Add live status updates: GPU allocated / generating / done

app.py

@@ -76,10 +76,12 @@ def run_tryon(
     num_steps: int,
     guidance_scale: float,
     seed: int,
-) -> tuple:
+):
     if person_image is None or garment_image is None:
         raise gr.Error("Please upload both a person photo and a garment image.")
 
+    yield [], [], "⏳ GPU allocated — loading model (first run: ~3 min download)…"
+
     pipe = _get_pipe()
 
     person  = _fit_to_square(person_image)
@@ -89,6 +91,8 @@ def run_tryon(
     rng = torch.Generator(device="cuda")
     rng.manual_seed(int(seed) if seed != -1 else torch.randint(0, 2**32, (1,)).item())
 
+    yield [], [], f"🎨 Running {num_steps} diffusion steps…"
+
     result = pipe(
         image=person,
         mask_image=mask,
@@ -106,7 +110,7 @@ def run_tryon(
         img.save(path, format="PNG")
         saved_paths.append(path)
 
-    return output_images, saved_paths
+    yield output_images, saved_paths, "✅ Done! Download your result below."
 
 # ---------------------------------------------------------------------------
 # Gradio UI
@@ -137,6 +141,7 @@ 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():
+            status_text    = gr.Textbox(label="Status", value="Ready", interactive=False, max_lines=1)
             output_gallery = gr.Gallery(label="Result", columns=1, height=380)
             output_files   = gr.File(
                 label="⬇ Download to your device",
@@ -147,7 +152,7 @@ with gr.Blocks(title="Virtual Try-On", theme=gr.themes.Soft()) as demo:
     try_btn.click(
         fn=run_tryon,
         inputs=[person_input, garment_input, cloth_type, num_steps, guidance, seed_input],
-        outputs=[output_gallery, output_files],
+        outputs=[output_gallery, output_files, status_text],
     )
 
     gr.Markdown(