Update app.py

app.py

@@ -1,4 +1,3 @@
-#+€+
 import os
 import json
 import mimetypes
@@ -48,8 +47,8 @@ print(f"Dossier pour les sessions serveur configuré : {app.config['SESSION_FILE
 server_session = Session(app)
 
 # --- Configuration de l'API Gemini ---
-MODEL_FLASH = 'gemini-2.0-flash'
-MODEL_PRO = 'gemini-2.5-pro-exp-03-25'
+MODEL_FLASH = 'gemini-1.5-flash-002'  # Mise à jour des modèles
+MODEL_PRO = 'gemini-1.5-pro-002'  # Mise à jour des modèles
 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."
 SAFETY_SETTINGS = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
@@ -247,16 +246,8 @@ def chat_api():
              current_gemini_parts.append(final_prompt_for_gemini)
 
     # --- Préparation des paramètres de configuration et modèle ---
-    config = {}
     selected_model_name = MODEL_PRO if use_advanced else MODEL_FLASH
     
-    # --- Intégration Google Search ---
-    if use_web_search and final_prompt_for_gemini:
-        print(f"--- LOG [/api/chat]: Activation Google Search pour: '{final_prompt_for_gemini[:60]}...'")
-        # Configurer Google Search comme outil
-        config["tools"] = [{"google_search": {}}]
-        print(f"  [/api/chat]: Configuration Google Search ajoutée")
-
     # Vérifier si on a quelque chose à envoyer
     if not current_gemini_parts:
         print("--- ERREUR [/api/chat]: Aucune donnée (texte ou fichier valide) à envoyer après traitement.")
@@ -299,10 +290,18 @@ def chat_api():
             system_instruction=SYSTEM_INSTRUCTION
         )
         print(f"--- LOG [/api/chat]: Envoi de la requête à {selected_model_name}...")
-        response = active_model.generate_content(
-            contents=contents_for_gemini,
-            config=config
-        )
+        
+        # Modification pour utiliser google_search_retrieval si la recherche web est activée
+        if use_web_search:
+            print(f"--- LOG [/api/chat]: Activation Google Search Retrieval...")
+            response = active_model.generate_content(
+                contents=contents_for_gemini,
+                tools='google_search_retrieval'
+            )
+        else:
+            response = active_model.generate_content(
+                contents=contents_for_gemini
+            )
 
         # --- Traitement de la réponse (avec logs) ---
         response_text_raw = ""
@@ -317,34 +316,25 @@ def chat_api():
                 # Récupération des métadonnées de recherche si disponibles
                 if use_web_search and hasattr(response, 'candidates') and response.candidates:
                     candidate = response.candidates[0]
-                    if hasattr(candidate, 'grounding_metadata'):
-                        metadata = candidate.grounding_metadata
+                    if hasattr(candidate, 'citation_metadata'):
+                        metadata = candidate.citation_metadata
                         search_metadata = {}
                         
-                        # Requêtes de recherche
-                        if hasattr(metadata, 'web_search_queries'):
-                            search_metadata["search_queries"] = metadata.web_search_queries
-                        
-                        # Pages de recherche utilisées
-                        if hasattr(metadata, 'grounding_chunks'):
-                            search_pages = [
-                                {
-                                    "title": site.web.title if hasattr(site, 'web') and hasattr(site.web, 'title') else "Sans titre",
-                                    "url": site.web.url if hasattr(site, 'web') and hasattr(site.web, 'url') else "#",
-                                    "snippet": site.web.snippet if hasattr(site, 'web') and hasattr(site.web, 'snippet') else ""
-                                }
-                                for site in metadata.grounding_chunks if hasattr(site, 'web')
-                            ]
-                            search_metadata["search_pages"] = search_pages
-                        
-                        # Contenu HTML rendu
-                        if hasattr(metadata, 'search_entry_point') and hasattr(metadata.search_entry_point, 'rendered_content'):
-                            search_metadata["rendered_content"] = metadata.search_entry_point.rendered_content
+                        # Extraire les citations et références
+                        if hasattr(metadata, 'citations'):
+                            search_pages = []
+                            for citation in metadata.citations:
+                                if hasattr(citation, 'start_index') and hasattr(citation, 'end_index'):
+                                    search_pages.append({
+                                        "title": citation.title if hasattr(citation, 'title') else "Sans titre",
+                                        "url": citation.uri if hasattr(citation, 'uri') else "#",
+                                        "snippet": f"Citation: {citation.start_index}-{citation.end_index}"
+                                    })
+                            if search_pages:
+                                search_metadata["search_pages"] = search_pages
                         
-                        print(f"--- LOG [/api/chat]: Métadonnées de recherche Google récupérées:")
-                        if search_metadata.get("search_queries"):
-                            print(f"  Requêtes: {search_metadata['search_queries']}")
-                        if search_metadata.get("search_pages"):
+                        print(f"--- LOG [/api/chat]: Métadonnées de citation récupérées:")
+                        if search_metadata and search_metadata.get("search_pages"):
                             print(f"  Pages: {len(search_metadata['search_pages'])} sources")
             else:
                 feedback_info = f"Feedback: {response.prompt_feedback}" if response.prompt_feedback else "Pas de feedback détaillé."