Update app.py

app.py

@@ -175,25 +175,9 @@ def gen(prompt: str, system_prompt: str, token: str, accountId: str):
         for i, p in enumerate(failed_prompts, 1):
             text += f"\n{i}. {p}"
     
-    # Create grid of images
-    if images:
-        cols = min(5, len(images))
-        rows = (len(images) + cols - 1) // cols
-        grid_width = cols * 512  # assuming each image is 512x512
-        grid_height = rows * 512
-        grid_image = Image.new('RGB', (grid_width, grid_height))
-        
-        for idx, img in enumerate(images):
-            x = (idx % cols) * 512
-            y = (idx // cols) * 512
-            img = img.resize((512, 512))
-            grid_image.paste(img, (x, y))
-            
-        return text, grid_image
-    else:
-        return text, None
+    return text, images
 
-# Gradio Interface
+# Modify Gradio interface
 with gr.Blocks(theme="ocean") as demo:
     gr.HTML("<h1><center>ImagenX</center></h1>")
     prompt = gr.Textbox(label='Enter Your Prompt', placeholder="Enter prompt...")
@@ -201,7 +185,15 @@ with gr.Blocks(theme="ocean") as demo:
         sendBtn = gr.Button(variant='primary')
         clearBtn = gr.ClearButton([prompt])
     gen_text = gr.Textbox(label="Optimization")
-    gen_img = gr.Image(type="pil", label='Generated Image', height=600)
+    # Ganti Image dengan Gallery
+    gen_img = gr.Gallery(
+        label="Generated Images",
+        show_label=True,
+        elem_id="gallery",
+        columns=[5],
+        rows=[2],
+        height="auto"
+    )
     with gr.Accordion("Advanced Options", open=True):
         system_prompt = gr.Textbox(
             value = PRESET_PROMPT,
@@ -231,7 +223,4 @@ with gr.Blocks(theme="ocean") as demo:
             accountId
         ],
         outputs=[gen_text, gen_img]
-    )
-
-if __name__ == "__main__":
-    demo.queue(api_open=False).launch(show_api=False, share=False)
+    )