Update app.py

app.py

@@ -2,58 +2,56 @@ import gradio as gr
 import os
 import sys
 
-# roop folder ko system path mein add karna
 sys.path.append('roop')
-
-# Sahi jagah se function import karna
 from roop.core import run as roop_run
 
-# --- Gradio Function ---
 def face_swap(source_img, target_media):
     if source_img is None or target_media is None:
-        raise gr.Error("Please upload both source image and target media.")
+        raise gr.Error("Dono file upload karein.")
 
     source_path = source_img
     target_path = target_media
     
-    # Output file ka naam tay karna
     output_filename = os.path.basename(target_path)
     os.makedirs("output", exist_ok=True)
     output_path = os.path.join("output", f"result_{output_filename}")
 
-    # ===== YAHAN AAKHRI CHANGE HAI =====
-    # OLD: "-e"
-    # NEW: Face enhancer ke liye poora command
+    # ===== KOI EXTRA LOGIC NAHI, SIMPLE SWAP =====
+    # Face Enhancer wala argument hata diya hai.
     args = [
         "run.py",
         "--source", source_path,
         "--target", target_path,
         "--output", output_path,
         "--execution-provider", "cpu",
-        # Face Enhancer ko chalu karne ka sahi tarika:
-        "--frame-processor", "face_swapper", "face_enhancer"
     ]
-    # ===================================
+    # ============================================
     sys.argv = args
     
     print(f"Running roop with args: {sys.argv}")
 
     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 failed. Error: {str(e)}")
+        raise gr.Error(f"Face swap fail hua. Error: {str(e)}")
 
-    print(f"Processing complete. Result at: {output_path}")
+    # 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.")
 
+    print(f"Process poora hua. Result yahan hai: {output_path}")
     return output_path
 
-# --- Gradio UI ---
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🎭 Roop Face Swap AI")
-    gr.Markdown("Upload a source face and a target image/video, then click 'Swap Face'.")
+    gr.Markdown("# 🎭 Simple Face Swap (Colab Port)")
     
     with gr.Row():
-        source_image = gr.Image(label="Source Face (Single Face)", type="filepath")
+        source_image = gr.Image(label="Source Face", type="filepath")
         target_media = gr.File(label="Target Image or Video", type="filepath")
     
     submit_btn = gr.Button("Swap Face", variant="primary")
@@ -64,8 +62,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         inputs=[source_image, target_media],
         outputs=result_output
     )
-    
-    gr.Markdown("---")
-    gr.Markdown("Created with help from AI. Make sure to use this tool responsibly.")
 
 app.launch()