Update main.py

main.py

@@ -584,38 +584,52 @@ async def chat_bot(input_data: ChatInput):
         intent = detect_intent(query)
         k_data = search_knowledge(query)  
         s_data = search_stats(query)
-        print(f"--- DEBUG STATS POUR {query} ---")
-        print(f"Contenu envoyé à l'IA : '{s_data}'")
-        print("---------------------------------")
+        s_internet = ""
+        if intent["needs_search"]:
+            print(f"--- ACTIVATION RECHERCHE WEB POUR : {query} ---")
+            s_internet = search_internet(query) or "Aucun résultat web récent trouvé."
+
         prediction_ctx = ""
-        pred = None
-        if intent["needs_prediction"] and intent["city"]:
-            pred = predict_pm25_for_city(intent["city"], days=7)
-            if pred:
+        pred_data = None
+        target_city = intent["city"]
+        
+        if target_city:
+            pred_results = predict_pm25_for_city(target_city, days=7)
+            if pred_results:
+                pred_data = pred_results
                 lines = []
-                for d in pred['days'][:7]: 
+                for d in pred_results['days']:
                     lines.append(
-                        f"- {d['date']} : {d['temp_max']}°C, {d['condition']}, "
-                        f"Pluie: {d['precip']}mm, Vent: {d['vent']}km/h, Humidité: {d['humidite']}%, "
+                        f"- {d['date']} ({target_city}): {d['temp_max']}°C, {d['condition']}, "
                         f"PM2.5 prévu: {d['pm25']} ({d['niveau']})"
                     )
-                    prediction_ctx = "\n".join(lines)
+                prediction_ctx = "\n".join(lines)
             else:
-                prediction_ctx = "Désolé, les prévisions sur 7 jours ne sont pas disponibles pour cette localité."
+                prediction_ctx = f"Données de prévision indisponibles pour {target_city}."
         prompt = build_prompt(
             user_query=query, 
             stats_ctx=s_data,      
             knowledge_ctx=k_data, 
             prediction_ctx=prediction_ctx, 
-            search_ctx=""
+            search_ctx=s_internet # Le bloc Web est maintenant rempli !
         )
-
         response = call_google(prompt) or call_openrouter(prompt)
-        return {"response": response.strip(), "prediction": pred}
+        
+        if not response:
+            raise Exception("Les services de génération IA sont indisponibles.")
+        return {
+            "response": response.strip(), 
+            "prediction": pred_data, 
+            "detected_city": target_city,
+            "source_web": True if s_internet else False
+        }
 
     except Exception as e:
-        print(f"Erreur : {e}") # C'est ici que tu voyais l'erreur
-        return {"response": "Désolé, j'ai un petit souci technique. ✨"}
+        print(f"ERREUR CRITIQUE CHATBOT: {str(e)}")
+        return {
+            "response": "Désolé, je rencontre une difficulté technique pour analyser cette demande. ✨",
+            "prediction": None
+        }
 @app.get("/api/weather-at-coords")
 async def get_weather_at_coords(lat: float, lon: float):
     prediction = predict_pm25_for_city(city_name="", days=1, user_lat=lat, user_lon=lon)