From 19f1f07c955a897008dd037e24ea1dd98297a527 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Thu, 22 Sep 2022 21:40:36 +0800
Subject: [PATCH] SD: remove dice button

---
 .../app/src/components/Editor/Editor.tsx      | 24 ++----------
 .../app/src/components/Header/PromptInput.tsx |  2 +-
 .../src/components/SidePanel/SidePanel.tsx    | 37 +++++++------------
 lama_cleaner/app/src/event.ts                 |  1 -
 lama_cleaner/app/src/store/Atoms.tsx          |  3 ++
 5 files changed, 20 insertions(+), 47 deletions(-)

diff --git a/lama_cleaner/app/src/components/Editor/Editor.tsx b/lama_cleaner/app/src/components/Editor/Editor.tsx
index aabca4d..2c00dd6 100644
--- a/lama_cleaner/app/src/components/Editor/Editor.tsx
+++ b/lama_cleaner/app/src/components/Editor/Editor.tsx
@@ -41,7 +41,7 @@ import {
 } from '../../store/Atoms'
 import useHotKey from '../../hooks/useHotkey'
 import Croper from '../Croper/Croper'
-import emitter, { EVENT_PROMPT, EVENT_RERUN } from '../../event'
+import emitter, { EVENT_PROMPT } from '../../event'
 
 const TOOLBAR_SIZE = 200
 const BRUSH_COLOR = '#ffcc00bb'
@@ -309,6 +309,8 @@ export default function Editor(props: EditorProps) {
     emitter.on(EVENT_PROMPT, () => {
       if (hadDrawSomething()) {
         runInpainting(promptVal)
+      } else if (lastLineGroup.length !== 0) {
+        runInpainting(promptVal, true)
       } else {
         setToastState({
           open: true,
@@ -323,26 +325,6 @@ export default function Editor(props: EditorProps) {
     }
   }, [hadDrawSomething, runInpainting, prompt])
 
-  useEffect(() => {
-    emitter.on(EVENT_RERUN, () => {
-      if (hadDrawSomething()) {
-        runInpainting(promptVal)
-      } else if (lastLineGroup.length !== 0) {
-        runInpainting(promptVal, true)
-      } else {
-        setToastState({
-          open: true,
-          desc: 'Please draw mask on picture',
-          state: 'error',
-          duration: 1500,
-        })
-      }
-    })
-    return () => {
-      emitter.off(EVENT_RERUN)
-    }
-  }, [lastLineGroup, hadDrawSomething, runInpainting, prompt])
-
   const hadRunInpainting = () => {
     return renders.length !== 0
   }
diff --git a/lama_cleaner/app/src/components/Header/PromptInput.tsx b/lama_cleaner/app/src/components/Header/PromptInput.tsx
index bd667bc..64b2faa 100644
--- a/lama_cleaner/app/src/components/Header/PromptInput.tsx
+++ b/lama_cleaner/app/src/components/Header/PromptInput.tsx
@@ -45,7 +45,7 @@ const PromptInput = () => {
         onClick={handleRepaintClick}
         disabled={prompt.length === 0 || app.isInpainting}
       >
-        RePaint
+        Dream
       </Button>
     </div>
   )
diff --git a/lama_cleaner/app/src/components/SidePanel/SidePanel.tsx b/lama_cleaner/app/src/components/SidePanel/SidePanel.tsx
index 15aca9b..1624038 100644
--- a/lama_cleaner/app/src/components/SidePanel/SidePanel.tsx
+++ b/lama_cleaner/app/src/components/SidePanel/SidePanel.tsx
@@ -7,8 +7,6 @@ import NumberInputSetting from '../Settings/NumberInputSetting'
 import SettingBlock from '../Settings/SettingBlock'
 import Selector from '../shared/Selector'
 import { Switch, SwitchThumb } from '../shared/Switch'
-import Button from '../shared/Button'
-import emitter, { EVENT_RERUN } from '../../event'
 
 const INPUT_WIDTH = 30
 
@@ -17,10 +15,6 @@ const SidePanel = () => {
   const [open, toggleOpen] = useToggle(false)
   const [setting, setSettingState] = useRecoilState(settingState)
 
-  const onReRunBtnClick = () => {
-    emitter.emit(EVENT_RERUN)
-  }
-
   return (
     <div className="side-panel">
       <PopoverPrimitive.Root open={open}>
@@ -103,6 +97,19 @@ const SidePanel = () => {
               }}
             />
 
+            <NumberInputSetting
+              title="Mask Blur"
+              width={INPUT_WIDTH}
+              value={`${setting.sdMaskBlur}`}
+              desc="TODO"
+              onValue={value => {
+                const val = value.length === 0 ? 0 : parseInt(value, 10)
+                setSettingState(old => {
+                  return { ...old, sdMaskBlur: val }
+                })
+              }}
+            />
+
             <SettingBlock
               className="sub-setting-block"
               title="Sampler"
@@ -132,24 +139,6 @@ const SidePanel = () => {
                     alignItems: 'center',
                   }}
                 >
-                  <Button
-                    onClick={onReRunBtnClick}
-                    icon={
-                      <svg
-                        version="1.1"
-                        xmlns="http://www.w3.org/2000/svg"
-                        width="20px"
-                        height="20px"
-                        viewBox="0 0 66.459 66.46"
-                      >
-                        <path
-                          d="M65.542,11.777L33.467,0.037c-0.133-0.049-0.283-0.049-0.42,0L0.916,11.748c-0.242,0.088-0.402,0.32-0.402,0.576 l0.09,40.484c0,0.25,0.152,0.475,0.385,0.566l31.047,12.399v0.072c0,0.203,0.102,0.393,0.27,0.508 c0.168,0.111,0.379,0.135,0.57,0.062l0.385-0.154l0.385,0.154c0.072,0.028,0.15,0.045,0.227,0.045c0.121,0,0.24-0.037,0.344-0.105 c0.168-0.115,0.27-0.305,0.27-0.508v-0.072l31.047-12.399c0.232-0.093,0.385-0.316,0.385-0.568l0.027-40.453 C65.943,12.095,65.784,11.867,65.542,11.777z M32.035,63.134L3.052,51.562V15.013l28.982,11.572L32.035,63.134L32.035,63.134z M33.259,24.439L4.783,13.066l28.48-10.498l28.735,10.394L33.259,24.439z M63.465,51.562L34.484,63.134V26.585l28.981-11.572 V51.562z M14.478,38.021c0-1.692,1.35-2.528,3.016-1.867c1.665,0.663,3.016,2.573,3.016,4.269 c-0.001,1.692-1.351,2.529-3.017,1.867C15.827,41.626,14.477,39.714,14.478,38.021z M5.998,25.375c0-1.693,1.351-2.529,3.017-1.866 c1.666,0.662,3.016,2.572,3.016,4.267c0,1.695-1.351,2.529-3.017,1.867C7.347,28.979,5.998,27.069,5.998,25.375z M22.959,32.124 c0-1.694,1.351-2.53,3.017-1.867c1.666,0.663,3.016,2.573,3.016,4.267c0,1.695-1.352,2.53-3.017,1.867 C24.309,35.728,22.959,33.818,22.959,32.124z M5.995,43.103c0.001-1.692,1.351-2.529,3.017-1.867 c1.666,0.664,3.016,2.573,3.016,4.269c0,1.694-1.351,2.53-3.017,1.867C7.344,46.709,5.995,44.797,5.995,43.103z M22.957,49.853 c0.001-1.695,1.351-2.529,3.017-1.867s3.016,2.572,3.016,4.269c0,1.692-1.351,2.528-3.017,1.866 C24.306,53.458,22.957,51.546,22.957,49.853z M27.81,12.711c-0.766,1.228-3.209,2.087-5.462,1.917 c-2.253-0.169-3.46-1.301-2.695-2.528c0.765-1.227,3.207-2.085,5.461-1.916C27.365,10.352,28.573,11.484,27.81,12.711z M43.928,13.921c-0.764,1.229-3.208,2.086-5.46,1.917c-2.255-0.169-3.46-1.302-2.696-2.528c0.764-1.229,3.209-2.086,5.462-1.918 C43.485,11.563,44.693,12.695,43.928,13.921z M47.04,42.328c-1.041-1.278-0.764-3.705,0.619-5.421 c1.381-1.716,3.344-2.069,4.381-0.792c1.041,1.276,0.764,3.704-0.617,5.42S48.079,43.604,47.04,42.328z"
-                          fill="currentColor"
-                        />
-                      </svg>
-                    }
-                  />
-
                   {/* 每次会从服务器返回更新该值 */}
                   <NumberInputSetting
                     title=""
diff --git a/lama_cleaner/app/src/event.ts b/lama_cleaner/app/src/event.ts
index 1a38e51..e4344fb 100644
--- a/lama_cleaner/app/src/event.ts
+++ b/lama_cleaner/app/src/event.ts
@@ -1,7 +1,6 @@
 import mitt from 'mitt'
 
 export const EVENT_PROMPT = 'prompt'
-export const EVENT_RERUN = 'rerun_last_mask'
 
 const emitter = mitt()
 
diff --git a/lama_cleaner/app/src/store/Atoms.tsx b/lama_cleaner/app/src/store/Atoms.tsx
index 955845d..2790a60 100644
--- a/lama_cleaner/app/src/store/Atoms.tsx
+++ b/lama_cleaner/app/src/store/Atoms.tsx
@@ -149,6 +149,7 @@ export interface Settings {
   zitsWireframe: boolean
 
   // For SD
+  sdMaskBlur: number
   sdMode: SDMode
   sdStrength: number
   sdSteps: number
@@ -206,6 +207,7 @@ const defaultHDSettings: ModelsHDSettings = {
 
 export enum SDSampler {
   ddim = 'ddim',
+  pndm = 'pndm',
 }
 
 export enum SDMode {
@@ -229,6 +231,7 @@ export const settingStateDefault: Settings = {
   zitsWireframe: true,
 
   // SD
+  sdMaskBlur: 0,
   sdMode: SDMode.inpainting,
   sdStrength: 0.75,
   sdSteps: 50,