from gradio_client import Client, file
import gradio as gr

client = Client("felixrosberg/face-swap")

def run_face_swap(target_img, source_img, slider_val, adv_slider_val, mode_list):
    result = client.predict(
        target=file(target_img),
        source=file(source_img),
        slider=slider_val,
        adv_slider=adv_slider_val,
        settings=mode_list,
        api_name="/run_inference"
    )
    return result

with gr.Blocks() as iface:
    gr.Markdown("## Ndërrimi i fytyrës me AI")

    with gr.Row():
        target = gr.Image(label="Fotoja objektiv (target)", type="filepath")
        source = gr.Image(label="Fotoja burim (source)", type="filepath")

    slider = gr.Slider(0, 100, value=100, label="Raporti i Anonimizimit (%)")
    adv_slider = gr.Slider(0, 100, value=100, label="Raporti i Mbrojtjes Kundër Sulmeve (%)")

    mode = gr.CheckboxGroup(
        choices=["Compare", "Anonymize", "Reconstruction Attack", "Adversarial Defense"],
        label="Modaliteti",
        value=[]
    )

    btn = gr.Button("🧙‍♂️ Ndërroni fytyrën")
    output_img = gr.Image(label="Rezultati")

    btn.click(
        fn=run_face_swap,
        inputs=[target, source, slider, adv_slider, mode],
        outputs=[output_img]
    )

iface.launch()