Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+import re
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
@@ -39,12 +40,14 @@ def get_air_quality(city: str) -> str:
     """
     try:
         search_tool = DuckDuckGoSearchTool()
-        query = f"current AQI in {city} site:aqicn.org"
+        query = f"current AQI in {city}"
         results = search_tool.forward(query)
 
-        if results:
-            top_result = results.split("\n\n")[0]  # Get the most relevant result
-            return f"AQI in {city}: {top_result}"
+        # Use regex to find numbers likely to be AQI values
+        aqi_match = re.search(r"AQI\s*(\d{1,3})", results)
+
+        if aqi_match:
+            return f"AQI in {city}: {aqi_match.group(1)}"
         return f"No AQI data found for {city}."
     except Exception as e:
         return f"Error fetching AQI: {e}"