update

app.py

@@ -100,36 +100,36 @@ def compare_all(image_file, image_url, prompt):
     gemini_img = run_gemini(raw_bytes, prompt)
     yield og_pil, qwen_img, flux_img, gemini_img
 
-with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 95% !important}") as demo:
-    gr.HTML("<h2 style='text-align: center;'>🛋️ Interior Design Model Arena</h2>")
+with gr.Blocks() as demo:
+    gr.HTML("<h2 style='text-align: center; margin-bottom: 20px;'>🛋️ 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)
+            input_prompt = gr.Textbox(label="Edit Prompt", value=DEFAULT_PROMPT, lines=4)
         with gr.Column(scale=2):
             with gr.Row():
-                input_file = gr.Image(label="Upload", type="filepath", height=100)
-                input_url = gr.Textbox(label="URL", placeholder="Optional...")
-            run_btn = gr.Button("🚀 Generate Comparison", variant="primary")
+                input_file = gr.Image(label="Upload Source", type="filepath", height=120)
+                input_url = gr.Textbox(label="OR: Image URL", placeholder="Paste URL here...")
+            run_btn = gr.Button("🚀 Generate Side-by-Side", variant="primary")
             
-    gr.HTML("<hr>")
+    gr.HTML("<hr style='margin: 20px 0;'>")
 
     with gr.Row():
         with gr.Column():
-            gr.HTML("<div style='text-align: center'><b>ORIGINAL</b><br><small>Source Reference</small></div>")
-            out_og = gr.Image(show_label=False, type="pil", height=350, object_fit="contain")
+            gr.HTML("<div style='text-align: center; margin-bottom: 10px;'><b>ORIGINAL</b><br><small>Input Reference</small></div>")
+            out_og = gr.Image(show_label=False, type="pil", height=380, object_fit="contain")
         
         with gr.Column():
-            gr.HTML("<div style='text-align: center'><b>QWEN-EDIT</b><br><small>$0.004 - $0.008</small></div>")
-            out_qwen = gr.Image(show_label=False, type="pil", height=350, object_fit="contain")
+            gr.HTML("<div style='text-align: center; margin-bottom: 10px;'><b>QWEN-EDIT</b><br><small>$0.004 - $0.008</small></div>")
+            out_qwen = gr.Image(show_label=False, type="pil", height=380, object_fit="contain")
             
         with gr.Column():
-            gr.HTML("<div style='text-align: center'><b>FLUX-2 EDIT</b><br><small>$0.03</small></div>")
-            out_fal = gr.Image(show_label=False, type="pil", height=350, object_fit="contain")
+            gr.HTML("<div style='text-align: center; margin-bottom: 10px;'><b>FLUX-2 EDIT</b><br><small>$0.03</small></div>")
+            out_fal = gr.Image(show_label=False, type="pil", height=380, object_fit="contain")
             
         with gr.Column():
-            gr.HTML("<div style='text-align: center'><b>GEMINI 2.5</b><br><small>$0.039</small></div>")
-            out_gemini = gr.Image(show_label=False, type="pil", height=350, object_fit="contain")
+            gr.HTML("<div style='text-align: center; margin-bottom: 10px;'><b>GEMINI 2.5</b><br><small>$0.039</small></div>")
+            out_gemini = gr.Image(show_label=False, type="pil", height=380, object_fit="contain")
 
     run_btn.click(
         fn=compare_all,
@@ -138,4 +138,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 95% !i
     )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(
+        server_name="0.0.0.0", 
+        server_port=7860,
+        theme=gr.themes.Soft(),
+        css=".gradio-container {max-width: 98% !important}"
+    )