add mask rerun button

This commit is contained in:
Qing 2022-11-15 21:53:16 +08:00
parent d7c3149f67
commit c54950c05c
6 changed files with 70 additions and 30 deletions

View File

@ -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",

View File

@ -1229,7 +1229,7 @@ export default function Editor() {
onClick={download}
/>
{settings.runInpaintingManually && (
{settings.runInpaintingManually && !isSD && (
<Button
toolTip="Run Inpainting"
tooltipPosition="top"

View File

@ -1,7 +1,13 @@
import { ArrowLeftIcon, UploadIcon } from '@heroicons/react/outline'
import { PlayIcon } from '@radix-ui/react-icons'
import React, { useState } from 'react'
import { useRecoilState, useRecoilValue } from 'recoil'
import { fileState, isInpaintingState, isSDState } from '../../store/Atoms'
import {
fileState,
isInpaintingState,
isSDState,
maskState,
} from '../../store/Atoms'
import Button from '../shared/Button'
import Shortcuts from '../Shortcuts/Shortcuts'
import { ThemeChanger } from './ThemeChanger'
@ -13,6 +19,7 @@ import emitter, { EVENT_CUSTOM_MASK } from '../../event'
const Header = () => {
const isInpainting = useRecoilValue(isInpaintingState)
const [file, setFile] = useRecoilState(fileState)
const [mask, setMask] = useRecoilState(maskState)
const [uploadElemId] = useState(`file-upload-${Math.random().toString()}`)
const [maskUploadElemId] = useState(`mask-upload-${Math.random().toString()}`)
const isSD = useRecoilValue(isSDState)
@ -29,7 +36,11 @@ const Header = () => {
}}
>
<label htmlFor={uploadElemId}>
<Button icon={<UploadIcon />} style={{ border: 0 }}>
<Button
icon={<UploadIcon />}
style={{ border: 0 }}
disabled={isInpainting}
>
<input
style={{ display: 'none' }}
id={uploadElemId}
@ -47,10 +58,15 @@ const Header = () => {
</Button>
</label>
<label
htmlFor={maskUploadElemId}
style={{ visibility: file ? 'visible' : 'hidden' }}
<div
style={{
visibility: file ? 'visible' : 'hidden',
display: 'flex',
justifyContent: 'center',
alignItems: 'center',
}}
>
<label htmlFor={maskUploadElemId}>
<Button style={{ border: 0 }} disabled={isInpainting}>
<input
style={{ display: 'none' }}
@ -67,6 +83,7 @@ const Header = () => {
// TODO: check mask size
console.info('Send custom mask')
emitter.emit(EVENT_CUSTOM_MASK, { mask: newFile })
setMask(newFile)
}
}}
accept="image/png, image/jpeg"
@ -74,6 +91,18 @@ const Header = () => {
Mask
</Button>
</label>
<Button
style={{
visibility: mask ? 'visible' : 'hidden',
}}
icon={<PlayIcon />}
onClick={() => {
if (mask) {
emitter.emit(EVENT_CUSTOM_MASK, { mask })
}
}}
/>
</div>
</div>
{isSD && file ? <PromptInput /> : <></>}

View File

@ -18,6 +18,11 @@ export const fileState = atom<File | undefined>({
default: undefined,
})
export const maskState = atom<File | undefined>({
key: 'maskState',
default: undefined,
})
export interface Rect {
x: number
y: number

View File

@ -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"

View File

@ -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())