Update app.py

app.py

@@ -100,6 +100,10 @@ def load_system_prompt():
 def initialize_gemini_client():
     """Initialise le client Google Gemini."""
     try:
+        # NOTE: Using a placeholder key in the code. A real API key 
+        # should be loaded from an environment variable or a secret.
+        if GEMINI_API_KEY == "AIzaSyDXXY7uSXryTxZ51jQFsSLcPnC_Ivt9V1g":
+             print("⚠️ Clé Gemini par défaut. Assurez-vous d'utiliser une clé valide ou un secret.")
         return genai.Client(api_key=GEMINI_API_KEY)
     except Exception as e:
         print(f"❌ Erreur Gemini: {e}")
@@ -116,7 +120,7 @@ def setup_chromadb_collection(client, df, model_paraphrase):
     try:
         collection = client.get_or_create_collection(name=COLLECTION_NAME)
     except Exception as e:
-        print(f"❌ Erreur ChromaDB: {e}")
+        print(f"❌ Erreur get/create collection ChromaDB: {e}")
         raise
     
     if collection.count() == total_docs and total_docs > 0:
@@ -144,15 +148,26 @@ def setup_chromadb_collection(client, df, model_paraphrase):
         metadatas.append({**meta, "type": "reponse"})
         ids.append(f"id_{i}_R")
     
-    embeddings = model_paraphrase.encode(docs, show_progress_bar=False).tolist()
+    # --- Potentielle source d'erreur: Encodage ---
+    try:
+        embeddings = model_paraphrase.encode(docs, show_progress_bar=False).tolist()
+    except Exception as e:
+        print(f"❌ Erreur d'encodage des documents pour ChromaDB: {e}")
+        raise
     
     try:
+        # Tentative de suppression et recréation pour forcer la mise à jour
         client.delete_collection(name=COLLECTION_NAME)
-    except:
+    except Exception:
+        # Ignorer si la collection n'existe pas ou si la suppression échoue
         pass
     
-    collection = client.get_or_create_collection(name=COLLECTION_NAME)
-    collection.add(embeddings=embeddings, documents=docs, metadatas=metadatas, ids=ids)
+    try:
+        collection = client.get_or_create_collection(name=COLLECTION_NAME)
+        collection.add(embeddings=embeddings, documents=docs, metadatas=metadatas, ids=ids)
+    except Exception as e:
+        print(f"❌ Erreur d'ajout des données à la collection ChromaDB: {e}")
+        raise
     
     print(f"✅ Collection remplie: {collection.count()} documents.")
     return collection
@@ -296,19 +311,29 @@ def initialize_global_resources():
     os.makedirs(CHROMA_DB_PATH, exist_ok=True)
     
     try:
+        # 1. Chargement des modèles, données et prompt
         model_cross_encoder, model_paraphrase = load_models()
         df = load_data()
         system_prompt = load_system_prompt()
         gemini_client = initialize_gemini_client()
     except Exception:
+        # Si une erreur se produit ici (modèles/données), on ne continue pas.
         return False
     
     try:
-        chroma_client = chromadb.PersistentClient(path=CHROMA_DB_PATH)
+        # 2. Initialisation et configuration de ChromaDB
+        print(f"⏳ Initialisation de ChromaDB à l'emplacement: {CHROMA_DB_PATH}")
+        # Note : Le chemin doit être accessible en écriture
+        chroma_client = chromadb.PersistentClient(path=CHROMA_DB_PATH) 
         collection = setup_chromadb_collection(chroma_client, df, model_paraphrase)
+        
         print("✅ INITIALISATION COMPLÈTE\n")
         return True
-    except Exception:
+    except Exception as e:
+        # 3. Gérer spécifiquement les erreurs ChromaDB
+        print(f"❌ Erreur lors de l'initialisation de ChromaDB ou du remplissage: {e}")
+        # Pour les déploiements sur Hugging Face Spaces, vérifiez que 'data/bdd_ChromaDB' 
+        # est accessible en écriture, ou essayez de le placer dans '/tmp/bdd_ChromaDB'.
         return False
 
 # ======================================================================