Update app.py

app.py

@@ -1,10 +1,12 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from geopy.geocoders import Nominatim
+from timezonefinder import TimezoneFinder
+from tools.final_answer import FinalAnswerTool
 import datetime
 import requests
 import re
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
@@ -31,46 +33,67 @@ def get_stock_price(stock_symbol: str) -> str:
 
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
+def get_current_time_in_location(location: str) -> str:
+    """A tool that fetches the current local time for a specified location (city or zip code).
 
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        location: A string representing a city or zip code (e.g., 'New York', '90210').
 
     Returns:
-        A string displaying the local date and time in the specified timezone.
+        A string displaying the local date and time in the specified location.
     """
     try:
-        # Validate timezone
-        if timezone not in pytz.all_timezones:
-            return f" Error: '{timezone}' is not a valid timezone. Please provide a correct timezone (e.g., 'America/New_York')."
+        # Initialize geolocator and timezone finder
+        geolocator = Nominatim(user_agent="time-zone-fetcher")
+        timezone_finder = TimezoneFinder()
+
+        # Get location coordinates (latitude and longitude)
+        location_obj = geolocator.geocode(location)
+        if not location_obj:
+            return f"Error: Location '{location}' could not be found. Please provide a valid city or zip code."
+
+        latitude, longitude = location_obj.latitude, location_obj.longitude
+
+        # Get the timezone for the location
+        timezone_str = timezone_finder.timezone_at(lng=longitude, lat=latitude)
+        if timezone_str is None:
+            return f"Error: Could not determine the timezone for location '{location}'."
 
         # Get the current time in the specified timezone
-        tz = pytz.timezone(timezone)
+        tz = pytz.timezone(timezone_str)
         local_time = datetime.datetime.now(tz)
 
         # Format the output with day, date, and time
         formatted_time = local_time.strftime("%A, %Y-%m-%d %H:%M:%S %Z")
 
-        return f"The current local time in **{timezone}** is: **{formatted_time}**"
+        return f"The current local time in **{location}** ({timezone_str}) is: **{formatted_time}**"
 
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return f"Error fetching time for location '{location}': {str(e)}"
+
+ 
 
-  
 @tool
-def get_weather(location: str = "auto") -> str:
-    """Fetches the current weather for a specified location without using an API key.
+def get_weather(location: str = None, zipcode: int = None) -> str:
+    """Fetches the current weather for a specified city or zip code without using an API key.
     
     Args:
         location: A string representing a valid city name (e.g., 'New York').
-                  Defaults to 'auto' to detect the location automatically.
+        zipcode: An integer representing a valid zip code (e.g., 10001 for NYC).
     
     Returns:
-        A string describing the weather conditions.
+        A string describing the weather conditions, temperature, and humidity in one line.
     """
     
-    BASE_URL = f"https://wttr.in/{location}?format=%C+%t+%h"
+    # Determine the correct location (city or zip code)
+    if location:
+        query = location
+    elif zipcode:
+        query = str(zipcode)
+    else:
+        return "Error: Please provide either a city name or zip code."
+    
+    BASE_URL = f"https://wttr.in/{query}?format=%C|%t|%h"
     
     try:
         response = requests.get(BASE_URL)
@@ -78,8 +101,12 @@ def get_weather(location: str = "auto") -> str:
         
         # Split the response into separate parts
         weather_data = response.text.strip().split("|")
-                        
-        return f"The current Weather in {location} is: {weather_data[0]}, Temperature: {weather_data[1]}, Humidity: {weather_data[2]}"
+        
+        if len(weather_data) != 3:
+            return "Error: Unexpected response format"
+        
+        # Return the weather data in one line, comma separated
+        return f"The current Weather in {query}is: {weather_data[0]}, Temperature: {weather_data[1]}, Humidity: {weather_data[2]}"
     
     except requests.exceptions.RequestException as e:
         return f"Error fetching weather data: {str(e)}"