update

app.py

@@ -44,33 +44,19 @@ 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:
-        handler = fal_client.submit(
-            "fal-ai/flux-2/edit",
-            arguments={"prompt": prompt, "image_urls": [image_url]}
-        )
+        handler = fal_client.submit("fal-ai/flux-2/edit", arguments={"prompt": prompt, "image_urls": [image_url]})
         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
@@ -78,17 +64,11 @@ 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)
@@ -97,43 +77,46 @@ def compare_all(image_file, image_url, prompt):
         return
 
     og_pil = bytes_to_pil(raw_bytes)
-    qwen_img, flux_img, gemini_img = None, None, None
+    q_img, f_img, g_img = None, None, None
     
-    yield og_pil, qwen_img, flux_img, gemini_img
+    yield og_pil, q_img, f_img, g_img
     
-    qwen_img = run_qwen(raw_b64, prompt)
-    yield og_pil, qwen_img, flux_img, gemini_img
+    q_img = run_qwen(raw_b64, prompt)
+    yield og_pil, q_img, f_img, g_img
 
-    flux_img = run_fal_flux(web_url, prompt)
-    yield og_pil, qwen_img, flux_img, gemini_img
+    f_img = run_fal_flux(web_url, prompt)
+    yield og_pil, q_img, f_img, g_img
 
-    gemini_img = run_gemini(raw_bytes, prompt)
-    yield og_pil, qwen_img, flux_img, gemini_img
+    g_img = run_gemini(raw_bytes, prompt)
+    yield og_pil, q_img, f_img, g_img
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 95% !important}") as demo:
     gr.Markdown("# 🛋️ Interior Design Model Arena")
     
-    with gr.Row():
-        with gr.Column(scale=2):
-            input_prompt = gr.Textbox(label="Prompt", value=DEFAULT_PROMPT, lines=3)
+    with gr.Row(variant="panel"):
+        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 Source", type="filepath", height=180)
-            input_url = gr.Textbox(label="Source URL", placeholder="Optional URL...")
-            run_btn = gr.Button("Compare Models", variant="primary")
+            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")
             
+    gr.Separator()
+
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### Original\nInput Image")
-            out_og = gr.Image(label="Original", type="pil", height=320)
+            gr.Markdown("#### 🖼️ Original")
+            out_og = gr.Image(show_label=False, type="pil", height=400, show_download_button=False)
         with gr.Column():
-            gr.Markdown("### Qwen-Edit\n**$0.004 - $0.008**")
-            out_qwen = gr.Image(label="Qwen", type="pil", height=320)
+            gr.Markdown("#### ⚡ Qwen-Edit\n($0.006 avg)")
+            out_qwen = gr.Image(show_label=False, type="pil", height=400)
         with gr.Column():
-            gr.Markdown("### Flux-2 Edit\n**$0.03**")
-            out_fal = gr.Image(label="Flux", type="pil", height=320)
+            gr.Markdown("#### 🎨 Flux-2\n($0.03)")
+            out_fal = gr.Image(show_label=False, type="pil", height=400)
         with gr.Column():
-            gr.Markdown("### Gemini 2.5 Flash\n**$0.039**")
-            out_gemini = gr.Image(label="Gemini", type="pil", height=320)
+            gr.Markdown("#### 💎 Gemini 2.5\n($0.039)")
+            out_gemini = gr.Image(show_label=False, type="pil", height=400)
 
     run_btn.click(
         fn=compare_all,