update

app.py

@@ -101,35 +101,35 @@ def compare_all(image_file, image_url, prompt):
     yield og_pil, qwen_img, flux_img, gemini_img
 
 with gr.Blocks() as demo:
-    gr.HTML("<h2 style='text-align: center; margin-bottom: 20px;'>🛋️ Interior Design Model Arena</h2>")
+    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=4)
+            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 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")
+                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: 20px 0;'>")
+    gr.HTML("<hr style='margin: 10px 0;'>")
 
     with gr.Row():
         with gr.Column():
-            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")
+            gr.HTML("<center><b>ORIGINAL</b><br><small>Reference</small></center>")
+            out_og = gr.Image(show_label=False, type="pil", height=350)
         
         with gr.Column():
-            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")
+            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)
             
         with gr.Column():
-            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")
+            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)
             
         with gr.Column():
-            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")
+            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)
 
     run_btn.click(
         fn=compare_all,
@@ -142,5 +142,5 @@ if __name__ == "__main__":
         server_name="0.0.0.0", 
         server_port=7860,
         theme=gr.themes.Soft(),
-        css=".gradio-container {max-width: 98% !important}"
+        css=".gradio-container {max-width: 98% !important} img {object-fit: contain !important;}"
     )