From dcbba6d96d14e6f25bcb5004c20acbde52467d39 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Thu, 18 May 2023 20:52:18 +0800
Subject: [PATCH] show prev mask when hover rerun button

---
 .../app/src/components/Header/Header.tsx        | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/lama_cleaner/app/src/components/Header/Header.tsx b/lama_cleaner/app/src/components/Header/Header.tsx
index 007dc95..5e0d0a7 100644
--- a/lama_cleaner/app/src/components/Header/Header.tsx
+++ b/lama_cleaner/app/src/components/Header/Header.tsx
@@ -19,7 +19,12 @@ import { ThemeChanger } from './ThemeChanger'
 import SettingIcon from '../Settings/SettingIcon'
 import PromptInput from './PromptInput'
 import CoffeeIcon from '../CoffeeIcon/CoffeeIcon'
-import emitter, { EVENT_CUSTOM_MASK, RERUN_LAST_MASK } from '../../event'
+import emitter, {
+  DREAM_BUTTON_MOUSE_ENTER,
+  DREAM_BUTTON_MOUSE_LEAVE,
+  EVENT_CUSTOM_MASK,
+  RERUN_LAST_MASK,
+} from '../../event'
 import { useImage } from '../../utils'
 import useHotKey from '../../hooks/useHotkey'
 
@@ -53,6 +58,14 @@ const Header = () => {
     emitter.emit(RERUN_LAST_MASK)
   }, [])
 
+  const onRerunMouseEnter = () => {
+    emitter.emit(DREAM_BUTTON_MOUSE_ENTER)
+  }
+
+  const onRerunMouseLeave = () => {
+    emitter.emit(DREAM_BUTTON_MOUSE_LEAVE)
+  }
+
   useHotKey(
     'r',
     () => {
@@ -207,6 +220,8 @@ const Header = () => {
               disabled={isInpainting}
               toolTip="Rerun last mask [r]"
               onClick={handleRerunLastMask}
+              onMouseEnter={onRerunMouseEnter}
+              onMouseLeave={onRerunMouseLeave}
             />
           </div>
         </div>