Update app.py

app.py

@@ -7,6 +7,7 @@ import json
 from werkzeug.utils import secure_filename
 import tempfile
 from datetime import datetime
+import uuid
 
 app = Flask(__name__)
 app.secret_key = os.urandom(24)
@@ -35,17 +36,8 @@ model = genai.GenerativeModel('gemini-2.0-flash-exp',
                             safety_settings=safety_settings,
                             system_instruction=SYSTEM_PROMPT)
 
-# Store chat sessions with user isolation
+# Store chat sessions
 chat_sessions = {}
-chat_histories = {}
-
-def get_client_ip():
-    """Get a unique identifier for the client"""
-    if request.headers.getlist("X-Forwarded-For"):
-        ip = request.headers.getlist("X-Forwarded-For")[0]
-    else:
-        ip = request.remote_addr
-    return ip
 
 def perform_web_search(query):
     conn = http.client.HTTPSConnection("google.serper.dev")
@@ -70,7 +62,6 @@ def format_search_results(data):
         return "Aucun résultat trouvé"
     
     result = []
-    
     if 'knowledgeGraph' in data:
         kg = data['knowledgeGraph']
         result.append({
@@ -101,15 +92,14 @@ def allowed_file(filename):
 
 @app.route('/')
 def home():
-    client_id = get_client_ip()
-    if client_id not in chat_histories:
-        chat_histories[client_id] = []
-    return render_template('index.html', messages=chat_histories[client_id])
+    if 'session_id' not in session:
+        session['session_id'] = str(uuid.uuid4())
+        session['messages'] = []
+    return render_template('index.html', messages=session.get('messages', []))
 
 @app.route('/send_message', methods=['POST'])
 def send_message():
     try:
-        client_id = get_client_ip()
         data = request.json
         message = data.get('message')
         web_search_enabled = data.get('web_search', False)
@@ -117,10 +107,16 @@ def send_message():
         if not message:
             return jsonify({'error': 'No message provided'}), 400
 
-        # Initialize or get existing chat session for this client
-        if client_id not in chat_sessions:
-            chat_sessions[client_id] = model.start_chat(history=[])
-            chat_histories[client_id] = []
+        # Ensure session is initialized
+        if 'session_id' not in session:
+            session['session_id'] = str(uuid.uuid4())
+            session['messages'] = []
+
+        session_id = session['session_id']
+
+        # Initialize chat session if needed
+        if session_id not in chat_sessions:
+            chat_sessions[session_id] = model.start_chat(history=[])
 
         # Perform web search if enabled
         if web_search_enabled:
@@ -130,19 +126,22 @@ def send_message():
                 message = f"""Question: {message}\n\nRésultats de recherche web:\n{formatted_results}\n\nPourrais-tu analyser ces informations et me donner une réponse complète?"""
 
         # Send message to Gemini
-        response = chat_sessions[client_id].send_message(message)
+        response = chat_sessions[session_id].send_message(message)
         
-        # Update message history for this client
-        chat_histories[client_id].append({
+        # Update message history in session
+        if 'messages' not in session:
+            session['messages'] = []
+            
+        current_messages = session['messages']
+        current_messages.append({
             'role': 'user',
-            'content': message,
-            'timestamp': datetime.now().isoformat()
+            'content': message
         })
-        chat_histories[client_id].append({
+        current_messages.append({
             'role': 'assistant',
-            'content': response.text,
-            'timestamp': datetime.now().isoformat()
+            'content': response.text
         })
+        session['messages'] = current_messages
         
         return jsonify({
             'response': response.text
@@ -176,11 +175,10 @@ def upload_file():
 
 @app.route('/clear_chat', methods=['POST'])
 def clear_chat():
-    client_id = get_client_ip()
-    if client_id in chat_sessions:
-        del chat_sessions[client_id]
-    if client_id in chat_histories:
-        del chat_histories[client_id]
+    session_id = session.get('session_id')
+    if session_id in chat_sessions:
+        del chat_sessions[session_id]
+    session['messages'] = []
     return jsonify({'success': True})
 
 if __name__ == '__main__':