Update app.py

app.py

@@ -57,7 +57,7 @@ def control_ui():
                 with gr.Row():
                     use_inp = gr.Checkbox(label='启用', value=False)
                     overlay = gr.Checkbox(label='覆盖原图', value=True)
-                inp_img = gr.Image(label="上传图片", value=None, interactive=True, type="pil", tool="sketch")
+                inp_img = gr.ImageEditor(label="上传图片", value=None, interactive=True, type="pil", eraser=False, transforms=None, brush='imagemask')
                 inp_str = gr.Slider(label='重绘强度', value=0.7, minimum=0, maximum=0.99, step=0.01)
                 reuse_img_inp = gr.Button(value='使用上一次生成的图片')
     with gr.Row():
@@ -105,7 +105,7 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
 
 
 def preview_ui():
-    with gr.Blocks(css='#preview_image { height: 100%;}') as page:
+    with gr.Blocks(css='#preview_image { height: 100%;}'):
         image = gr.Image(elem_id='preview_image', interactive=False, type='pil')
         info = gr.JSON(value={}, label="Submitted Payload")
     return image, info
@@ -147,7 +147,7 @@ def ui():
                            inputs=[png_items] + paras,
                            outputs=paras)
             png2main.click(fn=None,
-                           _js="(x) => { if (x !== null) document.getElementById('client_ui_main-button').click(); return null; }",
+                           js="(x) => { if (x !== null) document.getElementById('client_ui_main-button').click(); return null; }",
                            inputs=image)
             image.change(read_info_from_image, inputs=image, outputs=[info, png_items])
     return website