Update app.py

app.py

@@ -7,7 +7,6 @@ from typing import List, Dict
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from datetime import datetime
-import time
 
 # ======================================================================
 # CONFIGURATION
@@ -42,10 +41,6 @@ MAX_CONVERSATION_HISTORY = 10
 API_HOST = '0.0.0.0'
 API_PORT = 1212  # Le port 1212 est conservé, il doit être configuré dans le README.md
 
-# Configuration retry pour erreur 503
-MAX_RETRIES = 10
-RETRY_DELAY = 1  # secondes entre chaque tentative
-
 # ======================================================================
 # VARIABLES GLOBALES
 # ======================================================================
@@ -275,34 +270,50 @@ def clear_history(session_id):
     conversation_histories[session_id] = []
 
 # ======================================================================
-# CALL GEMINI AVEC RETRY
+# CALL GEMINI
 # ======================================================================
-
 def call_gemini(rag_prompt, system_prompt, gemini_client):
-    """Appelle Google Gemini avec retry automatique en cas d'erreur 503."""
+    """
+    Appelle Google Gemini avec une logique de réessai en cas d'échec de l'API.
+    Maximum de 10 tentatives.
+    """
+    MAX_RETRIES = 10
+    
+    # L'API Gemini Python ne retourne pas directement les codes HTTP comme 503, 
+    # mais lève des exceptions `APIError` ou `ResourceExhaustedError` qui correspondent
+    # à des erreurs de service. Nous allons intercepter l'exception générique 
+    # pour capturer toutes les erreurs potentielles liées à l'appel.
+
     for attempt in range(MAX_RETRIES):
         try:
+            print(f"   📞 Tentative d'appel Gemini #{attempt + 1}...")
+            # L'API Python de Google lève des exceptions `APIError` pour les échecs, 
+            # y compris ceux qui correspondent aux 5xx.
             response = gemini_client.models.generate_content(
                 model=GEMINI_MODEL,
                 contents=f"{system_prompt}\n\n{rag_prompt}"
             )
+            # Si la réponse réussit, on sort de la boucle
             return response.text.replace("*", "")
+
         except Exception as e:
+            # Ici, on capture toute erreur d'API ou de connexion.
+            # On considère cela comme une erreur de service transitoire pour les réessais.
             error_message = str(e)
-            # Vérifie si c'est une erreur 503
-            if "503" in error_message or "Service Unavailable" in error_message:
-                print(f"⚠️ Erreur 503 détectée (tentative {attempt + 1}/{MAX_RETRIES}). Nouvelle tentative dans {RETRY_DELAY}s...")
-                if attempt < MAX_RETRIES - 1:  # Pas de sleep à la dernière tentative
-                    time.sleep(RETRY_DELAY)
-                continue
+            print(f"   ❌ Erreur Gemini (Tentative {attempt + 1}/{MAX_RETRIES}): {error_message}")
+
+            if attempt < MAX_RETRIES - 1:
+                # Si ce n'est pas la dernière tentative, on attend avant de réessayer
+                sleep_time = 2  # Attente de 2 secondes
+                print(f"   😴 Attente de {sleep_time} secondes avant de réessayer...")
+                time.sleep(sleep_time)
             else:
-                # Si ce n'est pas une erreur 503, on arrête et on retourne l'erreur
-                print(f"❌ Erreur Gemini (non-503): {e}")
-                return f"Erreur: {str(e)}"
-    
-    # Si on arrive ici, c'est qu'on a épuisé toutes les tentatives
-    print(f"❌ Échec après {MAX_RETRIES} tentatives")
-    return "Erreur: Service temporairement indisponible après plusieurs tentatives"
+                # Dernière tentative échouée
+                print("   🛑 Toutes les tentatives de réessai ont échoué.")
+                return f"Erreur fatale après {MAX_RETRIES} tentatives: {error_message}"
+
+    # Ne devrait jamais être atteint, mais par sécurité
+    return "Erreur inconnue dans la boucle de réessai de Gemini."
 
 # ======================================================================
 # ANSWER PROCESS
@@ -388,7 +399,7 @@ def api_get_answer():
         # Génère prompt RAG
         rag_prompt = get_answer(query_text, collection, model_paraphrase, model_cross_encoder, history)
 
-        # Appelle Gemini avec retry automatique
+        # Appelle Gemini
         response = call_gemini(rag_prompt, system_prompt, gemini_client)
 
         # Sauvegarde réponse