Update app.py

app.py

@@ -6,6 +6,7 @@ import http.client
 import json
 from werkzeug.utils import secure_filename
 import tempfile
+from datetime import datetime
 
 app = Flask(__name__)
 app.secret_key = os.urandom(24)
@@ -34,8 +35,17 @@ model = genai.GenerativeModel('gemini-2.0-flash-exp',
                             safety_settings=safety_settings,
                             system_instruction=SYSTEM_PROMPT)
 
-# Store chat sessions in a dictionary
+# Store chat sessions with user isolation
 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")
@@ -91,15 +101,15 @@ def allowed_file(filename):
 
 @app.route('/')
 def home():
-    if 'user_id' not in session:
-        session['user_id'] = os.urandom(16).hex()
-    if 'messages' not in session:
-        session['messages'] = []
-    return render_template('index.html', messages=session['messages'])
+    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])
 
 @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)
@@ -107,9 +117,10 @@ def send_message():
         if not message:
             return jsonify({'error': 'No message provided'}), 400
 
-        user_id = session.get('user_id')
-        if user_id not in chat_sessions:
-            chat_sessions[user_id] = model.start_chat(history=[])
+        # 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] = []
 
         # Perform web search if enabled
         if web_search_enabled:
@@ -119,19 +130,18 @@ 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[user_id].send_message(message)
-        
-        # Update message history in session
-        if 'messages' not in session:
-            session['messages'] = []
+        response = chat_sessions[client_id].send_message(message)
         
-        session['messages'].append({
+        # Update message history for this client
+        chat_histories[client_id].append({
             'role': 'user',
-            'content': message
+            'content': message,
+            'timestamp': datetime.now().isoformat()
         })
-        session['messages'].append({
+        chat_histories[client_id].append({
             'role': 'assistant',
-            'content': response.text
+            'content': response.text,
+            'timestamp': datetime.now().isoformat()
         })
         
         return jsonify({
@@ -139,6 +149,7 @@ def send_message():
         })
 
     except Exception as e:
+        print(f"Error in send_message: {e}")
         return jsonify({'error': str(e)}), 500
 
 @app.route('/upload', methods=['POST'])
@@ -165,10 +176,11 @@ def upload_file():
 
 @app.route('/clear_chat', methods=['POST'])
 def clear_chat():
-    user_id = session.get('user_id')
-    if user_id in chat_sessions:
-        del chat_sessions[user_id]
-    session['messages'] = []
+    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]
     return jsonify({'success': True})
 
 if __name__ == '__main__':