Update app.py

app.py

@@ -21,10 +21,8 @@ TELEGRAM_TOKEN = "8584350410:AAEuXqopGMfgdZ1BvLntA-e6FpoZl5uunEk"
 TELEGRAM_CHAT_ID = "1278265595"
 # Activer/Désactiver l'envoi de notifications
 TELEGRAM_NOTIFICATIONS_ENABLED = True
-# 🚨 FIX POUR NAME RESOLUTION ERROR: Utilisation d'une IP connue de Telegram
-# La vraie solution est de fixer le DNS dans l'environnement du conteneur.
-TELEGRAM_API_HOST = "149.154.167.220"
-TELEGRAM_API_DOMAIN = "api.telegram.org"
+# 🚨 RETOUR À L'URL STANDARD (Celle qui marche dans le notebook)
+TELEGRAM_API_URL_BASE = "https://api.telegram.org/bot"
 
 
 # ======================================================================
@@ -32,8 +30,6 @@ TELEGRAM_API_DOMAIN = "api.telegram.org"
 # ======================================================================
 
 DATA_FILE_PATH = "data/QR.csv"
-
-# CORRECTION CRITIQUE: Déplacement de la DB vers /tmp
 CHROMA_DB_PATH = "/tmp/bdd_ChromaDB"
 COLLECTION_NAME = "qr_data_dual_embeddings"
 
@@ -47,13 +43,11 @@ SRC_PARAPHRASE = "models/paraphrase-mpnet-base-v2"
 N_RESULTS_RETRIEVAL = 10
 N_RESULTS_RERANK = 3
 
-# Récupération de la clé depuis l'environnement
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "AIzaSyDXXY7uSXryTxZ51jQFsSLcPnC_Ivt9V1g")
 GEMINI_MODEL = "gemini-2.5-flash"
 
 MAX_CONVERSATION_HISTORY = 10
 
-# Configuration pour l'accès externe (host et port)
 API_HOST = '0.0.0.0'
 API_PORT = 1212
 
@@ -71,21 +65,17 @@ conversation_histories: Dict[str, List[Dict[str, str]]] = {}
 conversation_start_times: Dict[str, str] = {}
 
 # ======================================================================
-# 🤖 FONCTION D'ENVOI TÉLÉGRAM (MODIFIÉE)
+# 🤖 FONCTION D'ENVOI TÉLÉGRAM (SIMPLIFIÉE ET STANDARD)
 # ======================================================================
 
 def send_telegram_message(message: str, token: str, chat_id: str):
     """
-    Fonction générique pour envoyer un message à Telegram.
-    Utilise l'IP pour contourner la NameResolutionError.
+    Fonction générique pour envoyer un message à Telegram en utilisant l'URL standard.
     """
     if not TELEGRAM_NOTIFICATIONS_ENABLED:
         return
 
-    # Construction de l'URL pour la méthode 'sendMessage' en utilisant l'IP de l'hôte
-    # NOTE: Nous utilisons l'IP mais nous devons injecter le nom d'hôte dans le header
-    # pour que le serveur Telegram sache à quel domaine la requête est destinée.
-    url = f"https://{TELEGRAM_API_HOST}/bot{token}/sendMessage"
+    url = f"{TELEGRAM_API_URL_BASE}{token}/sendMessage"
     
     params = {
         "chat_id": chat_id,
@@ -93,23 +83,21 @@ def send_telegram_message(message: str, token: str, chat_id: str):
         "parse_mode": "Markdown"
     }
 
-    # IMPORTANT: Nous ajoutons le header Host pour que la requête soit bien traitée
-    # par l'hôte virtuel sur le serveur Telegram, malgré l'utilisation de l'IP.
-    headers = {
-        "Host": TELEGRAM_API_DOMAIN
-    }
-
     try:
-        response = requests.post(url, params=params, headers=headers, timeout=5)
+        # Tente l'envoi avec un timeout standard
+        response = requests.post(url, params=params, timeout=10)
         response.raise_for_status()
 
-        if response.json().get("ok"):
-            print(f"✅ Message Telegram envoyé (via IP: {TELEGRAM_API_HOST}).")
+        response_data = response.json()
+
+        if response_data.get("ok"):
+            print(f"✅ Message Telegram envoyé (URL standard).")
         else:
-            print(f"❌ Échec envoi Telegram: {response.json().get('description')}")
+            print(f"❌ Échec envoi Telegram: {response_data.get('description')}")
 
     except requests.exceptions.RequestException as e:
-        print(f"❌ Erreur connexion Telegram (Vérifiez le TOKEN/IP de secours) : {e}")
+        # L'erreur de connexion est gérée ici.
+        print(f"❌ Erreur connexion Telegram (Vérifiez le TOKEN/Réseau/Pare-feu HF Spaces): {e}")
     except Exception as e:
         print(f"❌ Erreur inattendue Telegram : {e}")
 
@@ -117,7 +105,6 @@ def send_llm_interaction_to_telegram(question: str, reponse_llm: str, session_id
     """
     Construit le message d'interaction Q/R et l'envoie via Telegram.
     """
-    # Construction du message formaté
     MESSAGE = f"""
 *🔔 Nouvelle Interaction LLM 🔔*
 *Session ID:* `{session_id}`
@@ -127,7 +114,7 @@ def send_llm_interaction_to_telegram(question: str, reponse_llm: str, session_id
 {question}
 
 *Réponse (LLM - Début):*
-{reponse_llm[:200]}... (Voir le log pour la réponse complète)
+{reponse_llm[:200]}...
 """
     send_telegram_message(MESSAGE, token, chat_id)
 
@@ -332,7 +319,6 @@ def add_to_history(session_id, role, content):
 
     conversation_histories[session_id].append({"role": role, "content": content})
 
-    # Limiter la taille de l'historique conservé en mémoire
     if len(conversation_histories[session_id]) > MAX_CONVERSATION_HISTORY * 2:
         conversation_histories[session_id] = conversation_histories[session_id][-(MAX_CONVERSATION_HISTORY * 2):]
 
@@ -428,21 +414,14 @@ def api_get_answer():
             generic_message = "Requête vide."
             return jsonify({"error": generic_message}), 400
 
-        # Récupère historique
         history = get_conversation_history(session_id)
-
-        # Génère prompt RAG
         rag_prompt = get_answer(query_text, collection, model_paraphrase, model_cross_encoder, history)
-
-        # Appelle Gemini
         response = call_gemini(rag_prompt, system_prompt, gemini_client)
 
-        # Sauvegarde réponse dans l'historique de la session
         add_to_history(session_id, "user", query_text)
         add_to_history(session_id, "assistant", response)
 
-        # 🚀 ÉTAPE CRUCIALE : ENVOI DE LA NOTIFICATION TÉLÉGRAM
-        # Ceci archive l'interaction (Question + Réponse)
+        # 🚀 ENVOI DE LA NOTIFICATION TÉLÉGRAM
         send_llm_interaction_to_telegram(
             question=query_text,
             reponse_llm=response,
@@ -450,7 +429,6 @@ def api_get_answer():
             token=TELEGRAM_TOKEN,
             chat_id=TELEGRAM_CHAT_ID
         )
-        # --------------------------------------------------------
 
         return jsonify({"generated_response": response})
 
@@ -480,10 +458,9 @@ if __name__ == '__main__':
     print("start app.py")
     if initialize_global_resources():
 
-        # ➡️ ENVOI DE LA NOTIFICATION TÉLÉGRAM AU DÉMARRAGE (même appel que la fonction générique)
+        # ➡️ ENVOI DE LA NOTIFICATION TÉLÉGRAM AU DÉMARRAGE
         startup_message = f"🚀 **Application RAG Démarrée**\nHeure: {time.strftime('%Y-%m-%d %H:%M:%S')}\nToutes les ressources sont chargées."
         send_telegram_message(startup_message, TELEGRAM_TOKEN, TELEGRAM_CHAT_ID)
-        # ------------------------------------------------------
 
         try:
             import socket