diff --git a/invokeai/frontend/web/src/features/canvas/components/IAICanvasStagingAreaToolbar.tsx b/invokeai/frontend/web/src/features/canvas/components/IAICanvasStagingAreaToolbar.tsx index 5c41626c95..1a539a8909 100644 --- a/invokeai/frontend/web/src/features/canvas/components/IAICanvasStagingAreaToolbar.tsx +++ b/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" /> } diff --git a/invokeai/frontend/web/src/features/canvas/components/IAICanvasToolbar/IAICanvasToolbar.tsx b/invokeai/frontend/web/src/features/canvas/components/IAICanvasToolbar/IAICanvasToolbar.tsx index 65a46d580a..5642269bda 100644 --- a/invokeai/frontend/web/src/features/canvas/components/IAICanvasToolbar/IAICanvasToolbar.tsx +++ b/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(