update

app.py

@@ -4,14 +4,24 @@ import fal_client
 import os
 import json
 import time
+import base64
+import io
+from PIL import Image
 
 RUNPOD_API_KEY = os.getenv("RUNPOD_API_KEY")
 FAL_KEY = os.getenv("FAL_KEY")
 
+QWEN_ENDPOINT_ID = os.environ["ENDPOINTID"]
+
 os.environ["FAL_KEY"] = FAL_KEY if FAL_KEY else ""
 
 DEFAULT_PROMPT = "Add furnishings and accessories to this room as an interior designer would do for a real estate staging. The generated image shall have the exact same dimensions as the original image and architectural details. Respect doorways and windows and make sure they are consistent with the source image and not blocked by furniture. Use cute accessories and with appropriate wall space, add smart simple graphic paintings. Use neutral colors with light colored accents to match the colors of the room. Give the area an attractive glow."
 
+def get_as_base64(url):
+    """Helper to convert image URL to base64 for the Qwen endpoint"""
+    response = requests.get(url)
+    return base64.b64encode(response.content).decode('utf-8')
+
 def run_banana(image_url, prompt):
     url = "https://api.runpod.ai/v2/nano-banana-edit/runsync"
     headers = {"Content-Type": "application/json", "Authorization": f"Bearer {RUNPOD_API_KEY}"}
@@ -23,31 +33,36 @@ def run_banana(image_url, prompt):
         }
     }
     try:
-        response = requests.post(url, headers=headers, json=payload, timeout=60)
+        response = requests.post(url, headers=headers, json=payload, timeout=90)
         return response.json()["output"][0] 
     except Exception as e:
-        return f"Error: {str(e)}"
+        return None
 
 def run_qwen(image_url, prompt):
-    url = "https://api.runpod.ai/v2/qwen-image-edit-2511/runsync"
+    """Updated to use endpoint jzpm1xin5cprff with base64 logic"""
+    url = f"https://api.runpod.ai/v2/{QWEN_ENDPOINT_ID}/runsync"
     headers = {"Content-Type": "application/json", "Authorization": f"Bearer {RUNPOD_API_KEY}"}
-    payload = {
-        "input": {
-            "enable_base64_output": False,
-            "enable_sync_mode": False,
-            "images": [image_url],
-            "output_format": "jpeg",
-            "prompt": prompt,
-            "seed": -1,
-            "size": "1536*1080"
-        }
-    }
+    
     try:
-        response = requests.post(url, headers=headers, json=payload, timeout=60)
+        img_b64 = get_as_base64(image_url)
+        payload = {
+            "input": {
+                "image": img_b64,
+                "prompt": prompt,
+                "seed": 42,
+                "use_lightning": True,
+                "true_guidance_scale": 2.5,
+                "num_inference_steps": 4
+            }
+        }
+        response = requests.post(url, headers=headers, json=payload, timeout=90)
         res_data = response.json()
-        return res_data["output"] if "output" in res_data else "Processing..."
+        
+        img_data = res_data["output"]["images"][0]
+        return f"data:image/png;base64,{img_data}"
     except Exception as e:
-        return f"Error: {str(e)}"
+        print(f"Qwen Error: {e}")
+        return None
 
 def run_fal_flux(image_url, prompt):
     try:
@@ -61,9 +76,11 @@ def run_fal_flux(image_url, prompt):
         result = handler.get()
         return result['images'][0]['url']
     except Exception as e:
-        return f"Error: {str(e)}"
+        return None
 
 def compare_models(image_url, prompt):
+    if not image_url:
+        return None, None, None
     res_banana = run_banana(image_url, prompt)
     res_qwen = run_qwen(image_url, prompt)
     res_fal = run_fal_flux(image_url, prompt)
@@ -71,7 +88,7 @@ def compare_models(image_url, prompt):
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🛋️ Interior Design Model Arena")
-    gr.Markdown("Compare RunPod Nano-Banana, Qwen-Edit, and Fal-Flux Side-by-Side.")
+    gr.Markdown("Compare specialized staging models side-by-side.")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -81,16 +98,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### Nano-Banana (RunPod)")
-            out_banana = gr.Image(label="Banana Result", show_download_button=True, interactive=False)
+            gr.Markdown("### Qwen-Edit (New)")
+            gr.Markdown("**Price: $0.004 - $0.008 /req**")
+            out_qwen = gr.Image(label="Qwen Result", interactive=False)
             
         with gr.Column():
-            gr.Markdown("### Qwen-Edit (RunPod)")
-            out_qwen = gr.Image(label="Qwen Result", show_download_button=True, interactive=False)
-            
+            gr.Markdown("### Flux-2 Edit")
+            gr.Markdown("**Price: $0.024 /req**")
+            out_fal = gr.Image(label="Flux Result", interactive=False)
+
         with gr.Column():
-            gr.Markdown("### Flux-2 Edit (Fal.ai)")
-            out_fal = gr.Image(label="Flux Result", show_download_button=True, interactive=False)
+            gr.Markdown("### Nano-Banana")
+            gr.Markdown("**Price: $0.039 /req**")
+            out_banana = gr.Image(label="Banana Result", interactive=False)
 
     run_btn.click(
         fn=compare_models,
@@ -98,4 +118,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=[out_banana, out_qwen, out_fal]
     )
 
-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)