Update run.py

run.py

@@ -10,45 +10,30 @@ import shutil
 os.environ['INSIGHTFACE_ROOT'] = '/tmp/.insightface'
 
 def swap_face(source_face, target_face, frame):
-    """Remplace le visage cible par le visage source"""
     src_emb = source_face.normed_embedding
     tgt_bbox = target_face.bbox.astype(int)
-
-    # Obtenir le visage redimensionné
-    src_face_img = source_face.img
-    resized_face = cv2.resize(src_face_img, (tgt_bbox[2]-tgt_bbox[0], tgt_bbox[3]-tgt_bbox[1]))
-
-    # Créer un masque doux
+    resized_face = cv2.resize(source_face.img, (tgt_bbox[2]-tgt_bbox[0], tgt_bbox[3]-tgt_bbox[1]))
     mask = np.zeros_like(resized_face)
     center = (mask.shape[1]//2, mask.shape[0]//2)
     radius = int(min(mask.shape) * 0.45)
     cv2.circle(mask, center, radius, (255,255,255), -1)
     mask = cv2.GaussianBlur(mask, (15,15), 5)
-
-    # Positionner le visage
     center = ((tgt_bbox[0]+tgt_bbox[2])//2, (tgt_bbox[1]+tgt_bbox[3])//2)
-
-    # Échange réaliste
     result = cv2.seamlessClone(resized_face, frame, mask, center, cv2.NORMAL_CLONE)
     return result
 
-
 def process_video(source_img, target_video):
     try:
-        # Initialiser le modèle INSIDE la fonction
         face_app = FaceAnalysis(name="buffalo_l", root="/tmp/.insightface")
         face_app.prepare(ctx_id=0, det_size=(640, 640))
 
-        # Charger le visage source
         source_faces = face_app.get(source_img)
         if not source_faces:
             raise ValueError("Aucun visage trouvé dans l'image source.")
         source_face = source_faces[0]
 
-        # Fichier temporaire pour sauvegarder la sortie
         temp_output = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
 
-        # Ouvrir la vidéo cible
         cap = cv2.VideoCapture(target_video)
         fps = cap.get(cv2.CAP_PROP_FPS)
         frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
@@ -62,7 +47,6 @@ def process_video(source_img, target_video):
             if not ret:
                 break
 
-            # Détecter les visages dans la frame
             target_faces = face_app.get(frame)
             for face in target_faces:
                 frame = swap_face(source_face, face, frame)
@@ -72,7 +56,6 @@ def process_video(source_img, target_video):
         cap.release()
         out.release()
 
-        # Copier vers un nouveau fichier stable
         final_path = tempfile.mktemp(suffix=".mp4")
         shutil.copy(temp_output.name, final_path)
 
@@ -82,11 +65,7 @@ def process_video(source_img, target_video):
         print(f"Erreur lors du traitement : {str(e)}")
         return None
 
-
 # Interface Gradio
-title = "🎬 FaceSwap Pro - Swap Video"
-description = "Télécharge une image avec un visage et une vidéo. Le visage sera remplacé automatiquement."
-
 demo = gr.Interface(
     fn=process_video,
     inputs=[
@@ -94,8 +73,8 @@ demo = gr.Interface(
         gr.Video(label="Vidéo Cible"),
     ],
     outputs=gr.Video(label="Vidéo Résultat"),
-    title=title,
-    description=description,
+    title="🎬 FaceSwap Pro",
+    description="Échangez des visages dans une vidéo.",
     allow_flagging="never"
 )