Update app.py

app.py

@@ -175,62 +175,90 @@ def root():
     """Sert la page HTML principale."""
     return render_template('index.html')
 
-
 @app.route('/api/history', methods=['GET'])
 def get_history():
-    # Retourne toujours un historique vide
-    return jsonify({'success': True, 'history': []})
-
+    """Fournit l'historique de chat stocké en session au format JSON."""
+    if 'chat_history' not in session:
+        session['chat_history'] = []
+
+    # Préparer l'historique pour l'affichage (contient déjà le HTML pour l'assistant)
+    display_history = [
+        {'role': msg.get('role', 'unknown'), 'text': msg.get('text', '')}
+        for msg in session.get('chat_history', [])
+    ]
+    print(f"API: Récupération de l'historique ({len(display_history)} messages)")
+    return jsonify({'success': True, 'history': display_history})
 
 @app.route('/api/chat', methods=['POST'])
 def chat_api():
+    """Gère les nouvelles requêtes de chat via AJAX."""
+    if not GEMINI_CONFIGURED:
+        print("API ERREUR: Tentative d'appel à /api/chat sans configuration Gemini valide.")
+        return jsonify({'success': False, 'error': "Le service IA n'est pas configuré correctement."}), 503 # Service Unavailable
+
     # Récupération des données du formulaire
     prompt = request.form.get('prompt', '').strip()
-    use_web_search_str = request.form.get('web_search', 'false')
+    use_web_search_str = request.form.get('web_search', 'false') # 'true' ou 'false'
     use_web_search = use_web_search_str.lower() == 'true'
     file = request.files.get('file')
-    use_advanced_str = request.form.get('advanced_reasoning', 'false')
+    use_advanced_str = request.form.get('advanced_reasoning', 'false') # Get the new flag
     use_advanced = use_advanced_str.lower() == 'true'
 
+    # Validation simple
     if not prompt and not file:
         return jsonify({'success': False, 'error': 'Veuillez fournir un message ou un fichier.'}), 400
 
-    # Gestion éventuelle de l'upload de fichier et recherche web...
-    # Préparation du prompt pour Gemini, etc.
-    raw_user_text = prompt
-    final_prompt_for_gemini = raw_user_text
-    if use_web_search and raw_user_text:
-        search_data = perform_web_search(raw_user_text)
-        if search_data:
-            formatted_results = format_search_results(search_data)
-            final_prompt_for_gemini = f"""Voici la question originale de l'utilisateur:
-"{raw_user_text}"
+    print(f"\n--- Nouvelle requête /api/chat ---")
+    print(f"Prompt reçu: '{prompt[:50]}...'")
+    print(f"Recherche Web activée: {use_web_search}")
+    print(f"Raisonnement avancé demandé: {use_advanced}")
+    print(f"Fichier reçu: {file.filename if file else 'Aucun'}")
 
-J'ai effectué une recherche web et voici les informations pertinentes trouvées:
---- DEBUT RESULTATS WEB ---
-{formatted_results}
---- FIN RESULTATS WEB ---
+    # Initialiser l'historique de session si nécessaire
+    if 'chat_history' not in session:
+        session['chat_history'] = []
 
-En te basant sur ces informations ET sur ta connaissance générale, fournis une réponse complète et bien structurée à la question originale de l'utilisateur."""
-    
-    # Préparation des parts pour l'appel Gemini
-    current_gemini_parts = [final_prompt_for_gemini]
-    # (Gestion de fichier si applicable...)
-
-    try:
-        # Appel à Gemini avec current_gemini_parts et sans historique précédent
-        active_model = genai.GenerativeModel(
-            model_name=MODEL_PRO if use_advanced else MODEL_FLASH,
-            safety_settings=SAFETY_SETTINGS,
-            system_instruction=SYSTEM_INSTRUCTION
-        )
-        response = active_model.generate_content(current_gemini_parts)
-        response_text_raw = response.text
-        response_html = markdown.markdown(response_text_raw, extensions=['fenced_code', 'tables', 'nl2br'])
-        return jsonify({'success': True, 'message': response_html})
-    except Exception as e:
-        return jsonify({'success': False, 'error': f"Erreur interne: {e}"}), 500
+    uploaded_gemini_file = None # L'objet fichier retourné par genai.upload_file
+    uploaded_filename = None    # Juste le nom du fichier pour référence/affichage
+    filepath_to_delete = None   # Chemin du fichier local à supprimer après traitement
 
+    # --- Gestion de l'upload de fichier ---
+    if file and 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 # Marquer pour suppression
+                uploaded_filename = filename
+                print(f"Fichier '{filename}' sauvegardé temporairement dans '{filepath}'")
+
+                # Détecter le MimeType pour Gemini
+                mime_type = mimetypes.guess_type(filepath)[0]
+                if not mime_type:
+                    mime_type = 'application/octet-stream' # Fallback
+                    print(f"AVERTISSEMENT: Impossible de deviner le MimeType pour '{filename}', utilisation de '{mime_type}'.")
+
+                # Uploader vers Google AI (peut prendre du temps)
+                print(f"Upload du fichier vers Google AI (MimeType: {mime_type})...")
+                # Note: L'API upload_file est générique et ne dépend pas du modèle Flash/Pro
+                uploaded_gemini_file = genai.upload_file(path=filepath, mime_type=mime_type)
+                print(f"Fichier '{uploaded_gemini_file.name}' uploadé avec succès vers Google AI.")
+
+            except Exception as e:
+                print(f"ERREUR Critique lors du traitement/upload du fichier '{filename}': {e}")
+                # Supprimer le fichier local même en cas d'erreur d'upload Gemini
+                if filepath_to_delete and os.path.exists(filepath_to_delete):
+                    try:
+                        os.remove(filepath_to_delete)
+                        print(f"Fichier temporaire '{filepath_to_delete}' supprimé après erreur.")
+                    except OSError as del_e:
+                         print(f"Erreur lors de la suppression du fichier temporaire après erreur: {del_e}")
+                # Renvoyer une erreur claire au client
+                return jsonify({'success': False, 'error': f"Erreur lors du traitement du fichier: {e}"}), 500
+        else:
+            print(f"ERREUR: Type de fichier non autorisé: {file.filename}")
+            return jsonify({'success': False, 'error': f"Type de fichier non autorisé. Extensions permises: {', '.join(ALLOWED_EXTENSIONS)}"}), 400
 
     # --- Préparation du message utilisateur pour l'historique et Gemini ---
     # Texte brut pour Gemini (et pour l'historique interne)