Update app.py

app.py

@@ -5,11 +5,14 @@ import sys
 sys.path.append('roop')
 from roop.core import run as roop_run
 
-# CORE LOGIC - ISMEIN KOI CHANGE NAHI HAI
+# CORE LOGIC - Ismein hum 'yield' use karenge status update ke liye
 def face_swap(source_img, target_media):
     if source_img is None or target_media is None:
         raise gr.Error("Dono file upload karein.")
 
+    # 1. PEHLA STATUS UPDATE: Jaise hi button dabe, user ko batao
+    yield "Processing shuru ho gaya hai... Please wait.", None
+
     source_path = source_img
     target_path = target_media
     
@@ -29,6 +32,7 @@ def face_swap(source_img, target_media):
     print(f"Running roop with args: {sys.argv}")
 
     try:
+        # 2. ASLI KAAM YAHAN HOGA (Ismein time lagega)
         roop_run()
     except SystemExit:
         pass
@@ -36,32 +40,39 @@ def face_swap(source_img, target_media):
         raise gr.Error(f"Face swap fail hua. Error: {str(e)}")
 
     if not os.path.exists(output_path):
-        raise gr.Error("Processing ho raha hai... Pehli baar run hone mein 10-15 minute lag sakte hain. Please sabr rakhein.")
+        # Pehli baar model download hone par yeh message dikhega
+        yield "Model download ho raha hai (sirf pehli baar). Ismein 10-15 min lag sakte hain. Please sabr rakhein.", None
+        return # Yahan function rok dein
 
     print(f"Process poora hua. Result yahan hai: {output_path}")
-    return output_path
 
-# --- Gradio UI (SIRF YAHAN CHANGE HUA HAI) ---
+    # 3. AAKHRI STATUS UPDATE: Jab kaam poora ho jaaye
+    yield "✅ Process poora hua! Result taiyaar hai.", output_path
+
+# --- Gradio UI (Dono changes ke saath) ---
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🎭 Simple Face Swap (Colab Port)")
+    gr.Markdown("# 🎭 Roop Face Swap AI (with Live Status)")
     
     with gr.Row():
         source_image = gr.Image(label="Source Face", type="filepath")
         
-        # ===== CHANGE 1: gr.File ko gr.Video se badla =====
-        target_media = gr.Video(label="Target Image or Video", type="filepath")
-        # ==================================================
+        # ERROR THEEK KIYA: gr.Video se 'type="filepath"' hata diya
+        target_media = gr.Video(label="Target Image or Video")
+    
+    # NAYA FEATURE: Status dikhane ke liye ek Textbox
+    status_box = gr.Textbox(label="Status", value="Ready", interactive=False)
     
     submit_btn = gr.Button("Swap Face", variant="primary")
     
-    # ===== CHANGE 2: Result ke liye bhi gr.Video use kiya taaki result ka bhi preview dikhe =====
+    # Result ke liye bhi gr.Video use kiya
     result_output = gr.Video(label="Result", interactive=False)
-    # ======================================================================================
     
+    # Button click par ab 'face_swap' function chalega
+    # Aur woh STATUS BOX aur RESULT BOX dono ko update karega
     submit_btn.click(
         fn=face_swap,
         inputs=[source_image, target_media],
-        outputs=result_output
+        outputs=[status_box, result_output] # Output ab do jagah jayega
     )
 
 app.launch()