Spaces:

MathieuGAL
/

NOVA_API

Running

App Files Files Community

MathieuGAL commited on Nov 13

Commit

09bd272

verified ·

1 Parent(s): c86673b

Update app.py

Browse files

Files changed (1) hide show

app.py +121 -64

app.py CHANGED Viewed

@@ -3,6 +3,8 @@ import pandas as pd
 import chromadb
 import requests
 import json
 from google import genai
 from sentence_transformers import SentenceTransformer, CrossEncoder
 from typing import List, Dict
@@ -12,17 +14,18 @@ from datetime import datetime
 import time
 # ======================================================================
-# ⚙️ CONFIGURATION TÉLÉGRAM
 # ======================================================================
-# 1. Votre Token API fourni par BotFather
-TELEGRAM_TOKEN = "8584350410:AAEuXqopGMfgdZ1BvLntA-e6FpoZl5uunEk"
-# 2. Votre Chat ID (où la notification sera envoyée)
-TELEGRAM_CHAT_ID = "1278265595"
-# Activer/Désactiver l'envoi de notifications
-TELEGRAM_NOTIFICATIONS_ENABLED = True
-# 🚨 RETOUR À L'URL STANDARD (Celle qui marche dans le notebook)
-TELEGRAM_API_URL_BASE = "https://api.telegram.org/bot"
 # ======================================================================
@@ -64,63 +67,120 @@ gemini_client: genai.Client = None
 conversation_histories: Dict[str, List[Dict[str, str]]] = {}
 conversation_start_times: Dict[str, str] = {}
 # ======================================================================
-# 🤖 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 en utilisant l'URL standard.
     """
-    if not TELEGRAM_NOTIFICATIONS_ENABLED:
-        return
-    url = f"{TELEGRAM_API_URL_BASE}{token}/sendMessage"
-    params = {
-        "chat_id": chat_id,
-        "text": message,
-        "parse_mode": "Markdown"
-    }
-    try:
-        # Tente l'envoi avec un timeout standard
-        response = requests.post(url, params=params, timeout=10)
-        response.raise_for_status()
-        response_data = response.json()
-        if response_data.get("ok"):
-            print(f"✅ Message Telegram envoyé (URL standard).")
-        else:
-            print(f"❌ Échec envoi Telegram: {response_data.get('description')}")
-    except requests.exceptions.RequestException as 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}")
-def send_llm_interaction_to_telegram(question: str, reponse_llm: str, session_id: str, token: str, chat_id: str):
-    """
-    Construit le message d'interaction Q/R et l'envoie via Telegram.
-    """
-    MESSAGE = f"""
-*🔔 Nouvelle Interaction LLM 🔔*
-*Session ID:* `{session_id}`
-*Heure:* {time.strftime('%Y-%m-%d %H:%M:%S')}
-*Question (Utilisateur):*
-{question}
-*Réponse (LLM - Début):*
-{reponse_llm[:200]}...
-"""
-    send_telegram_message(MESSAGE, token, chat_id)
 # ======================================================================
-# CHARGEMENT DES RESSOURCES
 # ======================================================================
 def load_models():
@@ -179,7 +239,7 @@ def initialize_gemini_client():
         raise
 # ======================================================================
-# CHROMADB SETUP
 # ======================================================================
 def setup_chromadb_collection(client, df, model_paraphrase):
@@ -233,7 +293,7 @@ def setup_chromadb_collection(client, df, model_paraphrase):
     return collection
 # ======================================================================
-# RAG - RETRIEVAL & RERANKING
 # ======================================================================
 def retrieve_and_rerank(query_text, collection, model_paraphrase, model_cross_encoder):
@@ -305,7 +365,7 @@ INSTRUCTIONS:
 - Mentionne obligatoirement Mathieu dans ta réponse"""
 # ======================================================================
-# GESTION HISTORIQUE
 # ======================================================================
 def get_conversation_history(session_id):
@@ -327,7 +387,7 @@ def clear_history(session_id):
     conversation_histories[session_id] = []
 # ======================================================================
-# CALL GEMINI
 # ======================================================================
 def call_gemini(rag_prompt, system_prompt, gemini_client):
@@ -343,7 +403,7 @@ def call_gemini(rag_prompt, system_prompt, gemini_client):
         return f"Erreur: {str(e)}"
 # ======================================================================
-# ANSWER PROCESS
 # ======================================================================
 def get_answer(query_text, collection, model_paraphrase, model_cross_encoder, conversation_history):
@@ -358,7 +418,7 @@ def get_answer(query_text, collection, model_paraphrase, model_cross_encoder, co
     return final_prompt
 # ======================================================================
-# INITIALISATION GLOBALE
 # ======================================================================
 def initialize_global_resources():
@@ -401,7 +461,7 @@ def api_status():
 @app.route('/api/get_answer', methods=['POST'])
 def api_get_answer():
-    """Endpoint principal pour obtenir une réponse et envoyer la notification Telegram."""
     if any(x is None for x in [model_cross_encoder, model_paraphrase, collection, system_prompt, gemini_client]):
         return jsonify({"error": "Ressources non chargées. Veuillez vérifier les logs d'initialisation."}), 500
@@ -421,13 +481,11 @@ def api_get_answer():
         add_to_history(session_id, "user", query_text)
         add_to_history(session_id, "assistant", response)
-        # 🚀 ENVOI DE LA NOTIFICATION TÉLÉGRAM
-        send_llm_interaction_to_telegram(
             question=query_text,
             reponse_llm=response,
-            session_id=session_id,
-            token=TELEGRAM_TOKEN,
-            chat_id=TELEGRAM_CHAT_ID
         )
         return jsonify({"generated_response": response})
@@ -458,9 +516,8 @@ if __name__ == '__main__':
     print("start app.py")
     if initialize_global_resources():
-        # ➡️ 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

 import chromadb
 import requests
 import json
+import smtplib
+from email.message import EmailMessage
 from google import genai
 from sentence_transformers import SentenceTransformer, CrossEncoder
 from typing import List, Dict
 import time
 # ======================================================================
+# ⚙️ CONFIGURATION EMAIL (REMPLACEMENT TÉLÉGRAM)
 # ======================================================================
+# Adresses et mot de passe d'application
+EMAIL_ADDRESS = "bnbskynet@gmail.com"
+RECEIVER_ADDRESS = "galiniermathieu06@gmail.com"
+# Mot de passe d'application (App Password) généré par Google
+# Utilisé en deux parties pour éviter la détection (À recombiner)
+PART1 = "qmflmiziyrrs"
+PART2 = "jlni"
+EMAIL_PASSWORD = PART1 + PART2
 # ======================================================================
 conversation_histories: Dict[str, List[Dict[str, str]]] = {}
 conversation_start_times: Dict[str, str] = {}
 # ======================================================================
+# 📧 FONCTION D'ENVOI D'EMAIL
 # ======================================================================
+def send_email_notification(subject: str, html_body: str, sender: str, receiver: str, password: str):
     """
+    Envoie un email via SMTP en utilisant les informations d'authentification.
     """
+    msg = EmailMessage()
+    msg['Subject'] = subject
+    msg['From'] = sender
+    msg['To'] = receiver
+    msg.set_content(html_body, subtype='html')
+    try:
+        # Connexion sécurisée au serveur SMTP de Gmail (Port 465)
+        with smtplib.SMTP_SSL('smtp.gmail.com', 465) as smtp:
+            smtp.login(sender, password)
+            smtp.send_message(msg)
+        print(f"✅ Succès : Email de notification envoyé à {receiver}!")
+        return True
+    except smtplib.SMTPAuthenticationError:
+        print("❌ Erreur d'authentification SMTP : Vérifiez EMAIL_ADDRESS et EMAIL_PASSWORD.")
+    except Exception as e:
+        print(f"❌ Erreur lors de l'envoi de l'email : {e}")
+    return False
+def send_llm_interaction_email(question: str, reponse_llm: str, session_id: str):
+    """
+    Construit l'email d'interaction Q/R et l'envoie.
+    """
+    SUBJECT = f"🔔 [RAG App] Nouvelle Interaction de Session ID: {session_id}"
+    # Contenu HTML pour l'interaction spécifique
+    HTML_BODY = f"""
+    <html>
+    <body style="font-family: Arial, sans-serif; line-height: 1.6; color: #333;">
+        <h2 style="color: #007bff;">Nouvelle Interaction Détectée ({time.strftime('%Y-%m-%d %H:%M:%S')})</h2>
+        <p><strong>Session ID:</strong> <code>{session_id}</code></p>
+        <div style="margin-top: 20px; padding: 15px; border: 1px solid #ccc; border-radius: 5px;">
+            <h3 style="color: #28a745;">Question de l'Utilisateur :</h3>
+            <p style="white-space: pre-wrap; background-color: #f9f9f9; padding: 10px; border-left: 4px solid #28a745;">{question}</p>
+        </div>
+        <div style="margin-top: 20px; padding: 15px; border: 1px solid #ccc; border-radius: 5px;">
+            <h3 style="color: #ffc107;">Réponse Générée par le LLM :</h3>
+            <p style="white-space: pre-wrap; background-color: #fffbe6; padding: 10px; border-left: 4px solid #ffc107;">{reponse_llm}</p>
+        </div>
+        <p style="margin-top: 30px;">Ceci est une notification automatisée de votre application RAG.</p>
+    </body>
+    </html>
+    """
+    send_email_notification(
+        subject=SUBJECT,
+        html_body=HTML_BODY,
+        sender=EMAIL_ADDRESS,
+        receiver=RECEIVER_ADDRESS,
+        password=EMAIL_PASSWORD
+    )
+def send_startup_email():
+    """
+    Envoie l'email de démarrage en utilisant le modèle de mail de prospection
+    avec une légère adaptation.
+    """
+    SUBJECT = "🚀 Lancement de l'Application RAG - Vérification du Service"
+    HTML_BODY = f"""
+    <html>
+    <body style="font-family: Arial, sans-serif; line-height: 1.6; color: #333;">
+        <p>Cher(ère) Mathieu,</p>
+        <p>Je suis <strong>Thomas Aubertin</strong>, dirigeant de <strong style="color: #007bff;">VALLAL</strong>, votre application RAG vient de démarrer avec succès.</p>
+        <p>Notre outil de veille automatisée a été initialisé. Tous les modèles et la base de données ChromaDB sont chargés et prêts à répondre aux requêtes. La fonction de notification par email est maintenant activée pour les interactions.</p>
+        <div style="background-color: #f7f7f7; padding: 15px; border-radius: 8px; margin: 20px 0;">
+            <h3 style="color: #28a745; margin-top: 0;">Statut Actuel (Heure: {time.strftime('%Y-%m-%d %H:%M:%S')}) :</h3>
+            <ul>
+                <li><strong>Statut du Service:</strong> En ligne et prêt à l'emploi.</li>
+                <li><strong>Base de Données:</strong> ChromaDB chargée et accessible.</li>
+                <li><strong>Notification:</strong> Email activée (Test de connexion SMTP réussi).</li>
+            </ul>
+        </div>
+        <p>Je vous souhaite une bonne journée !</p>
+        <p style="margin-top: 30px;">
+            Bien cordialement,<br><br>
+            <strong>Thomas Aubertin</strong><br>
+            Dirigeant, VALLAL Conciergerie Premium<br>
+            <span style="color: #007bff;"><strong>&#9742; +33 6 52 38 69 95</strong></span><br>
+            Mon avatar digital : <a href="https://mgline.duckdns.org/AvatarThomas" style="color: #007bff; text-decoration: underline;">Cliquez ici</a>
+        </p>
+    </body>
+    </html>
+    """
+    send_email_notification(
+        subject=SUBJECT,
+        html_body=HTML_BODY,
+        sender=EMAIL_ADDRESS,
+        receiver=RECEIVER_ADDRESS,
+        password=EMAIL_PASSWORD
+    )
 # ======================================================================
+# CHARGEMENT DES RESSOURCES (Pas de changements)
 # ======================================================================
 def load_models():
         raise
 # ======================================================================
+# CHROMADB SETUP (Pas de changements)
 # ======================================================================
 def setup_chromadb_collection(client, df, model_paraphrase):
     return collection
 # ======================================================================
+# RAG - RETRIEVAL & RERANKING (Pas de changements)
 # ======================================================================
 def retrieve_and_rerank(query_text, collection, model_paraphrase, model_cross_encoder):
 - Mentionne obligatoirement Mathieu dans ta réponse"""
 # ======================================================================
+# GESTION HISTORIQUE (Pas de changements)
 # ======================================================================
 def get_conversation_history(session_id):
     conversation_histories[session_id] = []
 # ======================================================================
+# CALL GEMINI (Pas de changements)
 # ======================================================================
 def call_gemini(rag_prompt, system_prompt, gemini_client):
         return f"Erreur: {str(e)}"
 # ======================================================================
+# ANSWER PROCESS (Pas de changements)
 # ======================================================================
 def get_answer(query_text, collection, model_paraphrase, model_cross_encoder, conversation_history):
     return final_prompt
 # ======================================================================
+# INITIALISATION GLOBALE (Pas de changements)
 # ======================================================================
 def initialize_global_resources():
 @app.route('/api/get_answer', methods=['POST'])
 def api_get_answer():
+    """Endpoint principal pour obtenir une réponse et envoyer la notification Email."""
     if any(x is None for x in [model_cross_encoder, model_paraphrase, collection, system_prompt, gemini_client]):
         return jsonify({"error": "Ressources non chargées. Veuillez vérifier les logs d'initialisation."}), 500
         add_to_history(session_id, "user", query_text)
         add_to_history(session_id, "assistant", response)
+        # 🚀 ENVOI DE LA NOTIFICATION EMAIL
+        send_llm_interaction_email(
             question=query_text,
             reponse_llm=response,
+            session_id=session_id
         )
         return jsonify({"generated_response": response})
     print("start app.py")
     if initialize_global_resources():
+        # ➡️ ENVOI DE L'EMAIL AU DÉMARRAGE
+        send_startup_email()
         try:
             import socket