Update app.py

app.py

@@ -146,6 +146,9 @@ def get_conversation(conversation_id):
         return jsonify({'error': 'Conversation non trouvée'}), 404
     return jsonify(conversations[conversation_id])
 
+
+# Dans app.py, modifiez la fonction send_message :
+
 @app.route('/send_message', methods=['POST'])
 def send_message():
     """Traite l'envoi d'un message"""
@@ -161,6 +164,19 @@ def send_message():
         # Crée une nouvelle conversation si nécessaire
         if not conversation_id or conversation_id not in conversations:
             conversation_id = create_new_conversation()
+        
+        # Récupère ou crée la session de chat
+        if conversation_id not in chat_sessions:
+            # Reconstruit l'historique complet pour Gemini
+            history = []
+            if conversation_id in conversations:
+                for msg in conversations[conversation_id]['messages']:
+                    role = 'user' if msg['role'] == 'user' else 'model'
+                    history.append({
+                        'role': role,
+                        'parts': [msg['content']]
+                    })
+            chat_sessions[conversation_id] = model.start_chat(history=history)
 
         # Effectue une recherche web si activée
         search_results = ""
@@ -194,6 +210,38 @@ def send_message():
         print(f"Erreur lors de l'envoi du message: {e}")
         return jsonify({'error': str(e)}), 500
 
+# Ajoutez cette nouvelle route pour gérer le changement de conversation
+@app.route('/switch_conversation', methods=['POST'])
+def switch_conversation():
+    """Gère le changement de conversation"""
+    try:
+        data = request.json
+        conversation_id = data.get('conversation_id')
+        
+        if not conversation_id:
+            return jsonify({'error': 'ID de conversation manquant'}), 400
+            
+        if conversation_id not in conversations:
+            return jsonify({'error': 'Conversation non trouvée'}), 404
+            
+        # Recrée la session avec l'historique complet
+        history = []
+        for msg in conversations[conversation_id]['messages']:
+            role = 'user' if msg['role'] == 'user' else 'model'
+            history.append({
+                'role': role,
+                'parts': [msg['content']]
+            })
+            
+        chat_sessions[conversation_id] = model.start_chat(history=history)
+        
+        return jsonify({'success': True})
+        
+    except Exception as e:
+        print(f"Erreur lors du changement de conversation: {e}")
+        return jsonify({'error': str(e)}), 500
+
+
 @app.route('/upload', methods=['POST'])
 def upload_file():
     """Gère l'upload de fichiers"""