Update app.py

app.py

@@ -7,6 +7,7 @@ from typing import List, Dict
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from datetime import datetime
+import time
 
 # ======================================================================
 # CONFIGURATION
@@ -41,6 +42,10 @@ 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
 # ======================================================================
@@ -270,20 +275,34 @@ def clear_history(session_id):
     conversation_histories[session_id] = []
 
 # ======================================================================
-# CALL GEMINI
+# CALL GEMINI AVEC RETRY
 # ======================================================================
 
 def call_gemini(rag_prompt, system_prompt, gemini_client):
-    """Appelle Google Gemini."""
-    try:
-        response = gemini_client.models.generate_content(
-            model=GEMINI_MODEL,
-            contents=f"{system_prompt}\n\n{rag_prompt}"
-        )
-        return response.text.replace("*", "")
-    except Exception as e:
-        print(f"❌ Erreur Gemini: {e}")
-        return f"Erreur: {str(e)}"
+    """Appelle Google Gemini avec retry automatique en cas d'erreur 503."""
+    for attempt in range(MAX_RETRIES):
+        try:
+            response = gemini_client.models.generate_content(
+                model=GEMINI_MODEL,
+                contents=f"{system_prompt}\n\n{rag_prompt}"
+            )
+            return response.text.replace("*", "")
+        except Exception as e:
+            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
+            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"
 
 # ======================================================================
 # ANSWER PROCESS
@@ -369,7 +388,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
+        # Appelle Gemini avec retry automatique
         response = call_gemini(rag_prompt, system_prompt, gemini_client)
 
         # Sauvegarde réponse