feat(ui): save the current staging image with shift+s

invokeai/frontend/web/src/features/canvas/components/IAICanvasStagingAreaToolbar.tsx

@@ -107,6 +107,17 @@ const IAICanvasStagingAreaToolbar = () => {
     );
   }, [dispatch, imageDTO]);
 
+  useHotkeys(
+    ['shift+s'],
+    () => {
+      shouldShowStagingImage && handleSaveToGallery();
+    },
+    {
+      preventDefault: true,
+    },
+    [shouldShowStagingImage, handleSaveToGallery]
+  );
+
   const handleDiscardStagingArea = useCallback(() => {
     dispatch(discardStagedImages());
   }, [dispatch]);
@@ -167,7 +178,7 @@ const IAICanvasStagingAreaToolbar = () => {
           colorScheme="invokeBlue"
         />
         <IconButton
-          tooltip={t('unifiedCanvas.saveToGallery')}
+          tooltip={`${t('unifiedCanvas.saveToGallery')} (Shift+S)`}
           aria-label={t('unifiedCanvas.saveToGallery')}
           isDisabled={!imageDTO || !imageDTO.is_intermediate}
           icon={<PiFloppyDiskBold />}

invokeai/frontend/web/src/features/canvas/components/IAICanvasToolbar/IAICanvasToolbar.tsx

@@ -88,13 +88,13 @@ const IAICanvasToolbar = () => {
   useHotkeys(
     ['shift+s'],
     () => {
-      handleSaveToGallery();
+      !isStaging && handleSaveToGallery();
     },
     {
-      enabled: () => !isStaging,
+      enabled: true,
       preventDefault: true,
     },
-    []
+    [isStaging]
   );
 
   useHotkeys(