update

app.py

@@ -43,14 +43,17 @@ def update_chat(_provider: str, _chat, _model: str, _temperature: float, _max_to
 
 
 def image_progress(_image):
+    mask = _image['layers'][0]
+    mask = mask.convert('L')
     _input = {
-        'img': _image['background'],
-        'mask': _image['layers'][0],
+        'img': _image['background'].convert('RGB'),
+        'mask': mask,
     }
     inpainting = pipeline(Tasks.image_inpainting, model='damo/cv_fft_inpainting_lama')
     result = inpainting(_input)
     vis_img = result[OutputKeys.OUTPUT_IMG]
-    return vis_img, _image['layers'][0]
+    vis_img = vis_img[..., ::-1]
+    return vis_img, mask
 
 
 with gr.Blocks() as app:
@@ -66,9 +69,14 @@ with gr.Blocks() as app:
                     additional_inputs=[chat_engine]
                 )
             with gr.Column(scale=1, min_width=300):
-                with gr.Accordion('Select Model', open=True):
+                with gr.Accordion('参数设置', open=True):
                     with gr.Column():
-                        provider = gr.Dropdown(label='Provider', choices=['DeepSeek', 'OpenRouter', 'Tongyi'], value='DeepSeek')
+                        provider = gr.Dropdown(
+                            label='模型厂商',
+                            choices=['DeepSeek', 'OpenRouter', 'Tongyi'],
+                            value='DeepSeek',
+                            info='不同模型厂商参数，效果和价格略有不同，请先设置好对应模型厂商的 API Key。',
+                        )
 
                     @gr.render(inputs=provider)
                     def show_model_config_panel(_provider):
@@ -192,14 +200,16 @@ with gr.Blocks() as app:
             with gr.Column(scale=2, min_width=600):
                 image = gr.ImageMask(
                     type='pil',
-                    brush=gr.Brush(colors=["rgba(0, 0, 0, 0.7)"]),
+                    brush=gr.Brush(colors=["rgba(255, 255, 255, 0.9)"]),
                 )
                 with gr.Row():
                     image_preview = gr.Image(label='图片预览')
                     mask_preview = gr.Image(label='蒙板预览')
-                image.change(image_progress, inputs=[image], outputs=[image_preview, mask_preview])
             with gr.Column(scale=1, min_width=300):
-                gr.Textbox(label="LoRA")
+                with gr.Accordion(label="参数设置"):
+                    gr.Checkbox(label="Refined", info="只支持 GPU, 开启将获得更好的效果")
+                    convert = gr.Button('转换')
+            convert.click(fn=image_progress, inputs=[image], outputs=[image_preview, mask_preview])
 
 
 app.launch(debug=settings.debug, show_api=False)

pyproject.toml

@@ -20,6 +20,7 @@ dependencies = [
     "setuptools==69.5.1",
     "oss2>=2.18.6",
     "kornia>=0.7.3",
+    "opencv-python>=4.10.0.84",
 ]
 readme = "README.md"
 requires-python = ">= 3.8"

requirements-dev.lock

@@ -194,12 +194,15 @@ numpy==1.26.4
     # via gradio
     # via langchain
     # via matplotlib
+    # via opencv-python
     # via pandas
     # via pyarrow
     # via scipy
     # via torchvision
 openai==1.35.13
     # via langchain-openai
+opencv-python==4.10.0.84
+    # via aitoolkits-webui
 orjson==3.10.6
     # via fastapi
     # via gradio

requirements.lock

@@ -194,12 +194,15 @@ numpy==1.26.4
     # via gradio
     # via langchain
     # via matplotlib
+    # via opencv-python
     # via pandas
     # via pyarrow
     # via scipy
     # via torchvision
 openai==1.35.13
     # via langchain-openai
+opencv-python==4.10.0.84
+    # via aitoolkits-webui
 orjson==3.10.6
     # via fastapi
     # via gradio

requirements.txt

@@ -193,12 +193,15 @@ numpy==1.26.4
     # via gradio
     # via langchain
     # via matplotlib
+    # via opencv-python
     # via pandas
     # via pyarrow
     # via scipy
     # via torchvision
 openai==1.35.13
     # via langchain-openai
+opencv-python==4.10.0.84
+    # via aitoolkits-webui
 orjson==3.10.6
     # via fastapi
     # via gradio