created new ui using blocks

app.py

@@ -11,15 +11,42 @@ def generate_image():
     image.save("trial.png")
     img = plt.imread("trial.png")
     # Display the image (optional)
-    plt.imshow(img)
-    plt.axis("off")
-    plt.show()
+    # plt.imshow(img)
+    # plt.axis("off")
+    # plt.show()
     return img
 
 # gradio interface
+# import gradio as gr
+# iface = gr.Interface(fn=generate_image, inputs=None, outputs=[gr.Image(label="Generated Image", type="numpy", tool='editor')],
+#                 title="Image Data Generator",
+#                 description="This tool generates synthetic images using the DiffusionPipeline model.",
+#                 article="### Using the Image Data Generator\n\nSimply click 'Generate Image' to create a synthetic image. The generated image will be displayed below.")
+# iface.launch(debug=True)
+
+# blocks ui
 import gradio as gr
-iface = gr.Interface(fn=generate_image, inputs=None, outputs=[gr.Image(label="Generated Image", type="numpy", tool='editor')],
-                title="Image Data Generator",
-                description="This tool generates synthetic images using the DiffusionPipeline model.",
-                article="### Using the Image Data Generator\n\nSimply click 'Generate Image' to create a synthetic image. The generated image will be displayed below.")
-iface.launch(debug=True)
+
+def generate_multiple(num):
+    images = []
+    for i in range(num):
+      images.append(generate_image())
+
+with gr.Blocks() as app:
+    title = "Synthetic Image Generator"
+    with gr.Tab("Generate Single Image"):
+      gr.Markdown("## Using the Synthetic Generator\n\nSimply click 'Generate Image' to create a synthetic image. The generated image will be displayed below.")
+      gen_img = gr.Image(shape=(256,256), tool="select", type="numpy", label="Generated Image", container=True)
+      gen_button = gr.Button("Generate")
+    with gr.Tab("Generate Multiple Images"):
+      gr.Markdown(
+          """
+          ## Using the Synthetic Image Generator to generate multiple images
+          """
+      )
+      gen_number = gr.Slider(2, 5, label="Number of Images", info="Generate multiple images")
+      gen_images = gr.Gallery(label="Generated Images", object_fit="contain")
+      gen_m_button = gr.Button("Generate Images")
+    gen_button.click(generate_image, inputs=None, outputs=gen_img)
+    gen_m_button.click(generate_multiple, inputs=gen_number, outputs=gen_images)
+app.launch(debug=True)