Update app.py

app.py

@@ -8,7 +8,7 @@ from werkzeug.utils import secure_filename
 import tempfile
 
 app = Flask(__name__)
-app.secret_key = os.urandom(24)  # For session management
+app.secret_key = os.urandom(24)
 load_dotenv()
 
 # Configure Google AI
@@ -34,6 +34,9 @@ model = genai.GenerativeModel('gemini-2.0-flash-exp',
                             safety_settings=safety_settings,
                             system_instruction=SYSTEM_PROMPT)
 
+# Store chat sessions in a dictionary
+chat_sessions = {}
+
 def perform_web_search(query):
     conn = http.client.HTTPSConnection("google.serper.dev")
     payload = json.dumps({"q": query})
@@ -76,14 +79,6 @@ def format_search_results(data):
                 'link': item['link']
             })
     
-    if 'peopleAlsoAsk' in data:
-        for item in data['peopleAlsoAsk'][:2]:
-            result.append({
-                'type': 'question',
-                'question': item['question'],
-                'answer': item['snippet']
-            })
-    
     return result
 
 UPLOAD_FOLDER = 'temp'
@@ -96,9 +91,11 @@ def allowed_file(filename):
 
 @app.route('/')
 def home():
-    if 'chat_history' not in session:
-        session['chat_history'] = []
-    return render_template('index.html', messages=session['chat_history'])
+    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'])
 
 @app.route('/send_message', methods=['POST'])
 def send_message():
@@ -110,12 +107,11 @@ def send_message():
         if not message:
             return jsonify({'error': 'No message provided'}), 400
 
-        # Initialize chat if not in session
-        if 'chat' not in session:
-            session['chat'] = model.start_chat(history=[])
+        user_id = session.get('user_id')
+        if user_id not in chat_sessions:
+            chat_sessions[user_id] = model.start_chat(history=[])
 
         # Perform web search if enabled
-        web_results = None
         if web_search_enabled:
             web_results = perform_web_search(message)
             if web_results:
@@ -123,21 +119,23 @@ 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 = session['chat'].send_message(message)
+        response = chat_sessions[user_id].send_message(message)
+        
+        # Update message history in session
+        if 'messages' not in session:
+            session['messages'] = []
         
-        # Update chat history
-        session['chat_history'].append({
+        session['messages'].append({
             'role': 'user',
             'content': message
         })
-        session['chat_history'].append({
+        session['messages'].append({
             'role': 'assistant',
             'content': response.text
         })
         
         return jsonify({
-            'response': response.text,
-            'web_results': web_results
+            'response': response.text
         })
 
     except Exception as e:
@@ -167,9 +165,11 @@ def upload_file():
 
 @app.route('/clear_chat', methods=['POST'])
 def clear_chat():
-    session.pop('chat_history', None)
-    session.pop('chat', None)
+    user_id = session.get('user_id')
+    if user_id in chat_sessions:
+        del chat_sessions[user_id]
+    session['messages'] = []
     return jsonify({'success': True})
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run()