diff --git a/lama_cleaner/app/package.json b/lama_cleaner/app/package.json
index 47c148e..92d7905 100644
--- a/lama_cleaner/app/package.json
+++ b/lama_cleaner/app/package.json
@@ -6,6 +6,7 @@
"dependencies": {
"@heroicons/react": "^1.0.4",
"@radix-ui/react-dialog": "0.1.8-rc.25",
+ "@radix-ui/react-icons": "^1.1.1",
"@radix-ui/react-popover": "^1.0.0",
"@radix-ui/react-select": "0.1.2-rc.27",
"@radix-ui/react-switch": "^0.1.5",
diff --git a/lama_cleaner/app/src/components/Editor/Editor.tsx b/lama_cleaner/app/src/components/Editor/Editor.tsx
index 22da920..003313b 100644
--- a/lama_cleaner/app/src/components/Editor/Editor.tsx
+++ b/lama_cleaner/app/src/components/Editor/Editor.tsx
@@ -1229,7 +1229,7 @@ export default function Editor() {
onClick={download}
/>
- {settings.runInpaintingManually && (
+ {settings.runInpaintingManually && !isSD && (
-
+
+ }
+ onClick={() => {
+ if (mask) {
+ emitter.emit(EVENT_CUSTOM_MASK, { mask })
+ }
+ }}
+ />
+
{isSD && file ? : <>>}
diff --git a/lama_cleaner/app/src/store/Atoms.tsx b/lama_cleaner/app/src/store/Atoms.tsx
index c3aefc2..25002c6 100644
--- a/lama_cleaner/app/src/store/Atoms.tsx
+++ b/lama_cleaner/app/src/store/Atoms.tsx
@@ -18,6 +18,11 @@ export const fileState = atom({
default: undefined,
})
+export const maskState = atom({
+ key: 'maskState',
+ default: undefined,
+})
+
export interface Rect {
x: number
y: number
diff --git a/lama_cleaner/app/yarn.lock b/lama_cleaner/app/yarn.lock
index 1cba09b..2b3e340 100644
--- a/lama_cleaner/app/yarn.lock
+++ b/lama_cleaner/app/yarn.lock
@@ -1762,6 +1762,11 @@
"@radix-ui/react-primitive" "1.0.0"
"@radix-ui/react-use-callback-ref" "1.0.0"
+"@radix-ui/react-icons@^1.1.1":
+ version "1.1.1"
+ resolved "https://registry.npmmirror.com/@radix-ui/react-icons/-/react-icons-1.1.1.tgz#38d2aa548035dd3b799c169bd17177b1cec3152b"
+ integrity sha512-xc3wQC59rsFylVbSusQCrrM+6695ppF730Q6yqzhRdqDcRNWIm2R6ngpzBoSOQMcwnq4p805F+Gr7xo4fmtN1A==
+
"@radix-ui/react-id@0.1.5":
version "0.1.5"
resolved "https://registry.npmmirror.com/@radix-ui/react-id/-/react-id-0.1.5.tgz#010d311bedd5a2884c1e9bb6aaaa4e6cc1d1d3b8"
diff --git a/lama_cleaner/server.py b/lama_cleaner/server.py
index f7014b0..6211154 100644
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -57,7 +57,7 @@ BUILD_DIR = os.environ.get("LAMA_CLEANER_BUILD_DIR", "app/build")
class NoFlaskwebgui(logging.Filter):
def filter(self, record):
- return "GET /flaskwebgui-keep-server-alive" not in record.getMessage()
+ return "flaskwebgui-keep-server-alive" not in record.getMessage()
logging.getLogger("werkzeug").addFilter(NoFlaskwebgui())