Update app.py

app.py

@@ -78,7 +78,7 @@ def base64_to_image(b64: str) -> Image.Image | None:
 
 # ─────────  UI ↔ Backend bridge ─────────
 def call_backend_with_retry(input_image: Image.Image, category: str, gender: str, *, max_retries: int = MAX_RETRIES):
-    run_btn.update(interactive=False)
+    
     """Single‑shot call (no more than `max_retries` times)."""
 
     if input_image is None:
@@ -124,8 +124,8 @@ def call_backend_with_retry(input_image: Image.Image, category: str, gender: str
             if attempt == max_retries - 1:
                 return None, None, f"❌ {e}"
             time.sleep(1)
-    run_btn.update(interactive=True)
-    return None, None, "❌ Unknown error"
+    
+    return run_btn.update(interactive=False), None, None, "❌ Unknown error", run_btn.update(interactive=True)
 
 # ─────────  CSS ─────────
 custom_css = """
@@ -351,7 +351,7 @@ with gr.Blocks(css=custom_css, title="Jewellery Photography Preview") as demo:
     run_btn.click(
         fn=call_backend_with_retry,
         inputs=[input_img, category, gender],
-        outputs=[out_overlay, out_bg, out_status],
+        outputs=[run_btn,out_overlay, out_bg, out_status,run_btn],
         concurrency_limit=1,
         show_progress=True,
     )