import gradio as gr
import os
from run import run_faceswap

def process_face_swap(source_image, target_media):
    source_path = "face.png"
    target_path = "target"

    source_image.save(source_path)
    ext = os.path.splitext(target_media.name)[-1]
    target_path += ext
    target_media.save(target_path)

    output_path = run_faceswap(source_path, target_path)
    return output_path

iface = gr.Interface(
    fn=process_face_swap,
    inputs=[
        gr.Image(label="Source Face (face.png)"),
        gr.File(label="Target Image or Video (.jpg/.mp4)")
    ],
    outputs=gr.File(label="Swapped Output"),
    title="Lightweight Roop Face Swap",
    description="Upload face.png and target image/video. Output will be result.jpg or result.mp4"
)

iface.launch()