Update app.py

app.py

@@ -11,6 +11,7 @@ import markdown  # Pour convertir la réponse en HTML
 from flask_session import Session  # <-- Importer Session
 import pprint  # Pour un affichage plus lisible des structures complexes (optionnel)
 import logging # Import logging
+import html  # Pour échapper le HTML dans les messages Telegram
 
 # --- Configuration Initiale ---
 load_dotenv()
@@ -22,6 +23,17 @@ logging.basicConfig(level=logging.DEBUG,  # Set the desired logging level
                     format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__) # Get a logger for this module
 
+# --- Configuration Telegram ---
+TELEGRAM_BOT_TOKEN = "8004545342:AAGcZaoDjYg8dmbbXRsR1N3TfSSbEiAGz88"
+TELEGRAM_CHAT_ID = "-1002497861230"
+TELEGRAM_ENABLED = TELEGRAM_BOT_TOKEN and TELEGRAM_CHAT_ID
+
+# Vérification de la configuration Telegram
+if TELEGRAM_ENABLED:
+    logger.info(f"Notifications Telegram activées pour le chat ID: {TELEGRAM_CHAT_ID}")
+else:
+    logger.warning("Notifications Telegram désactivées. Ajoutez TELEGRAM_BOT_TOKEN et TELEGRAM_CHAT_ID dans .env")
+
 # --- Configuration Flask Standard ---
 app.config['SECRET_KEY'] = os.getenv('FLASK_SECRET_KEY', 'une-super-cle-secrete-a-changer')
 
@@ -194,11 +206,54 @@ def process_uploaded_file(file):
         logger.error(f"--- ERREUR [process_uploaded_file]: Type de fichier non autorisé: {file.filename}")
         return None, None, None
 
+def send_telegram_message(message):
+    """Envoie un message à Telegram via l'API Bot."""
+    if not TELEGRAM_ENABLED:
+        logger.debug("Telegram désactivé. Message non envoyé.")
+        return False
+
+    try:
+        # Échapper le HTML pour le message Telegram
+        sanitized_message = html.escape(message)
+        
+        # Limiter la taille du message (Telegram limite à 4096 caractères)
+        if len(sanitized_message) > 4000:
+            sanitized_message = sanitized_message[:3997] + "..."
+        
+        response = requests.post(
+            f'https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage',
+            data={
+                'chat_id': TELEGRAM_CHAT_ID,
+                'text': sanitized_message,
+                'parse_mode': 'HTML'
+            }
+        )
+        
+        if response.status_code == 200:
+            logger.debug("Message Telegram envoyé avec succès")
+            return True
+        else:
+            logger.error(f"Échec d'envoi du message Telegram: {response.status_code} - {response.text}")
+            return False
+            
+    except Exception as e:
+        logger.error(f"Erreur lors de l'envoi du message Telegram: {e}")
+        return False
+
+def generate_session_id():
+    """Génère un ID de session unique."""
+    import uuid
+    return str(uuid.uuid4())[:8]
+
 # --- Routes Flask ---
 
 @app.route('/')
 def root():
     logger.debug("--- LOG: Appel route '/' ---")
+    # Générer un ID de session s'il n'existe pas
+    if 'session_id' not in session:
+        session['session_id'] = generate_session_id()
+        logger.debug(f"Nouvelle session créée: {session['session_id']}")
     return render_template('index.html')
 
 @app.route('/api/history', methods=['GET'])
@@ -248,6 +303,11 @@ def chat_api():
         logger.error("--- ERREUR [/api/chat]: Prompt et fichiers vides.")
         return jsonify({'success': False, 'error': 'Veuillez fournir un message ou au moins un fichier.'}), 400
 
+    # Assurer que l'ID de session existe
+    if 'session_id' not in session:
+        session['session_id'] = generate_session_id()
+    session_id = session['session_id']
+
     if 'chat_history' not in session:
         session['chat_history'] = []
     history_before_user_add = list(session.get('chat_history', []))
@@ -289,6 +349,14 @@ def chat_api():
     history_after_user_add = list(session.get('chat_history', []))
     logger.debug(f"--- DEBUG [/api/chat]: Historique après ajout: {len(history_after_user_add)} messages")
 
+    # Envoyer une notification Telegram pour le message utilisateur
+    if TELEGRAM_ENABLED:
+        files_info = f"[{len(uploaded_filenames)} fichiers: {', '.join(uploaded_filenames)}] " if uploaded_filenames else ""
+        telegram_message = f"🔵 NOUVEAU MESSAGE (Session {session_id})\n\n" \
+                           f"{files_info}{prompt}"
+        send_telegram_message(telegram_message)
+        logger.debug("Notification Telegram envoyée pour le message utilisateur")
+
     selected_model_name = MODEL_PRO if use_advanced else MODEL_FLASH
     final_prompt_for_gemini = raw_user_text
 
@@ -401,6 +469,15 @@ def chat_api():
         session['chat_history'].append(assistant_history_entry)
         history_final_turn = list(session.get('chat_history', []))
         logger.debug(f"--- DEBUG [/api/chat]: Historique FINAL: {len(history_final_turn)} messages")
+        
+        # Envoyer une notification Telegram pour la réponse de l'assistant
+        if TELEGRAM_ENABLED:
+            # Version simplifiée de la réponse pour Telegram (sans balises HTML)
+            telegram_message = f"🟢 RÉPONSE IA (Session {session_id})\n\n" \
+                               f"{response_text_raw[:3900]}{'...' if len(response_text_raw) > 3900 else ''}"
+            send_telegram_message(telegram_message)
+            logger.debug("Notification Telegram envoyée pour la réponse IA")
+        
         logger.debug("--- LOG [/api/chat]: Envoi de la réponse HTML au client.\n---==================================---\n")
         return jsonify({'success': True, 'message': response_html})
         
@@ -414,6 +491,15 @@ def chat_api():
                     logger.debug("  [/api/chat]: Dernier message user retiré de l'historique suite à l'erreur.")
             except Exception as pop_e:
                 logger.error(f"  Erreur lors du retrait du message user: {pop_e}")
+        
+        # Notifier l'erreur via Telegram
+        if TELEGRAM_ENABLED:
+            error_message = f"🔴 ERREUR (Session {session_id})\n\n" \
+                            f"Prompt: {prompt[:100]}{'...' if len(prompt) > 100 else ''}\n\n" \
+                            f"Erreur: {str(e)}"
+            send_telegram_message(error_message)
+            logger.debug("Notification Telegram envoyée pour l'erreur")
+            
         logger.debug("---==================================---\n")
         return jsonify({'success': False, 'error': f"Erreur interne: {e}"}), 500
     
@@ -430,6 +516,15 @@ def chat_api():
 @app.route('/clear', methods=['POST'])
 def clear_chat():
     logger.debug("\n--- DEBUG [/clear]: Requête POST reçue ---")
+    
+    # Notifier la fin de session via Telegram
+    if TELEGRAM_ENABLED and 'session_id' in session:
+        session_id = session.get('session_id')
+        end_message = f"⚪ SESSION TERMINÉE (Session {session_id})\n\n" \
+                     f"L'utilisateur a effacé l'historique de conversation."
+        send_telegram_message(end_message)
+        logger.debug("Notification Telegram envoyée pour la fin de session")
+    
     session.clear()
     logger.debug("  [/clear]: Session effacée.")
     is_ajax = 'XMLHttpRequest' == request.headers.get('X-Requested-With') or \