update

app.py

@@ -92,11 +92,18 @@ def compare_all(image_file, image_url, prompt):
     og_pil = bytes_to_pil(raw_bytes)
     qwen_img, flux_img, gemini_img = None, None, None
     
+    # Step 0: Show Original
     yield og_pil, qwen_img, flux_img, gemini_img
+    
+    # Step 1: Qwen
     qwen_img = run_qwen(raw_b64, prompt)
     yield og_pil, qwen_img, flux_img, gemini_img
+
+    # Step 2: Flux
     flux_img = run_fal_flux(web_url, prompt)
     yield og_pil, qwen_img, flux_img, gemini_img
+
+    # Step 3: Gemini
     gemini_img = run_gemini(raw_bytes, prompt)
     yield og_pil, qwen_img, flux_img, gemini_img
 
@@ -104,32 +111,34 @@ with gr.Blocks() as demo:
     gr.HTML("<h2 style='text-align: center; margin-bottom: 10px;'>🛋️ Interior Design Model Arena</h2>")
     
     with gr.Row():
-        with gr.Column(scale=3):
-            input_prompt = gr.Textbox(label="Edit Prompt", value=DEFAULT_PROMPT, lines=3)
         with gr.Column(scale=2):
-            with gr.Row():
-                input_file = gr.Image(label="Upload", type="filepath", height=80)
-                input_url = gr.Textbox(label="OR URL", placeholder="Paste URL...")
-            run_btn = gr.Button("🚀 Run Comparison", variant="primary")
-            
-    gr.HTML("<hr style='margin: 10px 0;'>")
+            input_prompt = gr.Textbox(label="Edit Prompt", value=DEFAULT_PROMPT, lines=4)
+        with gr.Column(scale=1):
+            input_file = gr.Image(label="Upload Source Image", type="filepath", height=130)
+            input_url = gr.Textbox(label="OR: Image URL", placeholder="Paste URL...")
+            run_btn = gr.Button("🚀 Generate All Models", variant="primary")
 
+    gr.HTML("<div style='margin: 10px 0; border-bottom: 1px solid #ddd;'></div>")
+
+    # TOP: Original Image
     with gr.Row():
         with gr.Column():
-            gr.HTML("<center><b>ORIGINAL</b><br><small>Reference</small></center>")
-            out_og = gr.Image(show_label=False, type="pil", height=350)
-        
+            gr.HTML("<center><b>SOURCE ORIGINAL</b></center>")
+            out_og = gr.Image(show_label=False, type="pil", height=280)
+
+    # BOTTOM: Three Models
+    with gr.Row():
         with gr.Column():
             gr.HTML("<center><b>QWEN-EDIT</b><br><small>$0.004-$0.008</small></center>")
-            out_qwen = gr.Image(show_label=False, type="pil", height=350)
+            out_qwen = gr.Image(show_label=False, type="pil", height=280)
             
         with gr.Column():
             gr.HTML("<center><b>FLUX-2 EDIT</b><br><small>$0.03</small></center>")
-            out_fal = gr.Image(show_label=False, type="pil", height=350)
+            out_fal = gr.Image(show_label=False, type="pil", height=280)
             
         with gr.Column():
-            gr.HTML("<center><b>GEMINI 2.5</b><br><small>$0.039</small></center>")
-            out_gemini = gr.Image(show_label=False, type="pil", height=350)
+            gr.HTML("<center><b>GEMINI 2.5 FLASH</b><br><small>$0.039</small></center>")
+            out_gemini = gr.Image(show_label=False, type="pil", height=280)
 
     run_btn.click(
         fn=compare_all,
@@ -142,5 +151,5 @@ if __name__ == "__main__":
         server_name="0.0.0.0", 
         server_port=7860,
         theme=gr.themes.Soft(),
-        css=".gradio-container {max-width: 98% !important} img {object-fit: contain !important;}"
+        css=".gradio-container {max-width: 95% !important} img {object-fit: contain !important;}"
     )