update

app.py

@@ -4,6 +4,7 @@ import fal_client
 import os
 import base64
 import io
+import time
 from PIL import Image
 from google import genai
 from google.genai.types import GenerateContentConfig, ImageConfig, Part
@@ -11,7 +12,7 @@ from google.genai.types import GenerateContentConfig, ImageConfig, Part
 RUNPOD_API_KEY = os.getenv("RUNPOD_API_KEY")
 FAL_KEY = os.getenv("FAL_KEY")
 GEMINI_API_KEY = os.getenv("GEMINI_VLM_KEY")
-QWEN_ENDPOINT_ID = os.getenv("ENDPOINTID")
+QWEN_ENDPOINT_ID = "jzpm1xin5cprff"
 
 os.environ["FAL_KEY"] = FAL_KEY if FAL_KEY else ""
 gemini_client = genai.Client(api_key=GEMINI_API_KEY) if GEMINI_API_KEY else None
@@ -21,11 +22,7 @@ DEFAULT_PROMPT = "Add furnishings and accessories to this room as an interior de
 def get_closest_ratio(pil_img):
     w, h = pil_img.size
     ratio = w / h
-    ratios = {
-        "9:16": 0.56, "2:3": 0.66, "3:4": 0.75, "4:5": 0.8,
-        "1:1": 1.0, "5:4": 1.25, "4:3": 1.33, "3:2": 1.5,
-        "16:9": 1.77, "21:9": 2.33
-    }
+    ratios = {"9:16": 0.56, "2:3": 0.66, "3:4": 0.75, "4:5": 0.8, "1:1": 1.0, "5:4": 1.25, "4:3": 1.33, "3:2": 1.5, "16:9": 1.77, "21:9": 2.33}
     return min(ratios, key=lambda x: abs(ratios[x] - ratio))
 
 def b64_to_pil(b64_str):
@@ -80,20 +77,23 @@ def run_gemini(image_bytes, prompt, ratio_str):
 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, None
+        yield None, None, None, None, "", "", ""
         return
 
     og_pil = bytes_to_pil(raw_bytes)
     ratio_str = get_closest_ratio(og_pil)
-    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
-    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, ratio_str)
-    yield og_pil, qwen_img, flux_img, gemini_img
+    yield og_pil, q_img, f_img, g_img, "⏳ Processing (~5s)...", "🕒 Pending...", "🕒 Pending..."
+    q_img = run_qwen(raw_b64, prompt)
+    
+    yield og_pil, q_img, f_img, g_img, "✅ Complete", "⏳ Processing (~12s)...", "🕒 Pending..."
+    f_img = run_fal_flux(web_url, prompt)
+    
+    yield og_pil, q_img, f_img, g_img, "✅ Complete", "✅ Complete", "⏳ Processing (~15s)..."
+    g_img = run_gemini(raw_bytes, prompt, ratio_str)
+    
+    yield og_pil, q_img, f_img, g_img, "✅ Complete", "✅ Complete", "✅ Complete"
 
 with gr.Blocks() as demo:
     gr.HTML("<h2 style='text-align: center; margin: 10px 0;'>🛋️ Interior Design Model Arena</h2>")
@@ -115,20 +115,31 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             gr.HTML("<center><b>QWEN-EDIT</b><br><small>$0.004 - $0.008</small></center>")
+            stat_q = gr.Markdown("🕒 Waiting...", elem_classes="status-msg")
             out_qwen = gr.Image(show_label=False, type="pil", height=350)
             
         with gr.Column():
             gr.HTML("<center><b>FLUX-2 EDIT</b><br><small>$0.03</small></center>")
+            stat_f = gr.Markdown("🕒 Waiting...", elem_classes="status-msg")
             out_fal = gr.Image(show_label=False, type="pil", height=350)
             
         with gr.Column():
             gr.HTML("<center><b>GEMINI 2.5 FLASH</b><br><small>$0.039</small></center>")
+            stat_g = gr.Markdown("🕒 Waiting...", elem_classes="status-msg")
             out_gemini = gr.Image(show_label=False, type="pil", height=350)
 
-    run_btn.click(fn=compare_all, inputs=[input_file, input_url, input_prompt], outputs=[out_og, out_qwen, out_fal, out_gemini])
+    run_btn.click(
+        fn=compare_all, 
+        inputs=[input_file, input_url, input_prompt], 
+        outputs=[out_og, out_qwen, out_fal, out_gemini, stat_q, stat_f, stat_g]
+    )
 
 if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft(),
-        css=".gradio-container {max-width: 98% !important} img {object-fit: contain !important;}"
+        css="""
+        .gradio-container {max-width: 98% !important} 
+        img {object-fit: contain !important;}
+        .status-msg {text-align: center; margin-top: -10px; font-size: 0.85em; color: #666;}
+        """
     )