Update run.py

run.py

@@ -3,70 +3,86 @@ import cv2
 import numpy as np
 from insightface.app import FaceAnalysis
 
-# Initialisation du détecteur de visages
-face_app = FaceAnalysis(name="buffalo_l")
-face_app.prepare(ctx_id=0, det_size=(640, 640))
+# Initialisation sécurisée
+try:
+    face_app = FaceAnalysis(name="buffalo_l")
+    face_app.prepare(ctx_id=0, det_size=(640, 640))
+except Exception as e:
+    print(f"ERREUR initialisation : {str(e)}")
+    face_app = None
 
-def face_swap(source_img, target_img):
+def safe_face_swap(source_img, target_img):
     try:
-        # Conversion des images Gradio en numpy arrays
-        source = np.array(source_img)
-        target = np.array(target_img)
+        if face_app is None:
+            raise ValueError("Le modèle de détection n'est pas chargé")
         
-        # Détection des visages
-        source_faces = face_app.get(source)
-        target_faces = face_app.get(target)
+        # Conversion robuste
+        source = np.array(source_img, dtype=np.uint8)
+        target = np.array(target_img, dtype=np.uint8)
         
-        if not source_faces or not target_faces:
-            raise ValueError("Aucun visage détecté dans une des images")
+        # Vérification des dimensions
+        if source.ndim != 3 or target.ndim != 3:
+            raise ValueError("Format d'image invalide")
+            
+        # Détection sécurisée
+        source_faces = face_app.get(cv2.cvtColor(source, cv2.COLOR_RGB2BGR))
+        target_faces = face_app.get(cv2.cvtColor(target, cv2.COLOR_RGB2BGR))
         
-        # Récupération du premier visage détecté
-        src_face = source_faces[0]
-        tgt_face = target_faces[0]
+        if not source_faces:
+            raise ValueError("Aucun visage détecté dans l'image source")
+        if not target_faces:
+            raise ValueError("Aucun visage détecté dans l'image cible")
+            
+        # Swap simplifié mais stable
+        src_face = source_faces[0].bbox.astype(int)
+        tgt_face = target_faces[0].bbox.astype(int)
         
-        # Extraction des zones de visage (simplifié)
-        src_box = src_face.bbox.astype(int)
-        tgt_box = tgt_face.bbox.astype(int)
-        
-        # Échange des visages
         result = target.copy()
-        result[tgt_box[1]:tgt_box[3], tgt_box[0]:tgt_box[2]] = \
-            source[src_box[1]:src_box[3], src_box[0]:src_box[2]]
+        result[tgt_face[1]:tgt_face[3], tgt_face[0]:tgt_face[2]] = \
+            cv2.resize(source[src_face[1]:src_face[3], src_face[0]:src_face[2]], 
+                     (tgt_face[2]-tgt_face[0], tgt_face[3]-tgt_face[1]))
         
-        # Conversion BGR vers RGB pour Gradio
-        result = cv2.cvtColor(result, cv2.COLOR_BGR2RGB)
         return result
-        
     except Exception as e:
-        print(f"ERREUR: {str(e)}")
-        # Retourne l'image originale en cas d'erreur
-        return cv2.cvtColor(target, cv2.COLOR_BGR2RGB)
+        print(f"ERREUR traitement : {str(e)}")
+        # Retourne l'image cible avec message d'erreur
+        if target_img is not None:
+            return target_img
+        return None
 
-# Interface Gradio améliorée
-with gr.Blocks() as app:
-    gr.Markdown("## 🔄 FaceSwap Pro - Powered by InsightFace")
+# Interface renforcée
+with gr.Blocks(title="FaceSwap Pro") as app:
+    gr.Markdown("""
+    ## 🔄 FaceSwap Professionnel
+    *Échange de visages stable et sécurisé*
+    """)
     
     with gr.Row():
         with gr.Column():
-            src_img = gr.Image(label="Image Source", type="numpy")
-            tgt_img = gr.Image(label="Image Cible", type="numpy")
-        result_img = gr.Image(label="Résultat", interactive=False)
+            src = gr.Image(label="Source", type="numpy")
+            tgt = gr.Image(label="Cible", type="numpy")
+            btn = gr.Button("Exécuter", variant="primary")
+        result = gr.Image(label="Résultat", interactive=False)
     
-    btn = gr.Button("Échanger les visages", variant="primary")
+    # Gestion des erreurs dans l'UI
+    error_box = gr.Textbox(visible=False, label="Erreur")
     
-    # Exemples intégrés
-    gr.Examples(
-        examples=[
-            ["exemples/source1.jpg", "exemples/target1.jpg"],
-            ["exemples/source2.jpg", "exemples/target2.jpg"]
-        ],
-        inputs=[src_img, tgt_img],
-        outputs=result_img,
-        fn=face_swap,
-        cache_examples=True
-    )
+    def wrapped_face_swap(src_img, tgt_img):
+        try:
+            return safe_face_swap(src_img, tgt_img), ""
+        except Exception as e:
+            return None, f"Erreur : {str(e)}"
     
-    btn.click(face_swap, inputs=[src_img, tgt_img], outputs=result_img)
+    btn.click(
+        fn=wrapped_face_swap,
+        inputs=[src, tgt],
+        outputs=[result, error_box]
+    )
 
 if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", server_port=7860)
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        debug=True
+    )