Delete app.py
Browse files
app.py
DELETED
@@ -1,520 +0,0 @@
|
|
1 |
-
# --- START OF FILE app.py ---
|
2 |
-
|
3 |
-
import os
|
4 |
-
import json
|
5 |
-
import mimetypes
|
6 |
-
from flask import Flask, request, session, jsonify, redirect, url_for, flash, render_template
|
7 |
-
from dotenv import load_dotenv
|
8 |
-
from google import genai
|
9 |
-
from google.genai import types
|
10 |
-
import requests
|
11 |
-
from werkzeug.utils import secure_filename
|
12 |
-
import markdown # Pour convertir la réponse en HTML
|
13 |
-
from flask_session import Session # <-- Importer Session
|
14 |
-
import pprint # Pour un affichage plus lisible des structures complexes (optionnel)
|
15 |
-
|
16 |
-
# --- Configuration Initiale ---
|
17 |
-
load_dotenv()
|
18 |
-
|
19 |
-
app = Flask(__name__)
|
20 |
-
|
21 |
-
# --- Configuration Flask Standard ---
|
22 |
-
# Clé secrète FORTEMENT recommandée (vous l'avez déjà)
|
23 |
-
# Gardez-la secrète en production !
|
24 |
-
app.config['SECRET_KEY'] = os.getenv('FLASK_SECRET_KEY', 'une-super-cle-secrete-a-changer')
|
25 |
-
|
26 |
-
# Configuration pour les uploads (vous l'avez déjà)
|
27 |
-
UPLOAD_FOLDER = 'temp'
|
28 |
-
ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg'}
|
29 |
-
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
|
30 |
-
app.config['MAX_CONTENT_LENGTH'] = 25 * 1024 * 1024 # Limite de taille (ex: 25MB)
|
31 |
-
|
32 |
-
# Créer le dossier temp s'il n'existe pas (vous l'avez déjà)
|
33 |
-
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
|
34 |
-
print(f"Dossier d'upload configuré : {os.path.abspath(UPLOAD_FOLDER)}")
|
35 |
-
|
36 |
-
# --- Configuration pour Flask-Session (Backend Filesystem) ---
|
37 |
-
app.config['SESSION_TYPE'] = 'filesystem' # Indique d'utiliser le stockage par fichiers
|
38 |
-
app.config['SESSION_PERMANENT'] = False # La session expire quand le navigateur est fermé
|
39 |
-
app.config['SESSION_USE_SIGNER'] = True # Signe l'ID de session dans le cookie pour sécurité
|
40 |
-
app.config['SESSION_FILE_DIR'] = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'flask_session') # Chemin où stocker les fichiers de session
|
41 |
-
|
42 |
-
app.config['SESSION_COOKIE_SAMESITE'] = 'None'
|
43 |
-
# Nécessite HTTPS pour que 'None' fonctionne
|
44 |
-
app.config['SESSION_COOKIE_SECURE'] = True
|
45 |
-
|
46 |
-
# Crée le dossier pour les sessions filesystem s'il n'existe pas
|
47 |
-
os.makedirs(app.config['SESSION_FILE_DIR'], exist_ok=True)
|
48 |
-
print(f"Dossier pour les sessions serveur configuré : {app.config['SESSION_FILE_DIR']}")
|
49 |
-
|
50 |
-
# --- Initialisation de Flask-Session ---
|
51 |
-
server_session = Session(app)
|
52 |
-
|
53 |
-
# --- Configuration de l'API Gemini ---
|
54 |
-
MODEL_FLASH = 'gemini-2.0-flash'
|
55 |
-
MODEL_PRO = 'gemini-2.5-pro-exp-03-25'
|
56 |
-
SYSTEM_INSTRUCTION = "Tu es un assistant intelligent et amical nommé Mariam. Tu assistes les utilisateurs au mieux de tes capacités. Tu as été créé par Aenir."
|
57 |
-
|
58 |
-
# Nouveaux réglages de sécurité avec le nouveau SDK
|
59 |
-
SAFETY_SETTINGS = [
|
60 |
-
types.SafetySetting(
|
61 |
-
category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
|
62 |
-
threshold=types.HarmBlockThreshold.BLOCK_NONE,
|
63 |
-
),
|
64 |
-
types.SafetySetting(
|
65 |
-
category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
|
66 |
-
threshold=types.HarmBlockThreshold.BLOCK_NONE,
|
67 |
-
),
|
68 |
-
types.SafetySetting(
|
69 |
-
category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
|
70 |
-
threshold=types.HarmBlockThreshold.BLOCK_NONE,
|
71 |
-
),
|
72 |
-
types.SafetySetting(
|
73 |
-
category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
|
74 |
-
threshold=types.HarmBlockThreshold.BLOCK_NONE,
|
75 |
-
)
|
76 |
-
]
|
77 |
-
|
78 |
-
GEMINI_CONFIGURED = False
|
79 |
-
genai_client = None
|
80 |
-
try:
|
81 |
-
gemini_api_key = os.getenv("GOOGLE_API_KEY")
|
82 |
-
if not gemini_api_key:
|
83 |
-
print("ERREUR: Clé API GOOGLE_API_KEY manquante dans le fichier .env")
|
84 |
-
else:
|
85 |
-
# Initialisation du client avec le nouveau SDK
|
86 |
-
genai_client = genai.Client(api_key=gemini_api_key)
|
87 |
-
|
88 |
-
# Vérification de la disponibilité des modèles
|
89 |
-
try:
|
90 |
-
# Le nouveau SDK peut avoir une méthode différente pour lister les modèles
|
91 |
-
models = genai_client.list_models()
|
92 |
-
models_list = [model.name for model in models]
|
93 |
-
|
94 |
-
# Vérifier si les modèles requis sont disponibles
|
95 |
-
if any(MODEL_FLASH in model for model in models_list) and any(MODEL_PRO in model for model in models_list):
|
96 |
-
print(f"Configuration Gemini effectuée. Modèles requis ({MODEL_FLASH}, {MODEL_PRO}) disponibles.")
|
97 |
-
print(f"System instruction: {SYSTEM_INSTRUCTION}")
|
98 |
-
GEMINI_CONFIGURED = True
|
99 |
-
else:
|
100 |
-
print(f"ERREUR: Les modèles requis ({MODEL_FLASH}, {MODEL_PRO}) ne sont pas tous disponibles via l'API.")
|
101 |
-
print(f"Modèles trouvés: {models_list}")
|
102 |
-
except Exception as e_models:
|
103 |
-
print(f"ERREUR lors de la vérification des modèles: {e_models}")
|
104 |
-
# En cas d'échec de la vérification des modèles, supposons que les modèles sont disponibles
|
105 |
-
print("Tentative de continuer sans vérification des modèles disponibles.")
|
106 |
-
GEMINI_CONFIGURED = True
|
107 |
-
|
108 |
-
except Exception as e:
|
109 |
-
print(f"ERREUR Critique lors de la configuration initiale de Gemini : {e}")
|
110 |
-
print("L'application fonctionnera sans les fonctionnalités IA.")
|
111 |
-
|
112 |
-
# --- Fonctions Utilitaires ---
|
113 |
-
|
114 |
-
def allowed_file(filename):
|
115 |
-
"""Vérifie si l'extension du fichier est autorisée."""
|
116 |
-
return '.' in filename and \
|
117 |
-
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
|
118 |
-
|
119 |
-
def perform_web_search_with_gemini(query, client, model_id):
|
120 |
-
"""Effectue une recherche web via l'API Google Search intégrée à Gemini."""
|
121 |
-
try:
|
122 |
-
print(f"--- LOG WEBSEARCH: Recherche Google avec Gemini pour: '{query}'")
|
123 |
-
|
124 |
-
# Utiliser l'outil de recherche Google intégré à Gemini
|
125 |
-
response = client.models.generate_content(
|
126 |
-
model=model_id,
|
127 |
-
contents=query,
|
128 |
-
config=types.GenerateContentConfig(
|
129 |
-
tools=[{"google_search": {}}]
|
130 |
-
)
|
131 |
-
)
|
132 |
-
|
133 |
-
print("--- LOG WEBSEARCH: Résultats de recherche Google obtenus.")
|
134 |
-
|
135 |
-
# Extraction des informations de recherche pour le débogage
|
136 |
-
if hasattr(response, 'candidates') and response.candidates:
|
137 |
-
if hasattr(response.candidates[0], 'grounding_metadata'):
|
138 |
-
search_queries = getattr(response.candidates[0].grounding_metadata, 'web_search_queries', None)
|
139 |
-
if search_queries:
|
140 |
-
print(f"--- LOG WEBSEARCH: Requêtes utilisées: {search_queries}")
|
141 |
-
|
142 |
-
grounding_chunks = getattr(response.candidates[0].grounding_metadata, 'grounding_chunks', None)
|
143 |
-
if grounding_chunks:
|
144 |
-
sources = [getattr(chunk, 'web', None) for chunk in grounding_chunks]
|
145 |
-
source_titles = [getattr(source, 'title', 'Sans titre') for source in sources if source]
|
146 |
-
print(f"--- LOG WEBSEARCH: Sources utilisées: {', '.join(source_titles)}")
|
147 |
-
|
148 |
-
return response
|
149 |
-
|
150 |
-
except Exception as e:
|
151 |
-
print(f"--- LOG WEBSEARCH: Erreur lors de la recherche web avec Gemini : {e}")
|
152 |
-
return None
|
153 |
-
|
154 |
-
def format_search_response(response):
|
155 |
-
"""Extrait et met en forme le texte de la réponse de recherche web."""
|
156 |
-
if not response:
|
157 |
-
return "Aucun résultat de recherche web trouvé pertinent."
|
158 |
-
|
159 |
-
try:
|
160 |
-
return response.text
|
161 |
-
except Exception as e:
|
162 |
-
print(f"--- LOG WEBSEARCH: Erreur lors de l'extraction du texte de la réponse : {e}")
|
163 |
-
return "Impossible d'extraire les résultats de recherche."
|
164 |
-
|
165 |
-
def prepare_gemini_history(chat_history):
|
166 |
-
"""Convertit l'historique stocké en session au format attendu par Gemini API."""
|
167 |
-
print(f"--- DEBUG [prepare_gemini_history]: Entrée avec {len(chat_history)} messages") # LOG 1
|
168 |
-
# Avec le nouveau SDK, la structure de l'historique pourrait être différente
|
169 |
-
gemini_history = []
|
170 |
-
for i, message in enumerate(list(chat_history)): # Utiliser list() pour itérer sur une copie
|
171 |
-
role = 'user' if message.get('role') == 'user' else 'model' # 'model' au lieu de 'assistant' pour le nouveau SDK
|
172 |
-
text_part = message.get('raw_text')
|
173 |
-
# Log détaillé pour chaque message traité
|
174 |
-
print(f" [prepare_gemini_history] Message {i} (rôle session: {message.get('role')}, rôle gemini: {role}): raw_text présent? {'Oui' if text_part is not None else 'NON'}, contenu début: '{str(text_part)[:60]}...'") # LOG 2
|
175 |
-
|
176 |
-
if text_part: # Important: Ne pas ajouter de messages vides à l'historique Gemini
|
177 |
-
gemini_history.append({
|
178 |
-
'role': role,
|
179 |
-
'parts': [text_part]
|
180 |
-
})
|
181 |
-
else:
|
182 |
-
# Log si un message est ignoré car vide
|
183 |
-
print(f" AVERTISSEMENT [prepare_gemini_history]: raw_text vide ou absent pour le message {i}, ignoré pour l'historique Gemini.") # LOG 3
|
184 |
-
|
185 |
-
print(f"--- DEBUG [prepare_gemini_history]: Sortie avec {len(gemini_history)} messages formatés pour Gemini") # LOG 4
|
186 |
-
return gemini_history
|
187 |
-
|
188 |
-
# --- Routes Flask ---
|
189 |
-
|
190 |
-
@app.route('/')
|
191 |
-
def root():
|
192 |
-
"""Sert la page HTML principale."""
|
193 |
-
print("--- LOG: Appel route '/' ---")
|
194 |
-
return render_template('index.html')
|
195 |
-
|
196 |
-
@app.route('/api/history', methods=['GET'])
|
197 |
-
def get_history():
|
198 |
-
"""Fournit l'historique de chat stocké en session au format JSON."""
|
199 |
-
print("\n--- DEBUG [/api/history]: Début requête GET ---") # LOG 5
|
200 |
-
if 'chat_history' not in session:
|
201 |
-
session['chat_history'] = []
|
202 |
-
print(" [/api/history]: Session 'chat_history' initialisée (vide).")
|
203 |
-
|
204 |
-
display_history = []
|
205 |
-
current_history = session.get('chat_history', [])
|
206 |
-
print(f" [/api/history]: Historique récupéré de la session serveur: {len(current_history)} messages.") # LOG 6
|
207 |
-
|
208 |
-
# Optionnel: Afficher la structure brute pour un debug profond
|
209 |
-
# print(" [/api/history]: Contenu brut de l'historique session:")
|
210 |
-
# pprint.pprint(current_history)
|
211 |
-
|
212 |
-
for i, msg in enumerate(current_history):
|
213 |
-
# Vérifier la structure de chaque message récupéré
|
214 |
-
if isinstance(msg, dict) and 'role' in msg and 'text' in msg:
|
215 |
-
display_history.append({
|
216 |
-
'role': msg.get('role'),
|
217 |
-
'text': msg.get('text') # On envoie bien le HTML ('text') au frontend
|
218 |
-
})
|
219 |
-
else:
|
220 |
-
# Log si un message dans la session est mal formé
|
221 |
-
print(f" AVERTISSEMENT [/api/history]: Format invalide dans l'historique session au message {i}: {msg}") # LOG 7
|
222 |
-
|
223 |
-
print(f" [/api/history]: Historique préparé pour le frontend: {len(display_history)} messages.") # LOG 8
|
224 |
-
return jsonify({'success': True, 'history': display_history})
|
225 |
-
|
226 |
-
@app.route('/api/chat', methods=['POST'])
|
227 |
-
def chat_api():
|
228 |
-
"""Gère les nouvelles requêtes de chat via AJAX."""
|
229 |
-
print(f"\n---===================================---")
|
230 |
-
print(f"--- DEBUG [/api/chat]: Nouvelle requête POST ---")
|
231 |
-
|
232 |
-
if not GEMINI_CONFIGURED or not genai_client:
|
233 |
-
print("--- ERREUR [/api/chat]: Tentative d'appel sans configuration Gemini valide.")
|
234 |
-
return jsonify({'success': False, 'error': "Le service IA n'est pas configuré correctement."}), 503
|
235 |
-
|
236 |
-
# Récupération des données du formulaire
|
237 |
-
prompt = request.form.get('prompt', '').strip()
|
238 |
-
use_web_search = request.form.get('web_search', 'false').lower() == 'true'
|
239 |
-
file = request.files.get('file')
|
240 |
-
use_advanced = request.form.get('advanced_reasoning', 'false').lower() == 'true'
|
241 |
-
|
242 |
-
print(f" [/api/chat]: Prompt reçu: '{prompt[:50]}...'")
|
243 |
-
print(f" [/api/chat]: Recherche Web: {use_web_search}, Raisonnement Avancé: {use_advanced}")
|
244 |
-
print(f" [/api/chat]: Fichier: {file.filename if file else 'Aucun'}")
|
245 |
-
|
246 |
-
# Validation
|
247 |
-
if not prompt and not file:
|
248 |
-
print("--- ERREUR [/api/chat]: Prompt et fichier vides.")
|
249 |
-
return jsonify({'success': False, 'error': 'Veuillez fournir un message ou un fichier.'}), 400
|
250 |
-
|
251 |
-
# --- Log de l'état de l'historique AVANT toute modification ---
|
252 |
-
if 'chat_history' not in session:
|
253 |
-
session['chat_history'] = []
|
254 |
-
history_before_user_add = list(session.get('chat_history', [])) # Copie pour le log
|
255 |
-
print(f"--- DEBUG [/api/chat]: Historique en session AVANT ajout user message: {len(history_before_user_add)} messages") # LOG 9
|
256 |
-
|
257 |
-
uploaded_file_part = None
|
258 |
-
uploaded_filename = None
|
259 |
-
filepath_to_delete = None
|
260 |
-
|
261 |
-
# --- Gestion de l'upload de fichier (avec logs) ---
|
262 |
-
if file and file.filename != '':
|
263 |
-
print(f"--- LOG [/api/chat]: Traitement du fichier '{file.filename}'")
|
264 |
-
if allowed_file(file.filename):
|
265 |
-
try:
|
266 |
-
filename = secure_filename(file.filename)
|
267 |
-
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
268 |
-
file.save(filepath)
|
269 |
-
filepath_to_delete = filepath
|
270 |
-
uploaded_filename = filename
|
271 |
-
print(f" [/api/chat]: Fichier '{filename}' sauvegardé dans '{filepath}'")
|
272 |
-
mime_type = mimetypes.guess_type(filepath)[0] or 'application/octet-stream'
|
273 |
-
print(f" [/api/chat]: Préparation du fichier pour Gemini (Mime: {mime_type})...")
|
274 |
-
|
275 |
-
# Avec le nouveau SDK, la méthode d'upload a peut-être changé
|
276 |
-
with open(filepath, "rb") as f:
|
277 |
-
file_data = f.read()
|
278 |
-
|
279 |
-
# Création d'une partie de type fichier pour l'API
|
280 |
-
uploaded_file_part = {
|
281 |
-
"inline_data": {
|
282 |
-
"mime_type": mime_type,
|
283 |
-
"data": file_data
|
284 |
-
}
|
285 |
-
}
|
286 |
-
print(f" [/api/chat]: Fichier préparé pour Gemini.")
|
287 |
-
except Exception as e:
|
288 |
-
print(f"--- ERREUR [/api/chat]: Échec traitement/upload fichier '{filename}': {e}")
|
289 |
-
if filepath_to_delete and os.path.exists(filepath_to_delete):
|
290 |
-
try: os.remove(filepath_to_delete)
|
291 |
-
except OSError as del_e: print(f" Erreur suppression fichier temp après erreur: {del_e}")
|
292 |
-
return jsonify({'success': False, 'error': f"Erreur traitement fichier: {e}"}), 500
|
293 |
-
else:
|
294 |
-
print(f"--- ERREUR [/api/chat]: Type de fichier non autorisé: {file.filename}")
|
295 |
-
return jsonify({'success': False, 'error': f"Type de fichier non autorisé."}), 400
|
296 |
-
|
297 |
-
# --- Préparation du message utilisateur pour l'historique et Gemini ---
|
298 |
-
raw_user_text = prompt
|
299 |
-
display_user_text = f"[{uploaded_filename}] {prompt}" if uploaded_filename and prompt else (prompt or f"[{uploaded_filename}]")
|
300 |
-
user_history_entry = {
|
301 |
-
'role': 'user',
|
302 |
-
'text': display_user_text,
|
303 |
-
'raw_text': raw_user_text,
|
304 |
-
}
|
305 |
-
|
306 |
-
# Ajout à l'historique de session (vérifier que c'est une liste)
|
307 |
-
if not isinstance(session.get('chat_history'), list):
|
308 |
-
print("--- ERREUR [/api/chat]: 'chat_history' n'est pas une liste! Réinitialisation.")
|
309 |
-
session['chat_history'] = []
|
310 |
-
session['chat_history'].append(user_history_entry)
|
311 |
-
|
312 |
-
# --- Log de l'état de l'historique APRES ajout du message utilisateur ---
|
313 |
-
history_after_user_add = list(session.get('chat_history', [])) # Nouvelle copie
|
314 |
-
print(f"--- DEBUG [/api/chat]: Historique en session APRES ajout user message: {len(history_after_user_add)} messages") # LOG 10
|
315 |
-
|
316 |
-
# --- Sélection du modèle ---
|
317 |
-
selected_model_name = MODEL_PRO if use_advanced else MODEL_FLASH
|
318 |
-
|
319 |
-
# --- Préparation des contenus pour l'appel Gemini ---
|
320 |
-
final_prompt_for_gemini = raw_user_text
|
321 |
-
|
322 |
-
# Gérer le cas où seul un fichier est envoyé
|
323 |
-
if uploaded_file_part and not raw_user_text:
|
324 |
-
raw_user_text = f"Décris le contenu de ce fichier : {uploaded_filename}"
|
325 |
-
final_prompt_for_gemini = raw_user_text
|
326 |
-
print(f" [/api/chat]: Fichier seul détecté, prompt généré: '{final_prompt_for_gemini}'")
|
327 |
-
|
328 |
-
# --- Recherche Web si demandée ---
|
329 |
-
web_search_results = None
|
330 |
-
if use_web_search and final_prompt_for_gemini:
|
331 |
-
print(f"--- LOG [/api/chat]: Activation recherche web pour: '{final_prompt_for_gemini[:60]}...'")
|
332 |
-
web_search_response = perform_web_search_with_gemini(final_prompt_for_gemini, genai_client, selected_model_name)
|
333 |
-
if web_search_response:
|
334 |
-
web_search_results = format_search_response(web_search_response)
|
335 |
-
print(f" [/api/chat]: Résultats de recherche obtenus.")
|
336 |
-
# Avec le nouveau SDK et l'outil de recherche intégré, pas besoin de reformater le prompt
|
337 |
-
# car la recherche est gérée directement par l'API
|
338 |
-
|
339 |
-
# Préparer l'historique des messages PRÉCÉDENTS pour Gemini
|
340 |
-
history_for_gemini_prep = list(session.get('chat_history', []))[:-1] # Sans le dernier message utilisateur
|
341 |
-
gemini_history_to_send = prepare_gemini_history(history_for_gemini_prep)
|
342 |
-
|
343 |
-
# --- Appel à l'API Gemini ---
|
344 |
-
try:
|
345 |
-
print(f"--- DEBUG [/api/chat]: Préparation de l'envoi à l'API Gemini (Modèle: {selected_model_name}) ---") # LOG 11
|
346 |
-
|
347 |
-
# Construire le contenu pour l'appel avec le nouveau SDK
|
348 |
-
contents = []
|
349 |
-
|
350 |
-
# Ajouter l'historique
|
351 |
-
for message in gemini_history_to_send:
|
352 |
-
contents.append({
|
353 |
-
"role": message["role"],
|
354 |
-
"parts": message["parts"]
|
355 |
-
})
|
356 |
-
|
357 |
-
# Ajouter le message actuel de l'utilisateur
|
358 |
-
current_user_parts = []
|
359 |
-
|
360 |
-
# Si un fichier est présent, l'ajouter d'abord
|
361 |
-
if uploaded_file_part:
|
362 |
-
current_user_parts.append(uploaded_file_part)
|
363 |
-
|
364 |
-
# Ajouter le texte du message
|
365 |
-
if final_prompt_for_gemini:
|
366 |
-
current_user_parts.append(final_prompt_for_gemini)
|
367 |
-
|
368 |
-
if current_user_parts:
|
369 |
-
contents.append({
|
370 |
-
"role": "user",
|
371 |
-
"parts": current_user_parts
|
372 |
-
})
|
373 |
-
|
374 |
-
# Log détaillé de ce qui est envoyé
|
375 |
-
print(f" Nombre total de messages pour Gemini: {len(contents)}")
|
376 |
-
for i, content in enumerate(contents):
|
377 |
-
role = content.get("role")
|
378 |
-
parts = content.get("parts", [])
|
379 |
-
parts_info = []
|
380 |
-
|
381 |
-
for part in parts:
|
382 |
-
if isinstance(part, str):
|
383 |
-
parts_info.append(f"Text({len(part)} chars): '{part[:50]}...'")
|
384 |
-
elif isinstance(part, dict) and "inline_data" in part:
|
385 |
-
parts_info.append(f"File(mime={part['inline_data']['mime_type']})")
|
386 |
-
else:
|
387 |
-
parts_info.append(f"Part({type(part)})")
|
388 |
-
|
389 |
-
print(f" Message {i} (role: {role}): {', '.join(parts_info)}")
|
390 |
-
|
391 |
-
# Configuration de l'appel à l'API
|
392 |
-
generate_config = types.GenerateContentConfig(
|
393 |
-
system_instruction=SYSTEM_INSTRUCTION,
|
394 |
-
safety_settings=SAFETY_SETTINGS
|
395 |
-
)
|
396 |
-
|
397 |
-
# Ajouter la configuration de recherche web si demandée
|
398 |
-
if use_web_search:
|
399 |
-
generate_config.tools = [{"google_search": {}}]
|
400 |
-
|
401 |
-
# Appel à l'API avec le nouveau SDK
|
402 |
-
print(f"--- LOG [/api/chat]: Envoi de la requête à {selected_model_name}...")
|
403 |
-
response = genai_client.models.generate_content(
|
404 |
-
model=selected_model_name,
|
405 |
-
contents=contents,
|
406 |
-
config=generate_config
|
407 |
-
)
|
408 |
-
|
409 |
-
# --- Traitement de la réponse (avec logs) ---
|
410 |
-
response_text_raw = ""
|
411 |
-
response_html = ""
|
412 |
-
|
413 |
-
try:
|
414 |
-
# Extraction du texte de la réponse avec le nouveau SDK
|
415 |
-
if hasattr(response, 'text'):
|
416 |
-
response_text_raw = response.text
|
417 |
-
print(f"--- LOG [/api/chat]: Réponse reçue de Gemini (brute, début): '{response_text_raw[:100]}...'")
|
418 |
-
elif hasattr(response, 'parts'):
|
419 |
-
response_text_raw = ' '.join([str(part) for part in response.parts])
|
420 |
-
print(f"--- LOG [/api/chat]: Réponse extraite des parts: '{response_text_raw[:100]}...'")
|
421 |
-
else:
|
422 |
-
# Gestion des erreurs de sécurité ou autres
|
423 |
-
if hasattr(response, 'prompt_feedback'):
|
424 |
-
feedback = response.prompt_feedback
|
425 |
-
if feedback:
|
426 |
-
block_reason = getattr(feedback, 'block_reason', None)
|
427 |
-
if block_reason:
|
428 |
-
response_text_raw = f"Désolé, ma réponse a été bloquée ({block_reason})."
|
429 |
-
else:
|
430 |
-
response_text_raw = "Désolé, je n'ai pas pu générer de réponse (restrictions de sécurité)."
|
431 |
-
else:
|
432 |
-
response_text_raw = "Désolé, je n'ai pas pu générer de réponse."
|
433 |
-
|
434 |
-
print(f" [/api/chat]: Message d'erreur généré: '{response_text_raw}'")
|
435 |
-
|
436 |
-
# Conversion Markdown
|
437 |
-
response_html = markdown.markdown(response_text_raw, extensions=['fenced_code', 'tables', 'nl2br'])
|
438 |
-
if response_html != response_text_raw:
|
439 |
-
print(f" [/api/chat]: Réponse convertie en HTML.")
|
440 |
-
|
441 |
-
except Exception as e_resp:
|
442 |
-
print(f"--- ERREUR [/api/chat]: Erreur lors du traitement de la réponse Gemini : {e_resp}")
|
443 |
-
print(f" Réponse brute: {response}")
|
444 |
-
response_text_raw = f"Désolé, erreur inattendue ({type(e_resp).__name__})."
|
445 |
-
response_html = markdown.markdown(response_text_raw)
|
446 |
-
|
447 |
-
# --- Ajout de la réponse Assistant à l'historique de session ---
|
448 |
-
assistant_history_entry = {
|
449 |
-
'role': 'assistant',
|
450 |
-
'text': response_html,
|
451 |
-
'raw_text': response_text_raw
|
452 |
-
}
|
453 |
-
|
454 |
-
if not isinstance(session.get('chat_history'), list):
|
455 |
-
print("--- ERREUR [/api/chat]: 'chat_history' n'est pas liste avant ajout assistant! Réinitialisation.")
|
456 |
-
session['chat_history'] = [user_history_entry] # Garder au moins user msg
|
457 |
-
|
458 |
-
session['chat_history'].append(assistant_history_entry)
|
459 |
-
|
460 |
-
# --- Log de l'état final de l'historique pour ce tour ---
|
461 |
-
history_final_turn = list(session.get('chat_history', []))
|
462 |
-
print(f"--- DEBUG [/api/chat]: Historique en session FINAL après ajout assistant: {len(history_final_turn)} messages") # LOG 14
|
463 |
-
|
464 |
-
# --- Renvoyer la réponse au frontend ---
|
465 |
-
print(f"--- LOG [/api/chat]: Envoi de la réponse HTML au client.")
|
466 |
-
print(f"---==================================---\n")
|
467 |
-
return jsonify({'success': True, 'message': response_html})
|
468 |
-
|
469 |
-
except Exception as e:
|
470 |
-
print(f"--- ERREUR CRITIQUE [/api/chat]: Échec appel Gemini ou traitement réponse : {e}")
|
471 |
-
# Tentative de retrait du dernier message utilisateur en cas d'erreur
|
472 |
-
current_history = session.get('chat_history')
|
473 |
-
if isinstance(current_history, list) and current_history:
|
474 |
-
try:
|
475 |
-
if current_history[-1].get('role') == 'user':
|
476 |
-
current_history.pop()
|
477 |
-
print(" [/api/chat]: Dernier message user retiré de l'historique suite à l'erreur.")
|
478 |
-
else:
|
479 |
-
print(" [/api/chat]: Dernier message n'était pas 'user', historique non modifié après erreur.")
|
480 |
-
except Exception as pop_e:
|
481 |
-
print(f" Erreur lors tentative retrait message user: {pop_e}")
|
482 |
-
print(f"---==================================---\n")
|
483 |
-
return jsonify({'success': False, 'error': f"Erreur interne: {e}"}), 500
|
484 |
-
|
485 |
-
finally:
|
486 |
-
# --- Nettoyage des fichiers locaux ---
|
487 |
-
if filepath_to_delete and os.path.exists(filepath_to_delete):
|
488 |
-
try:
|
489 |
-
os.remove(filepath_to_delete)
|
490 |
-
print(f"--- LOG [/api/chat FINALLY]: Fichier temporaire local '{filepath_to_delete}' supprimé.")
|
491 |
-
except OSError as e_del_local:
|
492 |
-
print(f"--- ERREUR [/api/chat FINALLY]: Échec suppression fichier local '{filepath_to_delete}': {e_del_local}")
|
493 |
-
|
494 |
-
|
495 |
-
@app.route('/clear', methods=['POST'])
|
496 |
-
def clear_chat():
|
497 |
-
"""Efface l'historique de chat dans la session."""
|
498 |
-
print("\n--- DEBUG [/clear]: Requête POST reçue ---")
|
499 |
-
session.clear() # Efface toutes les données de la session serveur actuelle
|
500 |
-
print(" [/clear]: Session serveur effacée.")
|
501 |
-
is_ajax = 'XMLHttpRequest' == request.headers.get('X-Requested-With') or \
|
502 |
-
'application/json' in request.headers.get('Accept', '')
|
503 |
-
if is_ajax:
|
504 |
-
print(" [/clear]: Réponse JSON (AJAX).")
|
505 |
-
return jsonify({'success': True, 'message': 'Historique effacé.'})
|
506 |
-
else:
|
507 |
-
print(" [/clear]: Réponse Flash + Redirect (non-AJAX).")
|
508 |
-
flash("Conversation effacée.", "info")
|
509 |
-
return redirect(url_for('root'))
|
510 |
-
|
511 |
-
|
512 |
-
# --- Démarrage de l'application ---
|
513 |
-
if __name__ == '__main__':
|
514 |
-
print("--- Démarrage du serveur Flask ---")
|
515 |
-
port = int(os.environ.get('PORT', 5001))
|
516 |
-
# Mettre debug=True pour le développement et voir les logs facilement.
|
517 |
-
# ATTENTION: NE PAS UTILISER debug=True en production !
|
518 |
-
app.run(debug=True, host='0.0.0.0', port=port)
|
519 |
-
|
520 |
-
# --- END OF FILE app.py ---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|