Update app.py

app.py

@@ -13,41 +13,38 @@ 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.")
 
-    # ===== YAHAN MERI GALTI THI, AB THEEK HAI =====
-    # OLD: source_path = source_img.name
-    # NEW: Gradio 'filepath' type mein seedha path (string) deta hai.
     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}")
 
-    # Roop ko chalane ke liye arguments set karna
+    # ===== YAHAN AAKHRI CHANGE HAI =====
+    # OLD: "-e"
+    # NEW: Face enhancer ke liye poora command
     args = [
         "run.py",
         "--source", source_path,
         "--target", target_path,
         "--output", output_path,
         "--execution-provider", "cpu",
-        "-e"
+        # 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}")
 
-    # Asli face swap process yahan run hoga
     try:
         roop_run()
     except Exception as e:
-        # Error aane par user ko message dikhana
         raise gr.Error(f"Face swap failed. Error: {str(e)}")
 
     print(f"Processing complete. Result at: {output_path}")
 
-    # Result ko UI mein dikhana
     return output_path
 
 # --- Gradio UI ---