From 5429bf2e87271fe687bf53f7eff46cd7bf4ad353 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Thu, 22 Sep 2022 22:37:41 +0800
Subject: [PATCH] fix prompt input ctrl + z

---
 lama_cleaner/app/src/components/Editor/Editor.tsx     | 11 +++++++++--
 .../app/src/components/Header/PromptInput.tsx         |  7 +++++++
 .../app/src/components/SidePanel/SidePanel.tsx        |  2 +-
 3 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/lama_cleaner/app/src/components/Editor/Editor.tsx b/lama_cleaner/app/src/components/Editor/Editor.tsx
index 98f8ca3..57a2fca 100644
--- a/lama_cleaner/app/src/components/Editor/Editor.tsx
+++ b/lama_cleaner/app/src/components/Editor/Editor.tsx
@@ -640,6 +640,10 @@ export default function Editor(props: EditorProps) {
 
   // Handle Cmd+Z
   const undoPredicate = (event: KeyboardEvent) => {
+    // TODO: fix prompt input ctrl+z
+    if (isSD) {
+      return false
+    }
     const isCmdZ =
       (event.metaKey || event.ctrlKey) && !event.shiftKey && event.key === 'z'
     // Handle tab switch
@@ -654,7 +658,7 @@ export default function Editor(props: EditorProps) {
     return false
   }
 
-  useKey(undoPredicate, undo, undefined, [undoStroke, undoRender])
+  useKey(undoPredicate, undo, undefined, [undoStroke, undoRender, isSD])
 
   const disableUndo = () => {
     if (isInpainting) {
@@ -714,6 +718,9 @@ export default function Editor(props: EditorProps) {
 
   // Handle Cmd+shift+Z
   const redoPredicate = (event: KeyboardEvent) => {
+    if (isSD) {
+      return false
+    }
     const isCmdZ =
       (event.metaKey || event.ctrlKey) &&
       event.shiftKey &&
@@ -730,7 +737,7 @@ export default function Editor(props: EditorProps) {
     return false
   }
 
-  useKey(redoPredicate, redo, undefined, [redoStroke, redoRender])
+  useKey(redoPredicate, redo, undefined, [redoStroke, redoRender, isSD])
 
   const disableRedo = () => {
     if (isInpainting) {
diff --git a/lama_cleaner/app/src/components/Header/PromptInput.tsx b/lama_cleaner/app/src/components/Header/PromptInput.tsx
index 64b2faa..8b262e0 100644
--- a/lama_cleaner/app/src/components/Header/PromptInput.tsx
+++ b/lama_cleaner/app/src/components/Header/PromptInput.tsx
@@ -32,12 +32,19 @@ const PromptInput = () => {
     }
   })
 
+  const onKeyUp = (e: React.KeyboardEvent) => {
+    if (e.key === 'Enter') {
+      handleRepaintClick()
+    }
+  }
+
   return (
     <div className="prompt-wrapper">
       <TextInput
         ref={ref}
         value={prompt}
         onInput={handleOnInput}
+        onKeyUp={onKeyUp}
         placeholder="I want to repaint of..."
       />
       <Button
diff --git a/lama_cleaner/app/src/components/SidePanel/SidePanel.tsx b/lama_cleaner/app/src/components/SidePanel/SidePanel.tsx
index 1624038..a2faa18 100644
--- a/lama_cleaner/app/src/components/SidePanel/SidePanel.tsx
+++ b/lama_cleaner/app/src/components/SidePanel/SidePanel.tsx
@@ -12,7 +12,7 @@ const INPUT_WIDTH = 30
 
 // TODO: 添加收起来的按钮
 const SidePanel = () => {
-  const [open, toggleOpen] = useToggle(false)
+  const [open, toggleOpen] = useToggle(true)
   const [setting, setSettingState] = useRecoilState(settingState)
 
   return (