restore gradio

app.py

@@ -1,3 +1,11 @@
+import subprocess
+
+# 执行shell命令
+command = "pip install gradio==3.47"
+result = subprocess.run(command, shell=True, capture_output=True, text=True)
+
+
+
 import os
 import json
 import re
@@ -31,6 +39,7 @@ from torchvision.utils import save_image
 from diffusers.models.attention_processor import AttnProcessor2_0
 
 
+# 在 python 中使用 pip 安装 3.41 版本的 gradio
 
 
 css = """
@@ -281,7 +290,7 @@ class GlobalText:
 
 global_text = GlobalText()
 
-
+gr.ImageMask()
 def load_mask_images(source,style,source_mask,style_mask,device,width,height,out_dir=None):
     # invert the image into noise map
     if isinstance(source['image'], np.ndarray):
@@ -323,6 +332,7 @@ def load_mask_images(source,style,source_mask,style_mask,device,width,height,out
     return source_image,style_image,source_mask,style_mask
 
 
+
 def ui():
     with gr.Blocks(css=css) as demo:
         gr.Markdown(
@@ -393,14 +403,14 @@ def ui():
             with gr.Tab("Configs"):
 
                 with gr.Row():
-                    source_image = gr.Image(label="Source Image",  elem_id="img2maskimg", sources="upload", interactive=True, type="pil",image_mode="RGB", height=512)
-                    style_image = gr.Image(label="Style Image", elem_id="img2maskimg", sources="upload", interactive=True, type="pil", image_mode="RGB", height=512)
+                    source_image = gr.Image(label="Source Image",  elem_id="img2maskimg", source="upload", interactive=True, type="pil", tool="sketch", image_mode="RGB", height=512)
+                    style_image = gr.Image(label="Style Image", elem_id="img2maskimg", source="upload", interactive=True, type="pil", tool="sketch", image_mode="RGB", height=512)
                 with gr.Row():
                     prompt_textbox = gr.Textbox(label="Prompt", value='head', lines=1)
                     negative_prompt_textbox = gr.Textbox(label="Negative prompt", lines=1)
                     # output_dir = gr.Textbox(label="output_dir", value='./results/')
 
-                with gr.Row(equal_height=False):
+                with gr.Row().style(equal_height=False):
                     with gr.Column():
                         width_slider     = gr.Slider(label="Width", value=512, minimum=256, maximum=1024, step=64)
                         height_slider    = gr.Slider(label="Height", value=512, minimum=256, maximum=1024, step=64)
@@ -487,22 +497,23 @@ def ui():
                         sam_style_btn = gr.Button(value="SAM Style")
                         send_style_btn = gr.Button(value="Send Style")
                     with gr.Row():
-                        source_image_sam = gr.Image(label="Source Image SAM",  elem_id="SourceimgSAM", sources="upload", interactive=True, type="pil", image_mode="RGB", height=512)
-                        style_image_sam = gr.Image(label="Style Image SAM", elem_id="StyleimgSAM", sources="upload", interactive=True, type="pil", image_mode="RGB", height=512)
+                        source_image_sam = gr.Image(label="Source Image SAM",  elem_id="SourceimgSAM", source="upload", interactive=True, type="pil", image_mode="RGB", height=512)
+                        style_image_sam = gr.Image(label="Style Image SAM", elem_id="StyleimgSAM", source="upload", interactive=True, type="pil", image_mode="RGB", height=512)
 
                     with gr.Row():
                         source_image_with_points = gr.Image(label="source Image with points", elem_id="style_image_with_points", type="pil", image_mode="RGB", height=256)
-                        source_mask = gr.Image(label="Source Mask",  elem_id="img2maskimg", sources="upload", interactive=True, type="numpy", image_mode="RGB", height=256)
+                        source_mask = gr.Image(label="Source Mask",  elem_id="img2maskimg", source="upload", interactive=True, type="numpy", image_mode="RGB", height=256)
 
                         style_image_with_points = gr.Image(label="Style Image with points", elem_id="style_image_with_points", type="pil", image_mode="RGB", height=256)
-                        style_mask = gr.Image(label="Style Mask", elem_id="img2maskimg", sources="upload", interactive=True, type="numpy", image_mode="RGB", height=256)
+                        style_mask = gr.Image(label="Style Mask", elem_id="img2maskimg", source="upload", interactive=True, type="numpy", image_mode="RGB", height=256)
+        
         gr.Examples(
-                [[os.path.join(os.path.dirname(__file__), "images/content/1.jpg"),
-                os.path.join(os.path.dirname(__file__), "images/style/1.jpg")]],
-                #/home/jin.liu/liujin/mycode/PortraitDiffusion/images/style/1.jpg
-                [source_image, style_image]
+            [[os.path.join(os.path.dirname(__file__), "gradio_app/images/content/1.jpg"),
+              os.path.join(os.path.dirname(__file__), "gradio_app/images/style/1.jpg")],
+            
+            ],
+            [source_image, style_image]
         )
-        
         inputs = [
             source_image, style_image, source_mask, style_mask,
             start_step, start_layer, Style_attn_step, 

gr4_test.py

@@ -0,0 +1,15 @@
+import gradio as gr
+
+cnt = 0
+
+def test():
+    cnt += 1
+    return f'triggered!{cnt}'
+
+
+with gr.Blocks() as demo:
+    sketch_pad = gr.ImageEditor(type="pil")
+    output_text = gr.Textbox(label='Output Text')
+    sketch_pad.change(test, outputs=[output_text])
+    
+demo.launch()