Update app.py

app.py

@@ -1,28 +1,42 @@
+from gradio_client import Client, file
 import gradio as gr
-import os
-from PIL import Image
-
-def face_swap(source_path, target_path):
-    # Dummy output to simulate face swap; replace this with your actual model logic
-    try:
-        src = Image.open(source_path).resize((256, 256))
-        tgt = Image.open(target_path).resize((256, 256))
-        result = Image.blend(src, tgt, alpha=0.5)
-        return result
-    except Exception as e:
-        return f"Gabim gjatë përpunimit: {e}"
-
-with gr.Blocks(title="Ndërrim Fytyrash") as app:
-    gr.Markdown("## 🤖 Ndërrues Fytyrash (Face Swap)")
-    gr.Markdown("Ngarko dy fotografi për të ndërruar fytyrat midis tyre.")
+
+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():
-        source = gr.Image(label="Fotoja burimore", type="filepath")
-        target = gr.Image(label="Fotoja objektiv", type="filepath")
+        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ërro fytyrat")
-    output = gr.Image(label="Rezultati")
+    btn = gr.Button("🧙‍♂️ Ndërroni fytyrën")
+    output_img = gr.Image(label="Rezultati")
 
-    btn.click(fn=face_swap, inputs=[source, target], outputs=output)
+    btn.click(
+        fn=run_face_swap,
+        inputs=[target, source, slider, adv_slider, mode],
+        outputs=[output_img]
+    )
 
-app.launch()
+iface.launch()