Update run.py

run.py

@@ -6,35 +6,36 @@ from insightface.app import FaceAnalysis
 import tempfile
 from moviepy.editor import VideoFileClip
 
-# Désactiver les avertissements d'Albumentations
-os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1"
+# ===== Configuration précoce pour éviter les erreurs =====
 
-# Rediriger les chemins problématiques
+os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1"
 os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib'
 os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
 
 os.makedirs('/tmp/matplotlib', exist_ok=True)
 os.makedirs('/tmp/fontconfig', exist_ok=True)
 
-# Forcer InsightFace à utiliser un répertoire accessible
 os.environ['INSIGHTFACE_ROOT'] = '/tmp/.insightface'
-os.environ["ORT_DISABLE_CUDA"] = "1"  # Désactiver CUDA si GPU indisponible
+os.environ["ORT_DISABLE_CUDA"] = "1"
+
+# ===== Fonction de swap =====
 
 def swap_face(source_face, target_face, frame):
     src_emb = source_face.normed_embedding
     tgt_bbox = target_face.bbox.astype(int)
-    resized_face = cv2.resize(source_face.img, (tgt_bbox[2] - tgt_bbox[0], tgt_bbox[3] - tgt_bbox[1]))
-
+    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)
+    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)
-
-    center = ((tgt_bbox[0] + tgt_bbox[2]) // 2, (tgt_bbox[1] + tgt_bbox[3]) // 2)
+    cv2.circle(mask, center, radius, (255,255,255), -1)
+    mask = cv2.GaussianBlur(mask, (15,15), 5)
+    
+    center = ((tgt_bbox[0]+tgt_bbox[2])//2, (tgt_bbox[1]+tgt_bbox[3])//2)
     result = cv2.seamlessClone(resized_face, frame, mask, center, cv2.NORMAL_CLONE)
     return result
 
+
 def process_video(source_img, target_video):
     try:
         face_app = FaceAnalysis(name="buffalo_l", root="/tmp/.insightface")
@@ -52,7 +53,7 @@ def process_video(source_img, target_video):
         frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
         frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
 
-        fourcc = cv2.VideoWriter_fourcc(*'avc1')  # Codec compatible H.264
+        fourcc = cv2.VideoWriter_fourcc(*'avc1')
         out = cv2.VideoWriter(temp_output.name, fourcc, fps, (frame_width, frame_height))
 
         while True:
@@ -69,8 +70,6 @@ def process_video(source_img, target_video):
         cap.release()
         out.release()
 
-        print(f"Taille de la vidéo temporaire : {os.path.getsize(temp_output.name)} octets")
-
         # Réencodage final pour compatibilité Gradio
         clip = VideoFileClip(temp_output.name)
         final_path = tempfile.mktemp(suffix=".mp4")
@@ -79,21 +78,24 @@ def process_video(source_img, target_video):
         return final_path
 
     except Exception as e:
-        print(f"Erreur lors du traitement : {str(e)}")
+        print(f"Erreur : {str(e)}")
         return None
 
-# Interface Gradio
+
+# ===== Interface Gradio =====
+
 demo = gr.Interface(
     fn=process_video,
     inputs=[
-        gr.Image(label="Visage Source", type="numpy"),
-        gr.Video(label="Vidéo Cible"),
+        gr.Image(label="Visage Source", type="numpy"),  # ✅ Correct
+        gr.Video(label="Vidéo Cible"),                  # ✅ Correct
     ],
-    outputs=gr.Video(label="Vidéo Résultat"),
+    outputs=gr.Video(label="Vidéo Résultat"),           # ✅ Correct
     title="🎬 FaceSwap Pro",
     description="Échangez des visages dans une vidéo.",
     allow_flagging="never"
 )
 
+
 if __name__ == "__main__":
     demo.launch(share=True)