Update run.py

run.py

@@ -6,33 +6,20 @@ from insightface.app import FaceAnalysis
 import tempfile
 from moviepy.editor import VideoFileClip
 
-# ===== Configuration précoce pour éviter les erreurs =====
-
-# Désactiver les warnings d'Albumentations (optionnel)
+# Désactiver les avertissements d'Albumentations
 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">
-<fontconfig>
-</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
 
-# ===== Fonction de swap de visage =====
-
 def swap_face(source_face, target_face, frame):
     src_emb = source_face.normed_embedding
     tgt_bbox = target_face.bbox.astype(int)
@@ -48,7 +35,6 @@ def swap_face(source_face, target_face, frame):
     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")
@@ -96,9 +82,7 @@ def process_video(source_img, target_video):
         print(f"Erreur lors du traitement : {str(e)}")
         return None
 
-
-# ===== Interface Gradio =====
-
+# Interface Gradio
 demo = gr.Interface(
     fn=process_video,
     inputs=[
@@ -111,6 +95,5 @@ demo = gr.Interface(
     allow_flagging="never"
 )
 
-
 if __name__ == "__main__":
     demo.launch(share=True)