Update app.py

app.py

@@ -5,8 +5,6 @@ from dotenv import load_dotenv
 import http.client
 import json
 from werkzeug.utils import secure_filename
-import tempfile
-from datetime import datetime
 import uuid
 
 app = Flask(__name__)
@@ -31,12 +29,14 @@ SYSTEM_PROMPT = """
 """
 
 # Initialize Gemini model
-model = genai.GenerativeModel('gemini-2.0-flash-exp', 
-                            tools='code_execution',
-                            safety_settings=safety_settings,
-                            system_instruction=SYSTEM_PROMPT)
-
-# Store chat sessions
+model = genai.GenerativeModel(
+    'gemini-2.0-flash-exp',
+    tools='code_execution',
+    safety_settings=safety_settings,
+    system_instruction=SYSTEM_PROMPT
+)
+
+# Stockage des sessions de chat
 chat_sessions = {}
 
 def perform_web_search(query):
@@ -95,54 +95,69 @@ def home():
     if 'session_id' not in session:
         session['session_id'] = str(uuid.uuid4())
         session['messages'] = []
+        session['uploaded_files'] = []  # initialisation de la liste des fichiers uploadés
     return render_template('index.html', messages=session.get('messages', []))
 
 @app.route('/send_message', methods=['POST'])
 def send_message():
     try:
         data = request.json
-        message = data.get('message')
+        user_message = data.get('message')
         web_search_enabled = data.get('web_search', False)
         
-        if not message:
+        if not user_message:
             return jsonify({'error': 'No message provided'}), 400
 
-        # Ensure session is initialized
+        # Initialiser la session si nécessaire
         if 'session_id' not in session:
             session['session_id'] = str(uuid.uuid4())
             session['messages'] = []
+            session['uploaded_files'] = []
 
         session_id = session['session_id']
 
-        # Initialize chat session if needed
+        # Initialiser la session de chat si nécessaire
         if session_id not in chat_sessions:
             chat_sessions[session_id] = model.start_chat(history=[])
 
-        # Perform web search if enabled
+        # Intégration des fichiers uploadés dans le prompt
+        prompt = ""
+        if 'uploaded_files' in session and session['uploaded_files']:
+            prompt += "Fichiers uploadés:\n"
+            for file_info in session['uploaded_files']:
+                prompt += f"- Nom: {file_info['filename']}, Référence: {file_info['file_id']}\n"
+            prompt += "\n"
+        
+        prompt += f"Question: {user_message}"
+
+        # Réaliser une recherche web si activée
         if web_search_enabled:
-            web_results = perform_web_search(message)
+            web_results = perform_web_search(user_message)
             if web_results:
                 formatted_results = format_search_results(web_results)
-                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?"""
+                prompt += f"\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[session_id].send_message(message)
+        # Envoi du message à Gemini
+        response = chat_sessions[session_id].send_message(prompt)
         
-        # Update message history in session
+        # Mise à jour de l'historique de chat dans la session
         if 'messages' not in session:
             session['messages'] = []
             
         current_messages = session['messages']
         current_messages.append({
             'role': 'user',
-            'content': message
+            'content': prompt
         })
         current_messages.append({
             'role': 'assistant',
             'content': response.text
         })
         session['messages'] = current_messages
-        
+
+        # Optionnel : vider la liste des fichiers uploadés après utilisation
+        session['uploaded_files'] = []
+
         return jsonify({
             'response': response.text
         })
@@ -167,6 +182,13 @@ def upload_file():
         
         try:
             gemini_file = genai.upload_file(filepath)
+            # Stocker la référence du fichier dans la session
+            if 'uploaded_files' not in session:
+                session['uploaded_files'] = []
+            session['uploaded_files'].append({
+                'filename': filename,
+                'file_id': gemini_file.get('id') if isinstance(gemini_file, dict) else gemini_file
+            })
             return jsonify({'success': True, 'filename': filename})
         except Exception as e:
             return jsonify({'error': str(e)}), 500
@@ -179,6 +201,7 @@ def clear_chat():
     if session_id in chat_sessions:
         del chat_sessions[session_id]
     session['messages'] = []
+    session['uploaded_files'] = []
     return jsonify({'success': True})
 
 if __name__ == '__main__':