Update model_handler.py

model_handler.py

@@ -1,9 +1,8 @@
-# model_handler.py
+# model_handler.py (Version Corrigée et Améliorée)
 import sys
 from google import genai
 from google.genai.errors import APIError
 from google.genai.types import Part 
-# Importez os pour lire la variable d'environnement directement ici
 import os 
 from config import GEMINI_MODEL_NAME, DEFAULT_MAX_TOKENS, DEFAULT_TEMPERATURE, SYSTEM_PROMPT
 
@@ -12,6 +11,76 @@ from config import GEMINI_MODEL_NAME, DEFAULT_MAX_TOKENS, DEFAULT_TEMPERATURE, S
 client = None
 MODEL_LOADED_SUCCESSFULLY = False
 
+def run_api_self_test(client_instance):
+    """Effectue un test simple pour vérifier si l'API fonctionne."""
+    try:
+        print("INFO: Exécution du test de connexion à l'API Gemini...", file=sys.stderr)
+        
+        # Test simple de contenu
+        test_content = [
+            {"role": "user", "parts": [Part.from_text("Dis bonjour")]}
+        ]
+        
+        response = client_instance.models.generate_content(
+            model=GEMINI_MODEL_NAME, 
+            contents=test_content,
+            config=genai.types.GenerateContentConfig(max_output_tokens=10) # Limiter la réponse
+        )
+        
+        if response.text and response.text.strip():
+            print("INFO: ✅ Test de connexion API réussi. Le modèle répond.", file=sys.stderr)
+            print(f"INFO: Réponse du test (début): {response.text[:30]}...", file=sys.stderr)
+            return True
+        else:
+            print("ALERTE: ❌ Test de connexion API échoué. Réponse vide.", file=sys.stderr)
+            return False
+            
+    except APIError as e:
+        print(f"ALERTE: ❌ Test de connexion API échoué (APIError). Vérifiez la validité de la clé: {e.message}", file=sys.stderr)
+        return False
+    except Exception as e:
+        print(f"ALERTE: ❌ Test de connexion API échoué (Erreur Générale): {e}", file=sys.stderr)
+        return False
+
+def get_model_instance():
+    """Initialise ou retourne l'instance du client Gemini."""
+    global client, MODEL_LOADED_SUCCESSFULLY
+    
+    if client is None:
+        
+        print("INFO: Tentative d'initialisation du client Gemini...", file=sys.stderr)
+        api_key = os.environ.get("GEMINI_API_KEY")
+        
+        if not api_key:
+            print("ALERTE: ❌ Variable d'environnement 'GEMINI_API_KEY' non trouvée/vide.", file=sys.stderr)
+            MODEL_LOADED_SUCCESSFULLY = False
+            # Retourne pour s'assurer que 'client' reste None
+            return 
+            
+        print("INFO: Clé API 'GEMINI_API_KEY' trouvée dans les variables d'environnement.", file=sys.stderr)
+        
+        try:
+            # 1. Initialisation du client
+            client = genai.Client(api_key=api_key) 
+            print("INFO: Client Gemini initialisé. Tentative de self-test...", file=sys.stderr)
+            
+            # 2. Exécution du Self-Test
+            if run_api_self_test(client):
+                 MODEL_LOADED_SUCCESSFULLY = True
+                 print(f"INFO: Configuration API réussie. Modèle LLM prêt. Utilisant {GEMINI_MODEL_NAME}.", file=sys.stderr)
+            else:
+                 # Le self-test a échoué (clé invalide ou autre problème API)
+                 client = None # Forcer la réinitialisation si nécessaire
+                 MODEL_LOADED_SUCCESSFULLY = False
+                 print("ALERTE: ❌ Client Gemini non initialisé. Le self-test a échoué.", file=sys.stderr)
+                 
+        except Exception as e:
+            # Ceci attrape les erreurs qui ne sont pas des APIError (ex: problème de librairie, etc.)
+            print(f"ALERTE: ❌ Échec critique de l'initialisation du client Gemini: {e}", file=sys.stderr)
+            client = None
+            MODEL_LOADED_SUCCESSFULLY = False
+
+
 def get_model_instance():
     """Initialise ou retourne l'instance du client Gemini."""
     global client, MODEL_LOADED_SUCCESSFULLY