add rerun last mask button

lama_cleaner/app/src/components/Editor/Editor.tsx

@@ -63,6 +63,7 @@ import emitter, {
   EVENT_PROMPT,
   EVENT_CUSTOM_MASK,
   EVENT_PAINT_BY_EXAMPLE,
+  RERUN_LAST_MASK,
 } from '../../event'
 import FileSelect from '../FileSelect/FileSelect'
 import InteractiveSeg from '../InteractiveSeg/InteractiveSeg'
@@ -515,6 +516,28 @@ export default function Editor() {
     }
   }, [runInpainting])
 
+  useEffect(() => {
+    emitter.on(RERUN_LAST_MASK, () => {
+      if (lastLineGroup.length !== 0) {
+        // 使用上一次手绘的 mask 生成
+        runInpainting(true, undefined, prevInteractiveSegMask)
+      } else if (prevInteractiveSegMask) {
+        // 使用上一次 IS 的 mask 生成
+        runInpainting(false, undefined, prevInteractiveSegMask)
+      } else {
+        setToastState({
+          open: true,
+          desc: 'No mask to reuse',
+          state: 'error',
+          duration: 1500,
+        })
+      }
+    })
+    return () => {
+      emitter.off(RERUN_LAST_MASK)
+    }
+  }, [runInpainting])
+
   const hadRunInpainting = () => {
     return renders.length !== 0
   }

lama_cleaner/app/src/components/Header/Header.tsx

@@ -1,6 +1,6 @@
 import { FolderIcon, PhotoIcon } from '@heroicons/react/24/outline'
-import { PlayIcon } from '@radix-ui/react-icons'
-import React, { useState } from 'react'
+import { PlayIcon, ReloadIcon } from '@radix-ui/react-icons'
+import React, { useCallback, useState } from 'react'
 import { useRecoilState, useRecoilValue } from 'recoil'
 import * as PopoverPrimitive from '@radix-ui/react-popover'
 import {
@@ -19,7 +19,7 @@ import { ThemeChanger } from './ThemeChanger'
 import SettingIcon from '../Settings/SettingIcon'
 import PromptInput from './PromptInput'
 import CoffeeIcon from '../CoffeeIcon/CoffeeIcon'
-import emitter, { EVENT_CUSTOM_MASK } from '../../event'
+import emitter, { EVENT_CUSTOM_MASK, RERUN_LAST_MASK } from '../../event'
 import { useImage } from '../../utils'
 import useHotKey from '../../hooks/useHotkey'
 
@@ -49,6 +49,21 @@ const Header = () => {
     [showFileManager, enableFileManager, isInpainting]
   )
 
+  const handleRerunLastMask = useCallback(() => {
+    emitter.emit(RERUN_LAST_MASK)
+  }, [])
+
+  useHotKey(
+    'r',
+    () => {
+      if (!isInpainting) {
+        handleRerunLastMask()
+      }
+    },
+    {},
+    [isInpainting, handleRerunLastMask]
+  )
+
   const renderHeader = () => {
     return (
       <header>
@@ -57,7 +72,7 @@ const Header = () => {
             display: 'flex',
             justifyContent: 'center',
             alignItems: 'center',
-            gap: 8,
+            gap: 4,
           }}
         >
           {enableFileManager ? (
@@ -136,6 +151,7 @@ const Header = () => {
               </Button>
             </label>
 
+            {mask ? (
               <PopoverPrimitive.Root open={openMaskPopover}>
                 <PopoverPrimitive.Trigger
                   className="btn-primary side-panel-trigger"
@@ -171,6 +187,17 @@ const Header = () => {
                   </PopoverPrimitive.Content>
                 </PopoverPrimitive.Portal>
               </PopoverPrimitive.Root>
+            ) : (
+              <></>
+            )}
+
+            <Button
+              icon={<ReloadIcon style={{ height: 16, width: 16 }} />}
+              style={{ border: 0, gap: 0 }}
+              disabled={isInpainting}
+              toolTip="Rerun last mask [r]"
+              onClick={handleRerunLastMask}
+            />
           </div>
         </div>
 

lama_cleaner/app/src/components/Shortcuts/ShortcutsModal.tsx

@@ -64,6 +64,7 @@ export default function ShortcutsModal() {
         </div>
 
         <div className="shortcut-options-column">
+          <ShortCut content="Rerun last mask" keys={['R']} />
           <ShortCut content="Undo" keys={[CmdOrCtrl, 'Z']} />
           <ShortCut content="Redo" keys={[CmdOrCtrl, 'Shift', 'Z']} />
           <ShortCut content="Copy Result" keys={[CmdOrCtrl, 'C']} />

lama_cleaner/app/src/event.ts

@@ -12,6 +12,8 @@ export interface PaintByExampleEventData {
   image: File
 }
 
+export const RERUN_LAST_MASK = 'rerun_last_mask'
+
 const emitter = mitt()
 
 export default emitter