Update app.py

app.py

@@ -1,5 +1,3 @@
-# --- START OF FILE app.py ---
-
 import os
 import json
 import mimetypes
@@ -165,6 +163,37 @@ def prepare_gemini_history(chat_history):
     logger.debug(f"--- DEBUG [prepare_gemini_history]: Sortie avec {len(gemini_history)} messages")
     return gemini_history
 
+def process_uploaded_file(file):
+    """Traite un fichier uploadé et retourne les infos nécessaires pour Gemini."""
+    if not file or file.filename == '':
+        return None, None, None
+    
+    if allowed_file(file.filename):
+        try:
+            filename = secure_filename(file.filename)
+            filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+            file.save(filepath)
+            logger.debug(f"  [process_uploaded_file]: Fichier '{filename}' sauvegardé dans '{filepath}'")
+            
+            mime_type = mimetypes.guess_type(filepath)[0] or 'application/octet-stream'
+            with open(filepath, "rb") as f:
+                file_data = f.read()
+                
+            file_part = {
+                "inline_data": {
+                    "mime_type": mime_type,
+                    "data": file_data
+                }
+            }
+            
+            return file_part, filename, filepath
+        except Exception as e:
+            logger.error(f"--- ERREUR [process_uploaded_file]: Échec traitement fichier '{file.filename}': {e}")
+            return None, None, None
+    else:
+        logger.error(f"--- ERREUR [process_uploaded_file]: Type de fichier non autorisé: {file.filename}")
+        return None, None, None
+
 # --- Routes Flask ---
 
 @app.route('/')
@@ -206,65 +235,53 @@ def chat_api():
 
     prompt = request.form.get('prompt', '').strip()
     use_web_search = request.form.get('web_search', 'false').lower() == 'true'
-    file = request.files.get('file')
     use_advanced = request.form.get('advanced_reasoning', 'false').lower() == 'true'
 
+    # Récupération de tous les fichiers (multiple)
+    files = request.files.getlist('file')
+    
     logger.debug(f"  [/api/chat]: Prompt reçu: '{prompt[:50]}...'")
     logger.debug(f"  [/api/chat]: Recherche Web: {use_web_search}, Raisonnement Avancé: {use_advanced}")
-    logger.debug(f"  [/api/chat]: Fichier: {file.filename if file else 'Aucun'}")
-
-    if not prompt and not file:
-        logger.error("--- ERREUR [/api/chat]: Prompt et fichier vides.")
-        return jsonify({'success': False, 'error': 'Veuillez fournir un message ou un fichier.'}), 400
+    logger.debug(f"  [/api/chat]: Nombre de fichiers: {len(files) if files else 0}")
+    
+    if not prompt and not files:
+        logger.error("--- ERREUR [/api/chat]: Prompt et fichiers vides.")
+        return jsonify({'success': False, 'error': 'Veuillez fournir un message ou au moins un fichier.'}), 400
 
     if 'chat_history' not in session:
         session['chat_history'] = []
     history_before_user_add = list(session.get('chat_history', []))
     logger.debug(f"--- DEBUG [/api/chat]: Historique avant ajout: {len(history_before_user_add)} messages")
 
-    uploaded_file_part = None
-    uploaded_filename = None
-    filepath_to_delete = None
+    uploaded_file_parts = []
+    uploaded_filenames = []
+    filepaths_to_delete = []
 
-    if file and file.filename != '':
-        logger.debug(f"--- LOG [/api/chat]: Traitement du fichier '{file.filename}'")
-        if allowed_file(file.filename):
-            try:
-                filename = secure_filename(file.filename)
-                filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-                file.save(filepath)
-                filepath_to_delete = filepath
-                uploaded_filename = filename
-                logger.debug(f"  [/api/chat]: Fichier '{filename}' sauvegardé dans '{filepath}'")
-                mime_type = mimetypes.guess_type(filepath)[0] or 'application/octet-stream'
-                with open(filepath, "rb") as f:
-                    file_data = f.read()
-                uploaded_file_part = {
-                    "inline_data": {
-                        "mime_type": mime_type,
-                        "data": file_data
-                    }
-                }
-                logger.debug(f"  [/api/chat]: Fichier préparé pour Gemini.")
-            except Exception as e:
-                logger.error(f"--- ERREUR [/api/chat]: Échec traitement fichier '{filename}': {e}")
-                if filepath_to_delete and os.path.exists(filepath_to_delete):
-                    try: 
-                        os.remove(filepath_to_delete)
-                    except OSError as del_e: 
-                        logger.error(f"  Erreur suppression fichier temp: {del_e}")
-                return jsonify({'success': False, 'error': f"Erreur traitement fichier: {e}"}), 500
-        else:
-            logger.error(f"--- ERREUR [/api/chat]: Type de fichier non autorisé: {file.filename}")
-            return jsonify({'success': False, 'error': f"Type de fichier non autorisé."}), 400
+    # Traiter tous les fichiers
+    for file in files:
+        if file and file.filename != '':
+            logger.debug(f"--- LOG [/api/chat]: Traitement du fichier '{file.filename}'")
+            file_part, filename, filepath = process_uploaded_file(file)
+            
+            if file_part and filename and filepath:
+                uploaded_file_parts.append(file_part)
+                uploaded_filenames.append(filename)
+                filepaths_to_delete.append(filepath)
+                logger.debug(f"  [/api/chat]: Fichier '{filename}' ajouté à la liste")
+            else:
+                logger.warning(f"  [/api/chat]: Échec du traitement pour '{file.filename}'")
 
     raw_user_text = prompt
-    display_user_text = f"[{uploaded_filename}] {prompt}" if uploaded_filename and prompt else (prompt or f"[{uploaded_filename}]")
+    # Créer un texte d'affichage avec tous les noms de fichiers
+    files_text = ", ".join([f"[{filename}]" for filename in uploaded_filenames]) if uploaded_filenames else ""
+    display_user_text = f"{files_text} {prompt}" if files_text and prompt else (prompt or files_text)
+    
     user_history_entry = {
         'role': 'user',
         'text': display_user_text,
         'raw_text': raw_user_text,
     }
+    
     if not isinstance(session.get('chat_history'), list):
         logger.error("--- ERREUR [/api/chat]: 'chat_history' n'est pas une liste! Réinitialisation.")
         session['chat_history'] = []
@@ -275,10 +292,14 @@ def chat_api():
     selected_model_name = MODEL_PRO if use_advanced else MODEL_FLASH
     final_prompt_for_gemini = raw_user_text
 
-    if uploaded_file_part and not raw_user_text:
-        raw_user_text = f"Décris le contenu de ce fichier : {uploaded_filename}"
+    # Si seulement des fichiers sont fournis sans texte, générer un prompt approprié
+    if uploaded_filenames and not raw_user_text:
+        if len(uploaded_filenames) == 1:
+            raw_user_text = f"Décris le contenu de ce fichier : {uploaded_filenames[0]}"
+        else:
+            raw_user_text = f"Décris le contenu de ces fichiers : {', '.join(uploaded_filenames)}"
         final_prompt_for_gemini = raw_user_text
-        logger.debug(f"  [/api/chat]: Fichier seul détecté, prompt généré: '{final_prompt_for_gemini}'")
+        logger.debug(f"  [/api/chat]: Fichier(s) seul(s) détecté(s), prompt généré: '{final_prompt_for_gemini}'")
 
     # --- Appel séparé de la recherche web si demandée ---
     search_results_text = ""
@@ -295,14 +316,17 @@ def chat_api():
 
     # Préparer les parties du message utilisateur final
     current_user_parts = []
-    if uploaded_file_part:
-        current_user_parts.append(uploaded_file_part)
+    
+    # Ajouter tous les fichiers au message
+    for file_part in uploaded_file_parts:
+        current_user_parts.append(file_part)
     
     # Intégrer le prompt initial et, le cas échéant, le résumé des résultats de recherche
     combined_prompt = final_prompt_for_gemini
     if search_results_text:
         combined_prompt += "\n\nRésultats de recherche récents:\n" + search_results_text
     current_user_parts.append({"text": combined_prompt})
+    
     contents.append({
         "role": "user",
         "parts": current_user_parts
@@ -394,12 +418,14 @@ def chat_api():
         return jsonify({'success': False, 'error': f"Erreur interne: {e}"}), 500
     
     finally:
-        if filepath_to_delete and os.path.exists(filepath_to_delete):
-            try:
-                os.remove(filepath_to_delete)
-                logger.debug(f"--- LOG [/api/chat FINALLY]: Fichier temporaire '{filepath_to_delete}' supprimé.")
-            except OSError as e_del_local:
-                logger.error(f"--- ERREUR [/api/chat FINALLY]: Échec suppression fichier '{filepath_to_delete}': {e_del_local}")
+        # Nettoyage de tous les fichiers temporaires
+        for filepath in filepaths_to_delete:
+            if filepath and os.path.exists(filepath):
+                try:
+                    os.remove(filepath)
+                    logger.debug(f"--- LOG [/api/chat FINALLY]: Fichier temporaire '{filepath}' supprimé.")
+                except OSError as e_del_local:
+                    logger.error(f"--- ERREUR [/api/chat FINALLY]: Échec suppression fichier '{filepath}': {e_del_local}")
 
 @app.route('/clear', methods=['POST'])
 def clear_chat():
@@ -419,6 +445,4 @@ def clear_chat():
 if __name__ == '__main__':
     logger.info("--- Démarrage du serveur Flask ---")
     port = int(os.environ.get('PORT', 5001))
-    app.run(debug=True, host='0.0.0.0', port=port)
-
-# --- END OF FILE app.py ---
+    app.run(debug=True, host='0.0.0.0', port=port)