Update app.py

app.py

@@ -160,12 +160,26 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
         with gr.Column(scale=2):
             gallery2 = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery", interactive=False, show_share_button=True, format="png", preview=True, object_fit="cover", columns=2, rows=2)
 
-        for i, o in enumerate(output2):
-            num_images.change(lambda i, n: gr.update(visible=(i < n)), [i, num_images], o, queue=False)
-            gen_event2 = gr.on(triggers=[gen_button2.click, txt_input2.submit],
-                               fn=lambda i, n, m, t1, t2, n1, n2, n3, n4, n5: gen_fn(m, t1, t2, n1, n2, n3, n4, n5) if (i < n) else None,
-                               inputs=[i, num_images, model_choice2, txt_input2, neg_input2, height, width, steps, cfg, seed], outputs=[o])
-            o.change(add_gallery, [o, model_choice2, gallery2], [gallery2])
+for i, o in enumerate(output2):
+    # Change visibility based on num_images
+    num_images.change(
+        lambda num_images, i=i: gr.update(visible=(i < num_images)),  # `i` corresponds to the index of the image
+        [num_images],  # Only num_images needs to be an input
+        o,  # Outputs the updated visibility of the image `o`
+        queue=False
+    )
+    
+    # Image generation function
+    gen_event2 = gr.on(
+        triggers=[gen_button2.click, txt_input2.submit],
+        fn=gen_fn,
+        inputs=[i, num_images, model_choice2, txt_input2, neg_input2, height, width, steps, cfg, seed],
+        outputs=[o]
+    )
+    
+    # Update gallery when a new image is generated
+    o.change(add_gallery, [o, model_choice2, gallery2], [gallery2])
+
 
     gr.Markdown("")
 demo.launch(show_api=False, max_threads=400)