Update app.py

app.py

@@ -39,15 +39,38 @@ def get_air_quality(city: str) -> str:
         str: The AQI value along with air quality status.
     """
     try:
-        search_tool = DuckDuckGoSearchTool()
-        query = f"current AQI in {city}"
-        results = search_tool.forward(query)
+        # Step 1: Search for the city on AQI.in to find the correct country path
+        search_url = f"https://www.aqi.in/search?query={city.replace(' ', '+')}"
+        search_response = requests.get(search_url)
+
+        if search_response.status_code != 200:
+            return f"Failed to fetch AQI data for {city}. Status Code: {search_response.status_code}"
+
+        # Step 2: Extract the correct city URL from search results
+        import re
+        match = re.search(r'href="(/dashboard/[^"]+)"', search_response.text)
+
+        if not match:
+            return f"No AQI data found for {city}."
+
+        city_url_path = match.group(1)
+        city_url = f"https://www.aqi.in{city_url_path}"
 
-        # Print raw search results to debug
-        print(f"Raw DuckDuckGo search results for AQI in {city}:")
-        print(results)
+        # Step 3: Fetch AQI details from the city's AQI page
+        city_response = requests.get(city_url)
+
+        if city_response.status_code != 200:
+            return f"Failed to fetch AQI data for {city}. Status Code: {city_response.status_code}"
+
+        # Step 4: Extract AQI value using regex
+        match = re.search(r"AQI is (\d+)", city_response.text)
+
+        if match:
+            aqi_value = match.group(1)
+            return f"Air Quality in {city}: AQI {aqi_value} ({city_url})"
+        
+        return f"No AQI data found for {city}."
 
-        return f"Raw search results logged for debugging."
     except Exception as e:
         return f"Error fetching AQI: {e}"