update

app.py

@@ -165,7 +165,7 @@ def generate(
     randomize_cfg: bool,
     text_cfg_scale: float,
     image_cfg_scale: float,
-    weather_hide_video: bool,
+    weather_close_video: bool,
     decode_image_batch: int
 ):
     seed = random.randint(0, 100000) if randomize_seed else seed
@@ -226,9 +226,8 @@ def generate(
         edited_image = Image.fromarray(x_0.type(torch.uint8).cpu().numpy())
         edited_mask = Image.fromarray(x_1.type(torch.uint8).cpu().numpy())
 
-
         image_video_path = None
-        if not weather_hide_video:
+        if not weather_close_video:
             image_video = []
             
             for i in range(0, len(image_list), decode_image_batch):
@@ -286,21 +285,21 @@ def generate(
         return [int(seed), text_cfg_scale, image_cfg_scale, edited_image, mix_image, edited_mask_copy, mask_video_path, image_video_path, input_image_copy, mix_result_with_red_mask]
 
 def reset():
-    return [100, "Randomize Seed", 1372, "Fix CFG", 7.5, 1.5, None, None, None, None, None, None, None, "Hide Image Video", 10]
+    return [100, "Randomize Seed", 1372, "Fix CFG", 7.5, 1.5, None, None, None, None, None, None, None, "Close Image Video", 10]
 
 def get_example():
     return [
-        ["test/dufu.png", "black and white suit", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/girl.jpeg", "reflective sunglasses", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/road_sign.png", "stop sign", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/dufu.png", "blue medical mask", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/people_standing.png", "dark green pleated skirt", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/girl.jpeg", "shiny golden crown", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/dufu.png", "sunglasses", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/girl.jpeg", "diamond necklace", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/iron_man.jpg", "sunglasses", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/girl.jpeg", "the queen's crown", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
-        ["test/girl.jpeg", "gorgeous yellow gown", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/dufu.png", "black and white suit", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/girl.jpeg", "reflective sunglasses", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/road_sign.png", "stop sign", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/dufu.png", "blue medical mask", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/people_standing.png", "dark green pleated skirt", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/girl.jpeg", "shiny golden crown", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/dufu.png", "sunglasses", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/girl.jpeg", "diamond necklace", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/iron_man.jpg", "sunglasses", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/girl.jpeg", "the queen's crown", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
+        ["example_images/girl.jpeg", "gorgeous yellow gown", 100, "Fix Seed", 1372, "Fix CFG", 7.5, 1.5],
     ]
 
 with gr.Blocks(css="footer {visibility: hidden}") as demo:
@@ -348,20 +347,17 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
     
     with gr.Accordion('More outputs', open=False):
         with gr.Row():
-            # 单选框 选择是否显示视频
-            weather_hide_video = gr.Radio(
-                ["Show Image Video", "Hide Image Video"],
-                value="Hide Image Video",
+            weather_close_video = gr.Radio(
+                ["Show Image Video", "Close Image Video"],
+                value="Close Image Video",
                 type="index",
-                label="Dynamic Process",
-                show_label=False,
+                label="Image Generation Process Selection ()",
                 interactive=True,
             )
-            # decode的batch size，最大为steps的值
             decode_image_batch = gr.Number(value=10, precision=0, label="Decode Image Batch (<steps)", interactive=True)
         with gr.Row():
-            image_video = gr.Video(label="Real-time Image Output")
-            mask_video = gr.Video(label="Real-time Mask Output")
+            image_video = gr.Video(label="Image Video of Generation Process")
+            mask_video = gr.Video(label="Mask Video of Generation Process")
         with gr.Row():
             original_image = gr.Image(label=f"Original Image", type="pil", interactive=False)
             edited_image = gr.Image(label=f"Output Image", type="pil", interactive=False)
@@ -388,7 +384,7 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
             randomize_cfg,
             text_cfg_scale,
             image_cfg_scale,
-            weather_hide_video,
+            weather_close_video,
             decode_image_batch
         ],
         outputs=[seed, text_cfg_scale, image_cfg_scale, edited_image, mix_image, edited_mask, mask_video, image_video, original_image, mix_result_with_red_mask],
@@ -396,7 +392,7 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
     reset_button.click(
         fn=reset,
         inputs=[],
-        outputs=[steps, randomize_seed, seed, randomize_cfg, text_cfg_scale, image_cfg_scale, edited_image, mix_image, edited_mask, mask_video, image_video, original_image, mix_result_with_red_mask, weather_hide_video, decode_image_batch],
+        outputs=[steps, randomize_seed, seed, randomize_cfg, text_cfg_scale, image_cfg_scale, edited_image, mix_image, edited_mask, mask_video, image_video, original_image, mix_result_with_red_mask, weather_close_video, decode_image_batch],
     )