Update app.py

app.py

@@ -329,7 +329,10 @@ Please draw boxes accordingly on the sketch pad.""".format(len(boxes), len(groun
         fix_seed, rand_seed, actual_mask, style_cond_image, clip_model=clip_model,
     )
     blank_samples = batch_size % 2 if batch_size > 1 else 0
-    gen_images = [gr.Image.update(value=x, visible=True) for i,x in enumerate(gen_images)] 
+    gen_images = [gr.Image.update(value=x, visible=True) for i,x in enumerate(gen_images)] \
+                    + [gr.Image.update(value=None, visible=True) for _ in range(blank_samples)] \
+                    + [gr.Image.update(value=None, visible=False) for _ in range(4 - batch_size - blank_samples)]
+
     return gen_images + [state]
 
 
@@ -504,10 +507,13 @@ def clear(task, sketch_pad_trigger, batch_size, state,trigger_stage, switch_task
     sketch_pad_trigger = sketch_pad_trigger + 1
     trigger_stage = 0
     blank_samples = batch_size % 2 if batch_size > 1 else 0
-    out_images = [gr.Image.update(value=None, visible=True) for i in range(batch_size)] 
+    out_images = [gr.Image.update(value=None, visible=True) for i in range(batch_size)] \
+                    + [gr.Image.update(value=None, visible=True) for _ in range(blank_samples)] \
+                    + [gr.Image.update(value=None, visible=False) for _ in range(4 - batch_size - blank_samples)]
     state = {}
     return [None, sketch_pad_trigger, None, 1.0] + out_images + [state] + [trigger_stage]
 
+
 css = """
 #img2img_image, #img2img_image > .fixed-height, #img2img_image > .fixed-height > div, #img2img_image > .fixed-height > div > img
 {
@@ -593,7 +599,6 @@ with Blocks(
             with gr.Row():
                 sketch_pad = ImageMask(label="Sketch Pad", elem_id="img2img_image")
                 out_imagebox = gr.Image(type="pil",elem_id="my_image" ,label="Parsed Sketch Pad", shape=(512,512))
-                out_gen_1 = gr.Image(type="pil", visible=True, show_label=False)
             with gr.Row():
                 clear_btn = gr.Button(value='Clear')
                 gen_btn = gr.Button(value='Generate')
@@ -603,7 +608,7 @@ with Blocks(
             with gr.Accordion("Advanced Options", open=False):
                 with gr.Column():
                     alpha_sample = gr.Slider(minimum=0, maximum=1.0, step=0.1, value=0.3, label="Scheduled Sampling (τ)")
-                    guidance_scale = gr.Slider(minimum=0, maximum=50, step=0.5, value=7.5, label="CFG Guidance Scale")
+                    guidance_scale = gr.Slider(minimum=0, maximum=50, step=0.5, value=7.5, label="Guidance Scale")
                     batch_size = gr.Slider(minimum=1, maximum=4,visible=False, step=1, value=1, label="Number of Samples")
                     append_grounding = gr.Checkbox(value=True, label="Append grounding instructions to the caption")
                     use_actual_mask = gr.Checkbox(value=False, label="Use actual mask for inpainting", visible=False)
@@ -611,17 +616,17 @@ with Blocks(
                         fix_seed = gr.Checkbox(value=True, label="Fixed seed")
                         rand_seed = gr.Slider(minimum=0, maximum=1000, step=1, value=0, label="Seed")
                    
-                # with gr.Row():
-                #         use_style_cond = gr.Checkbox(value=False,visible=False, label="Enable Style Condition")
-                #         style_cond_image = gr.Image(type="pil",visible=False, label="Style Condition", interactive=True)
-        # with gr.Column(scale=4):
-        #     gr.HTML('<span style="font-size: 20px; font-weight: bold">Generated Images</span>')
-        #     with gr.Row():
-        #         out_gen_1 = gr.Image(type="pil", visible=True, show_label=False)
-        #         out_gen_2 = gr.Image(type="pil", visible=False, show_label=False)
-        #     with gr.Row():
-        #         out_gen_3 = gr.Image(type="pil", visible=False, show_label=False)
-        #         out_gen_4 = gr.Image(type="pil", visible=False, show_label=False)
+                with gr.Row():
+                        use_style_cond = gr.Checkbox(value=False,visible=False, label="Enable Style Condition")
+                        style_cond_image = gr.Image(type="pil",visible=False, label="Style Condition", interactive=True)
+        with gr.Column(scale=4):
+            gr.HTML('<span style="font-size: 20px; font-weight: bold">Generated Images</span>')
+            with gr.Row():
+                out_gen_1 = gr.Image(type="pil", visible=True, show_label=False)
+                out_gen_2 = gr.Image(type="pil", visible=False, show_label=False)
+            with gr.Row():
+                out_gen_3 = gr.Image(type="pil", visible=False, show_label=False)
+                out_gen_4 = gr.Image(type="pil", visible=False, show_label=False)
 
         state = gr.State({})
         
@@ -670,7 +675,7 @@ with Blocks(
         clear_btn.click(
             clear,
             inputs=[task, sketch_pad_trigger, batch_size,trigger_stage, state],
-            outputs=[sketch_pad, sketch_pad_trigger, out_imagebox, image_scale, out_gen_1, state, trigger_stage],
+            outputs=[sketch_pad, sketch_pad_trigger, out_imagebox, image_scale, out_gen_1, out_gen_2, out_gen_3, out_gen_4, state, trigger_stage],
             queue=False)
     
         sketch_pad_trigger.change(
@@ -689,7 +694,7 @@ with Blocks(
                 append_grounding, style_cond_image,
                 state,
             ],
-            outputs=[out_gen_1, state],
+            outputs=[out_gen_1, out_gen_2, out_gen_3, out_gen_4, state],
             queue=True
         )
         init_white_trigger.change(