feat(ui): add Replace Current open to Select Object -> Save As

invoke-ai · Oct 25, 2024 · 8e7cabd · 8e7cabd
1 parent a4c4237
commit 8e7cabd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/invokeai/frontend/web/public/locales/en.json b/invokeai/frontend/web/public/locales/en.json
@@ -1757,6 +1757,7 @@
         "newGallerySessionDesc": "This will clear the canvas and all settings except for your model selection. Generations will be sent to the gallery.",
         "newCanvasSession": "New Canvas Session",
         "newCanvasSessionDesc": "This will clear the canvas and all settings except for your model selection. Generations will be staged on the canvas.",
+        "replaceCurrent": "Replace Current",
         "controlMode": {
             "controlMode": "Control Mode",
             "balanced": "Balanced",

diff --git a/invokeai/frontend/web/src/features/controlLayers/components/SelectObject/SelectObject.tsx b/invokeai/frontend/web/src/features/controlLayers/components/SelectObject/SelectObject.tsx
@@ -42,6 +42,10 @@ const SelectObjectContent = memo(
     const hasImageState = useStore(adapter.segmentAnything.$hasImageState);
     const autoProcess = useAppSelector(selectAutoProcess);
 
+    const replaceCurrent = useCallback(() => {
+      adapter.segmentAnything.apply();
+    }, [adapter.segmentAnything]);
+
     const saveAsInpaintMask = useCallback(() => {
       adapter.segmentAnything.saveAs('inpaint_mask');
     }, [adapter.segmentAnything]);
@@ -142,17 +146,20 @@ const SelectObjectContent = memo(
               {t('controlLayers.selectObject.saveAs')}
             </MenuButton>
             <MenuList>
+              <MenuItem isDisabled={!hasImageState} onClick={replaceCurrent}>
+                {t('controlLayers.replaceCurrent')}
+              </MenuItem>
               <MenuItem isDisabled={!hasImageState} onClick={saveAsInpaintMask}>
-                {t('controlLayers.inpaintMask')}
+                {t('controlLayers.newInpaintMask')}
               </MenuItem>
               <MenuItem isDisabled={!hasImageState} onClick={saveAsRegionalGuidance}>
-                {t('controlLayers.regionalGuidance')}
+                {t('controlLayers.newRegionalGuidance')}
               </MenuItem>
               <MenuItem isDisabled={!hasImageState} onClick={saveAsControlLayer}>
-                {t('controlLayers.controlLayer')}
+                {t('controlLayers.newControlLayer')}
               </MenuItem>
               <MenuItem isDisabled={!hasImageState} onClick={saveAsRasterLayer}>
-                {t('controlLayers.rasterLayer')}
+                {t('controlLayers.newRasterLayer')}
               </MenuItem>
             </MenuList>
           </Menu>