fix: change UI to match UX

app.py

@@ -61,20 +61,23 @@ def generate(images, model_choice):
     return outputs
 
 
-with gr.Blocks() as demo:
-    gr.Markdown("# 🔥 Multi-API Image-to-Image Generator")
-
-    with gr.Row():
-        model_choice = gr.Dropdown(
-            ["Flux Kontext Pro", "Wan 2.5", "Qwen 3", "Nano Banana", "GPT"],
-            value="Flux Kontext Pro",
-            label="Select Model",
-            interactive=True,
-        )
-        btn = gr.Button("Generate", variant="primary")
-
-    image_input = gr.File(file_count="multiple", label="Upload your images")
-    gallery = gr.Gallery(label="Generated Results", columns=2)
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 🔥 Multi-API Image-to-Image Generator")
+
+    with gr.Row(equal_height=True):
+        # Left Column
+        with gr.Column(scale=1):
+            image_input = gr.File(file_count="multiple", label="Upload your images")
+            model_choice = gr.Dropdown(
+                ["Flux Kontext Pro", "Wan 2.5", "Qwen 3", "Nano Banana", "GPT"],
+                value="Flux Kontext Pro",
+                label="Select Model",
+            )
+
+        # Right Column
+        with gr.Column(scale=1):
+            gallery = gr.Gallery(label="Generated Results", columns=2, height=400)
+            btn = gr.Button("Generate", variant="primary", size="lg")
 
     btn.click(generate, [image_input, model_choice], gallery)