update

app.py

@@ -28,20 +28,17 @@ def bytes_to_pil(img_bytes):
     return Image.open(io.BytesIO(img_bytes))
 
 def get_image_inputs(image_file, image_url):
-    """Returns (raw_bytes, raw_b64, fal_url)"""
     if image_file:
         with open(image_file, "rb") as f:
             raw_bytes = f.read()
         raw_b64 = base64.b64encode(raw_bytes).decode('utf-8')
         fal_url = fal_client.upload_file(image_file)
         return raw_bytes, raw_b64, fal_url
-    
     elif image_url:
         resp = requests.get(image_url)
         raw_bytes = resp.content
         raw_b64 = base64.b64encode(raw_bytes).decode('utf-8')
         return raw_bytes, raw_b64, image_url
-    
     return None, None, None
 
 def run_qwen(raw_b64, prompt):
@@ -79,68 +76,70 @@ def run_gemini(image_bytes, prompt):
     if not gemini_client: return None
     try:
         response = gemini_client.models.generate_content(
-            model="gemini-2.5-flash-image", 
-            contents=[
-                Part.from_bytes(data=image_bytes, mime_type="image/jpeg"),
-                prompt
-            ],
+            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="4:3"), 
+                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 Exception as e:
-        print(f"Gemini Error: {e}")
+    except:
         return None
 
 def compare_all(image_file, image_url, prompt):
     raw_bytes, raw_b64, web_url = get_image_inputs(image_file, image_url)
     if not raw_bytes:
-        yield None, None, None
+        yield None, None, None, None
         return
 
+    og_pil = bytes_to_pil(raw_bytes)
     qwen_img, flux_img, gemini_img = None, None, None
-
+    
+    yield og_pil, qwen_img, flux_img, gemini_img
+    
     qwen_img = run_qwen(raw_b64, prompt)
-    yield qwen_img, flux_img, gemini_img
+    yield og_pil, qwen_img, flux_img, gemini_img
 
     flux_img = run_fal_flux(web_url, prompt)
-    yield qwen_img, flux_img, gemini_img
+    yield og_pil, qwen_img, flux_img, gemini_img
 
     gemini_img = run_gemini(raw_bytes, prompt)
-    yield qwen_img, flux_img, gemini_img
+    yield og_pil, qwen_img, flux_img, gemini_img
 
 with gr.Blocks(theme=gr.themes.Soft()) 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.Column(scale=1):
-            input_file = gr.Image(label="Upload Image", type="filepath")
-            input_url = gr.Textbox(label="OR: Image URL")
-            input_prompt = gr.Textbox(label="Prompt", value=DEFAULT_PROMPT, lines=4)
-            run_btn = gr.Button("Generate Comparison", variant="primary")
+            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")
             
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### Qwen-Edit\n**$0.004 - $0.008 /req**")
-            out_qwen = gr.Image(label="Qwen Result", type="pil")
-            
+            gr.Markdown("### Original\nInput Image")
+            out_og = gr.Image(label="Original", type="pil", height=320)
         with gr.Column():
-            gr.Markdown("### Flux-2 Edit\n**$0.03 /req**")
-            out_fal = gr.Image(label="Flux Result", type="pil")
-
+            gr.Markdown("### Qwen-Edit\n**$0.004 - $0.008**")
+            out_qwen = gr.Image(label="Qwen", type="pil", height=320)
+        with gr.Column():
+            gr.Markdown("### Flux-2 Edit\n**$0.03**")
+            out_fal = gr.Image(label="Flux", type="pil", height=320)
         with gr.Column():
-            gr.Markdown("### Gemini 2.5 Flash\n**$0.039 /req**")
-            out_gemini = gr.Image(label="Gemini Result", type="pil")
+            gr.Markdown("### Gemini 2.5 Flash\n**$0.039**")
+            out_gemini = gr.Image(label="Gemini", type="pil", height=320)
 
     run_btn.click(
         fn=compare_all,
         inputs=[input_file, input_url, input_prompt],
-        outputs=[out_qwen, out_fal, out_gemini]
+        outputs=[out_og, out_qwen, out_fal, out_gemini]
     )
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)