Update app.py

app.py

@@ -5,6 +5,7 @@ import sys
 sys.path.append('roop')
 from roop.core import run as roop_run
 
+# CORE LOGIC - ISMEIN KOI CHANGE NAHI HAI
 def face_swap(source_img, target_media):
     if source_img is None or target_media is None:
         raise gr.Error("Dono file upload karein.")
@@ -16,8 +17,6 @@ def face_swap(source_img, target_media):
     os.makedirs("output", exist_ok=True)
     output_path = os.path.join("output", f"result_{output_filename}")
 
-    # ===== KOI EXTRA LOGIC NAHI, SIMPLE SWAP =====
-    # Face Enhancer wala argument hata diya hai.
     args = [
         "run.py",
         "--source", source_path,
@@ -25,7 +24,6 @@ def face_swap(source_img, target_media):
         "--output", output_path,
         "--execution-provider", "cpu",
     ]
-    # ============================================
     sys.argv = args
     
     print(f"Running roop with args: {sys.argv}")
@@ -33,29 +31,32 @@ def face_swap(source_img, target_media):
     try:
         roop_run()
     except SystemExit:
-        # Roop argparse error par SystemExit deta hai, use pass kar rahe hain.
         pass
     except Exception as e:
         raise gr.Error(f"Face swap fail hua. Error: {str(e)}")
 
-    # Check karein ki output file bani hai ya nahi
     if not os.path.exists(output_path):
-        # Agar file nahi bani to user ko batayein ki model download ho raha hai
-        # Pehli baar run hone mein 5-10 min lag sakte hain.
-        raise gr.Error("Processing ho raha hai... Pehli baar run hone mein 10-15 minute lag sakte hain kyunki model download ho raha hai. Please sabr rakhein aur thodi der baad dobara try karein.")
+        raise gr.Error("Processing ho raha hai... Pehli baar run hone mein 10-15 minute lag sakte hain. Please sabr rakhein.")
 
     print(f"Process poora hua. Result yahan hai: {output_path}")
     return output_path
 
+# --- Gradio UI (SIRF YAHAN CHANGE HUA HAI) ---
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("# 🎭 Simple Face Swap (Colab Port)")
     
     with gr.Row():
         source_image = gr.Image(label="Source Face", type="filepath")
-        target_media = gr.File(label="Target Image or Video", type="filepath")
+        
+        # ===== CHANGE 1: gr.File ko gr.Video se badla =====
+        target_media = gr.Video(label="Target Image or Video", type="filepath")
+        # ==================================================
     
     submit_btn = gr.Button("Swap Face", variant="primary")
-    result_output = gr.File(label="Result", interactive=False)
+    
+    # ===== CHANGE 2: Result ke liye bhi gr.Video use kiya taaki result ka bhi preview dikhe =====
+    result_output = gr.Video(label="Result", interactive=False)
+    # ======================================================================================
     
     submit_btn.click(
         fn=face_swap,