From 15dcb43f69658459971fa7095084957c16ef4b75 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Wed, 29 Jun 2022 21:47:26 +0800
Subject: [PATCH] change workspace key generate method

if use useCallback, Workspace will reload when switch theme
---
 lama_cleaner/app/src/App.tsx | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/lama_cleaner/app/src/App.tsx b/lama_cleaner/app/src/App.tsx
index 012dd7f..f5b849d 100644
--- a/lama_cleaner/app/src/App.tsx
+++ b/lama_cleaner/app/src/App.tsx
@@ -1,4 +1,4 @@
-import React, { useCallback, useEffect, useState } from 'react'
+import React, { useEffect, useMemo } from 'react'
 import { useKeyPressEvent } from 'react-use'
 import { useRecoilState } from 'recoil'
 import { nanoid } from 'nanoid'
@@ -34,18 +34,14 @@ function App() {
     document.body.setAttribute('data-theme', theme)
   }, [theme])
 
-  const getWorkspaceKey = useCallback(() => {
+  const workspaceId = useMemo(() => {
     return nanoid()
   }, [file])
 
   return (
     <div className="lama-cleaner">
       <Header />
-      {file ? (
-        <Workspace file={file} key={getWorkspaceKey()} />
-      ) : (
-        <LandingPage />
-      )}
+      {file ? <Workspace file={file} key={workspaceId} /> : <LandingPage />}
     </div>
   )
 }