Fix: .then() chaining for instant status, remove gr.File, robust error return

app.py

@@ -45,7 +45,7 @@ def _make_mask(size: int, cloth_type: str) -> Image.Image:
     return mask
 
 # ---------------------------------------------------------------------------
-# GPU inference
+# GPU inference — returns images + status string
 # ---------------------------------------------------------------------------
 _pipe = None
 
@@ -59,7 +59,7 @@ def run_tryon(
     seed: int,
 ):
     if person_image is None or garment_image is None:
-        raise gr.Error("Please upload both a person photo and a garment image.")
+        return None, "❌ Please upload both a person photo and a garment image."
 
     global _pipe
     if _pipe is None:
@@ -91,10 +91,9 @@ def run_tryon(
 
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
     for i, img in enumerate(output_images):
-        path = os.path.join(OUTPUT_DIR, f"tryon_{timestamp}_{i}.png")
-        img.save(path, format="PNG")
+        img.save(os.path.join(OUTPUT_DIR, f"tryon_{timestamp}_{i}.png"), format="PNG")
 
-    return output_images
+    return output_images, "✅ Done! Right-click an image in the gallery to save it."
 
 # ---------------------------------------------------------------------------
 # Gradio UI
@@ -103,15 +102,14 @@ with gr.Blocks(title="Virtual Try-On", theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         "# 👗 Virtual Try-On\n"
         "Upload a **person photo** and a **garment image**, select the type, then click **Try On**.\n\n"
-        "> Runs on **Hugging Face ZeroGPU** (free A10G) — no local GPU needed.\n\n"
-        "> ⏳ After clicking, the button shows a **loading spinner** — that means it is working.  \n"
-        "> **First run:** ~2-3 min (model download). **Subsequent runs:** ~15-30s."
+        "> Runs on **Hugging Face ZeroGPU** (free A10G) — no local GPU needed.\n"
+        "> **First run:** ~2-3 min (model download ~5 GB).  **Subsequent runs:** ~15-30s."
     )
 
     with gr.Row():
         with gr.Column():
-            person_input  = gr.Image(label="Person Photo", type="pil", height=380)
-            garment_input = gr.Image(label="Garment Image", type="pil", height=380)
+            person_input  = gr.Image(label="Person Photo", type="pil", height=350)
+            garment_input = gr.Image(label="Garment Image", type="pil", height=350)
             cloth_type = gr.Radio(
                 ["upper", "lower", "overall"],
                 value="upper",
@@ -125,13 +123,21 @@ with gr.Blocks(title="Virtual Try-On", theme=gr.themes.Soft()) as demo:
             try_btn = gr.Button("👗 Try On", variant="primary", size="lg")
 
         with gr.Column():
-            gr.Markdown("*Click an image in the gallery then use its ⬇ icon to download.*")
-            output_gallery = gr.Gallery(label="Result", columns=1, height=420, show_download_button=True)
+            status_box = gr.Textbox(
+                label="Status", value="Ready — upload images and click Try On",
+                interactive=False, max_lines=2,
+            )
+            output_gallery = gr.Gallery(label="Result", columns=1, height=420)
 
+    # Chain: first update status immediately, then run inference
     try_btn.click(
+        fn=lambda: "⏳ Requesting GPU + loading model…  (first run ~3 min, please wait)",
+        inputs=None,
+        outputs=[status_box],
+    ).then(
         fn=run_tryon,
         inputs=[person_input, garment_input, cloth_type, num_steps, guidance, seed_input],
-        outputs=[output_gallery],
+        outputs=[output_gallery, status_box],
     )
 
     gr.Markdown(