Update run.py

run.py

@@ -1,34 +1,26 @@
 import gradio as gr
-from roop import core
+from roop_unleashed import Core  # Changement d'import
 
 def launch_interface():
-    # Initialisation EXPLICITE des composants Gradio
-    with gr.Blocks() as app:  # <-- Structure recommandée pour éviter les "NoneType"
-        gr.Markdown("## 🔄 Face Swap avec Roop")
-        
+    with gr.Blocks() as app:
+        gr.Markdown("## Face Swap")
         with gr.Row():
-            src_img = gr.Image(label="Image Source", type="filepath")
-            tgt_img = gr.Image(label="Image Cible", type="filepath")
-            result_img = gr.Image(label="Résultat", interactive=False)
-
-        btn = gr.Button("Exécuter le swap")
-
-        def swap_faces(source, target):
+            src = gr.Image(label="Source", type="filepath")
+            target = gr.Image(label="Cible", type="filepath")
+            output = gr.Image(label="Résultat")
+        
+        btn = gr.Button("Exécuter")
+        
+        def process(src_path, tgt_path):
             try:
-                # Appel à la logique roop (exemple simplifié)
                 output_path = "result.jpg"
-                core.process(source, target, output_path)  # Adaptez à la vraie API roop
+                Core().enhance_face(src_path, tgt_path, output_path)  # Adaptez à l'API réelle
                 return output_path
             except Exception as e:
-                print(f"ERREUR: {str(e)}")
+                print(f"Erreur : {e}")
                 return None
-
-        btn.click(
-            fn=swap_faces,
-            inputs=[src_img, tgt_img],
-            outputs=result_img
-        )
-
+        
+        btn.click(process, inputs=[src, target], outputs=output)
     return app
 
 if __name__ == "__main__":