app.py

import gradio as gr
import os
import sys

sys.path.append('roop')
from roop.core import run as roop_run

# CORE LOGIC - ISMEIN KOI BHI CHANGE NAHI HUA HAI. YEH PERFECT HAI.
def face_swap(source_img, target_media, progress=gr.Progress(track_tqdm=True)):
    if source_img is None or target_media is None:
        raise gr.Error("Dono file upload karein.")

    source_path = source_img
    target_path = target_media
    
    output_filename = os.path.basename(target_path)
    os.makedirs("output", exist_ok=True)
    output_path = os.path.join("output", f"result_{output_filename}")

    args = [
        "run.py",
        "--source", source_path,
        "--target", target_path,
        "--output", output_path,
        "--execution-provider", "cpu",
    ]
    sys.argv = args
    
    print(f"Running roop with args: {sys.argv}")

    try:
        roop_run()
    except SystemExit:
        pass
    except Exception as e:
        raise gr.Error(f"Face swap fail hua. Error: {str(e)}")

    if not os.path.exists(output_path):
        raise gr.Error("Processing fail hua. Model download nahi ho paaya ya koi aur error aayi. Logs check karein.")

    print(f"Process poora hua. Result yahan hai: {output_path}")
    return output_path

# --- Gradio UI (AAPKE IDEA KE SAATH) ---
with gr.Blocks(theme=gr.themes.Soft()) as app:
    gr.Markdown("# 🎭 Roop Face Swap AI")
    
    # Step 1: Source Face (Yeh dono tabs ke liye common rahega)
    gr.Markdown("### Step 1: Upload Source Face")
    source_face = gr.Image(label="Source Face (Jiska Chehra Lagana Hai)", type="filepath")

    # Step 2: Alag-alag Tabs banana (Image aur Video ke liye)
    with gr.Tabs():
        # TAB 1: IMAGE KE LIYE
        with gr.TabItem("🖼️ Swap on Image"):
            gr.Markdown("### Step 2: Upload Target Image")
            target_image = gr.Image(label="Target Image (Jispar Chehra Lagana Hai)", type="filepath")
            submit_btn_image = gr.Button("Swap Face on Image", variant="primary")
            gr.Markdown("### Step 3: Result")
            result_image = gr.Image(label="Result Image", interactive=False)

        # TAB 2: VIDEO KE LIYE
        with gr.TabItem("🎬 Swap on Video"):
            gr.Markdown("### Step 2: Upload Target Video")
            target_video = gr.Video(label="Target Video (Jispar Chehra Lagana Hai)")
            submit_btn_video = gr.Button("Swap Face on Video", variant="primary")
            gr.Markdown("### Step 3: Result")
            result_video = gr.Video(label="Result Video", interactive=False)

    # Dono buttons ke liye alag-alag click event
    # Yeh dono event ek hi 'face_swap' function ko call karte hain
    
    submit_btn_image.click(
        fn=face_swap,
        inputs=[source_face, target_image],
        outputs=result_image
    )
    
    submit_btn_video.click(
        fn=face_swap,
        inputs=[source_face, target_video],
        outputs=result_video
    )

app.launch()