Update run.py

run.py

@@ -6,14 +6,21 @@ from insightface.app import FaceAnalysis
 import tempfile
 from moviepy.editor import VideoFileClip
 
-# Rediriger Matplotlib et Fontconfig vers des répertoires temporaires
+# ===== Configuration précoce pour éviter les erreurs =====
+
+# Désactiver les warnings d'Albumentations (optionnel)
+os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1"
+
+# Rediriger les chemins problématiques
 os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib'
 os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
 os.environ['FONTCONFIG_FILE'] = '/tmp/fontconfig/fonts.conf'
 
+# Créer les répertoires temporaires nécessaires
 os.makedirs(os.environ['MPLCONFIGDIR'], exist_ok=True)
 os.makedirs(os.environ['FONTCONFIG_PATH'], exist_ok=True)
 
+# Créer un fichier minimal de configuration Fontconfig si nécessaire
 with open(os.environ['FONTCONFIG_FILE'], 'w') as f:
     f.write('''<?xml version="1.0"?>
 <!DOCTYPE fontconfig SYSTEM "fonts.dtd">
@@ -24,18 +31,20 @@ with open(os.environ['FONTCONFIG_FILE'], 'w') as f:
 os.environ['INSIGHTFACE_ROOT'] = '/tmp/.insightface'
 os.environ["ORT_DISABLE_CUDA"] = "1"  # Désactiver CUDA si GPU indisponible
 
+# ===== 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
 
@@ -57,7 +66,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
+        fourcc = cv2.VideoWriter_fourcc(*'avc1')  # Codec compatible H.264
         out = cv2.VideoWriter(temp_output.name, fourcc, fps, (frame_width, frame_height))
 
         while True:
@@ -88,7 +97,8 @@ def process_video(source_img, target_video):
         return None
 
 
-# Interface Gradio
+# ===== Interface Gradio =====
+
 demo = gr.Interface(
     fn=process_video,
     inputs=[
@@ -101,5 +111,6 @@ demo = gr.Interface(
     allow_flagging="never"
 )
 
+
 if __name__ == "__main__":
     demo.launch()