update

app.py

@@ -44,11 +44,17 @@ def get_image_inputs(image_file, image_url):
 def run_qwen(raw_b64, prompt):
     url = f"https://api.runpod.ai/v2/{QWEN_ENDPOINT_ID}/runsync"
     headers = {"Content-Type": "application/json", "Authorization": f"Bearer {RUNPOD_API_KEY}"}
-    payload = {"input": {"image": raw_b64, "prompt": prompt, "seed": 42, "use_lightning": True, "true_guidance_scale": 2.5, "num_inference_steps": 4}}
+    payload = {
+        "input": {
+            "image": raw_b64, "prompt": prompt, "seed": 42,
+            "use_lightning": True, "true_guidance_scale": 2.5, "num_inference_steps": 4
+        }
+    }
     try:
         response = requests.post(url, headers=headers, json=payload, timeout=60)
         return b64_to_pil(response.json()["output"]["images"][0])
-    except: return None
+    except:
+        return None
 
 def run_fal_flux(image_url, prompt):
     try:
@@ -56,7 +62,8 @@ def run_fal_flux(image_url, prompt):
         result = handler.get()
         resp = requests.get(result['images'][0]['url'])
         return bytes_to_pil(resp.content)
-    except: return None
+    except:
+        return None
 
 def run_gemini(image_bytes, prompt):
     if not gemini_client: return None
@@ -64,11 +71,17 @@ def run_gemini(image_bytes, prompt):
         response = gemini_client.models.generate_content(
             model="gemini-2.5-flash-image",
             contents=[Part.from_bytes(data=image_bytes, mime_type="image/jpeg"), prompt],
-            config=GenerateContentConfig(response_modalities=["IMAGE"], image_config=ImageConfig(aspect_ratio="16:9"), candidate_count=1),
+            config=GenerateContentConfig(
+                response_modalities=["IMAGE"],
+                image_config=ImageConfig(aspect_ratio="16:9"),
+                candidate_count=1,
+            ),
         )
         for part in response.candidates[0].content.parts:
-            if part.inline_data: return bytes_to_pil(part.inline_data.data)
-    except: return None
+            if part.inline_data:
+                return bytes_to_pil(part.inline_data.data)
+    except:
+        return None
 
 def compare_all(image_file, image_url, prompt):
     raw_bytes, raw_b64, web_url = get_image_inputs(image_file, image_url)
@@ -77,46 +90,46 @@ def compare_all(image_file, image_url, prompt):
         return
 
     og_pil = bytes_to_pil(raw_bytes)
-    q_img, f_img, g_img = None, None, None
+    qwen_img, flux_img, gemini_img = None, None, None
     
-    yield og_pil, q_img, f_img, g_img
-    
-    q_img = run_qwen(raw_b64, prompt)
-    yield og_pil, q_img, f_img, g_img
-
-    f_img = run_fal_flux(web_url, prompt)
-    yield og_pil, q_img, f_img, g_img
-
-    g_img = run_gemini(raw_bytes, prompt)
-    yield og_pil, q_img, f_img, g_img
+    yield og_pil, qwen_img, flux_img, gemini_img
+    qwen_img = run_qwen(raw_b64, prompt)
+    yield og_pil, qwen_img, flux_img, gemini_img
+    flux_img = run_fal_flux(web_url, prompt)
+    yield og_pil, qwen_img, flux_img, gemini_img
+    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.Markdown("# 🛋️ Interior Design Model Arena")
+    gr.HTML("<h2 style='text-align: center;'>🛋️ Interior Design Model Arena</h2>")
     
-    with gr.Row(variant="panel"):
+    with gr.Row():
         with gr.Column(scale=3):
-            input_prompt = gr.Textbox(label="Design Instructions", value=DEFAULT_PROMPT, lines=3)
-        with gr.Column(scale=1):
-            input_file = gr.Image(label="Upload (Fixed Width)", type="filepath", height=120)
-        with gr.Column(scale=1):
-            input_url = gr.Textbox(label="Or Image URL", placeholder="http://...")
-            run_btn = gr.Button("🚀 GENERATE ALL", variant="primary", size="lg")
+            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=100)
+                input_url = gr.Textbox(label="URL", placeholder="Optional...")
+            run_btn = gr.Button("🚀 Generate Comparison", variant="primary")
             
-    gr.Separator()
+    gr.HTML("<hr>")
 
     with gr.Row():
         with gr.Column():
-            gr.Markdown("#### 🖼️ Original")
-            out_og = gr.Image(show_label=False, type="pil", height=400, show_download_button=False)
+            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")
+        
         with gr.Column():
-            gr.Markdown("#### ⚡ Qwen-Edit\n($0.006 avg)")
-            out_qwen = gr.Image(show_label=False, type="pil", height=400)
+            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")
+            
         with gr.Column():
-            gr.Markdown("#### 🎨 Flux-2\n($0.03)")
-            out_fal = gr.Image(show_label=False, type="pil", height=400)
+            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")
+            
         with gr.Column():
-            gr.Markdown("#### 💎 Gemini 2.5\n($0.039)")
-            out_gemini = gr.Image(show_label=False, type="pil", height=400)
+            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")
 
     run_btn.click(
         fn=compare_all,