Update analyze_bob_hf.py

analyze_bob_hf.py

@@ -23,7 +23,7 @@ except Exception:
 BASE_DIR = Path(os.environ.get("BOB_BASE_DIR", Path(__file__).parent.parent))
 TRANSCRIPTIONS_DIR = Path(os.environ.get("BOB_TRANSCRIPTIONS_DIR", BASE_DIR / "output" / "transcriptions"))
 OUTPUT_FILE = Path(os.environ.get("BOB_OUTPUT_FILE", BASE_DIR / "output" / "resume_bob.txt"))
-HF_MODEL = os.environ.get("HF_MODEL", "meta-llama/Llama-3.2-1B-Instruct")  # par défaut
+HF_MODEL = os.environ.get("HF_MODEL", "facebook/opt-350m")  # Modèle léger par défaut
 
 def load_hf_model():
     """Charge un modèle Hugging Face"""
@@ -39,10 +39,11 @@ def load_hf_model():
             token=os.environ.get("HF_TOKEN")  # Pour les modèles privés
         )
         
-        print(f"Modèle {HF_MODEL} chargé avec succès")
+        print(f"✅ Modèle {HF_MODEL} chargé avec succès")
         return generator
     except Exception as e:
-        print(f"Erreur lors du chargement du modèle Hugging Face: {e}")
+        print(f"❌ Erreur lors du chargement du modèle Hugging Face: {e}")
+        print(f"Traceback: {traceback.format_exc()}")
         print("Assurez-vous que le modèle est disponible et que vous avez les permissions nécessaires")
         return None
 
@@ -462,7 +463,6 @@ def analyze_files_hf(transcriptions_dir: Path, input_dir: Path, output_file: Pat
         log("Assurez-vous d'avoir exécuté le script de transcription d'abord")
         return {"success": False, "count": 0}
 
-
     log(f"Trouvé {len(transcription_files)} fichier(s) de transcription:")
     for i, file in enumerate(transcription_files, 1):
         log(f"  {i}. {file.name}")
@@ -471,7 +471,20 @@ def analyze_files_hf(transcriptions_dir: Path, input_dir: Path, output_file: Pat
     # Initialisation du modèle Hugging Face
     generator = load_hf_model()
     if not generator:
-        return {"success": False, "error": "Modèle Hugging Face indisponible"}
+        log("❌ Modèle Hugging Face indisponible")
+        # Créer un fichier de résultat factice pour éviter l'erreur
+        try:
+            output_file.parent.mkdir(parents=True, exist_ok=True)
+            with open(output_file, 'w', encoding='utf-8') as f:
+                f.write("# ERREUR - Modèle indisponible\n")
+                f.write("Erreur|P|MODÈLE INDISPONIBLE|000\n")
+            log(f"📄 Fichier d'erreur créé: {output_file}")
+            return {"success": False, "error": "Modèle Hugging Face indisponible", "count": 0}
+        except Exception as write_error:
+            log(f"❌ Erreur lors de la création du fichier d'erreur: {write_error}")
+            return {"success": False, "error": "Modèle Hugging Face indisponible"}
+    
+    log("✅ Modèle chargé, début de l'analyse...")
 
     results = []
     success_count = 0
@@ -560,5 +573,15 @@ def analyze_files_hf(transcriptions_dir: Path, input_dir: Path, output_file: Pat
             log(f"Traceback: {traceback.format_exc()}")
             return {"success": False, "error": f"Erreur d'écriture: {write_error}"}
     else:
-        log("Aucune analyse réussie, pas de fichier de résumé généré")
-        return {"success": False, "count": total, "ok": 0}
+        log("⚠️ Aucune analyse réussie, création d'un fichier factice...")
+        # Créer un fichier factice pour éviter l'erreur
+        try:
+            output_file.parent.mkdir(parents=True, exist_ok=True)
+            with open(output_file, 'w', encoding='utf-8') as f:
+                f.write(f"# RÉSUMÉ DES BOB - {datetime.now().strftime('%d/%m/%Y %H:%M:%S')}\n")
+                f.write("# Aucune analyse réussie\n")
+            log(f"📄 Fichier factice créé: {output_file}")
+            return {"success": False, "count": total, "ok": 0}
+        except Exception as write_error:
+            log(f"❌ Erreur lors de la création du fichier factice: {write_error}")
+            return {"success": False, "error": f"Erreur d'écriture: {write_error}"}