Update app.py

app.py

@@ -1,32 +1,75 @@
-import cv2
-import os
 import gradio as gr
-from run import run_faceswap
+import os
+import sys
+
+# YEH SABSE ZAROORI LINE HAI
+# Hum 'roop' folder ko system path mein add kar rahe hain taaki Python use dhoondh sake.
+sys.path.append('roop')
+
+# AB HUM SAHI JAGAH SE IMPORT KAR RAHE HAIN
+# 'roop' folder ke andar 'core.py' file se 'run' function ko import kar rahe hain.
+# Yahi aapke Colab ke video wale code ka logic hai.
+from roop.core import run as roop_run
 
-def process_face_swap(source_image, target_file):
-    source_path = "face.png"
-    target_path = "target" + os.path.splitext(target_file.name)[-1]
+# --- 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.")
 
-    # Save uploaded numpy image using OpenCV
-    cv2.imwrite(source_path, cv2.cvtColor(source_image, cv2.COLOR_RGB2BGR))
+    # Gradio se mili files ke temporary paths
+    source_path = source_img.name
+    target_path = target_media.name
+    
+    # Output file ka naam tay karna
+    output_filename = os.path.basename(target_path)
+    # Output folder banayein agar nahi hai
+    os.makedirs("output", exist_ok=True)
+    output_path = os.path.join("output", f"result_{output_filename}")
 
-    # Save uploaded target file
-    with open(target_path, "wb") as f:
-        f.write(target_file.read())
+    # Roop ko chalane ke liye arguments set karna
+    args = [
+        "run.py", # yeh bas ek placeholder hai, asli kaam roop_run() function karega
+        "--source", source_path,
+        "--target", target_path,
+        "--output", output_path,
+        "--execution-provider", "cpu", # Free tier ke liye 'cpu'
+        "-e" # Face enhancer on
+    ]
+    sys.argv = args
+    
+    print(f"Running roop with args: {sys.argv}")
 
-    # Call face swap
-    output_path = run_faceswap(source_path, target_path)
+    # Asli face swap process yahan run hoga
+    try:
+        roop_run()
+    except Exception as e:
+        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
 
-iface = gr.Interface(
-    fn=process_face_swap,
-    inputs=[
-        gr.Image(label="Source Face (face.png)", type="numpy"),
-        gr.File(label="Target Image or Video (.jpg/.mp4)")
-    ],
-    outputs=gr.File(label="Swapped Output"),
-    title="Lightweight Face Swap (User Logic)",
-    description="Upload face.png and target image/video. Output will be result.jpg or result.mp4"
-)
-
-iface.launch()
+# --- 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'.")
+    
+    with gr.Row():
+        source_image = gr.Image(label="Source Face (Single Face)", type="filepath")
+        target_media = gr.File(label="Target Image or Video", type="filepath")
+    
+    submit_btn = gr.Button("Swap Face", variant="primary")
+    result_output = gr.File(label="Result", interactive=False)
+    
+    submit_btn.click(
+        fn=face_swap,
+        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 ko launch karein
+app.launch()