Update app.py

app.py

@@ -48,21 +48,19 @@ def get_air_quality(city: str) -> str:
 
 # Tool to get local time
 @tool
-def get_local_time(city: str) -> str:
-    """Fetches the current local time for a specified city.
-    
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches the current local time in a specified timezone.
     Args:
-        city: The name of the city to fetch the local time for.
-
-    Returns:
-        str: The current local time formatted as YYYY-MM-DD HH:MM:SS.
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
-        tz = pytz.timezone(city.replace(" ", "_"))  # Handle spaces in city names
+        # Create timezone object
+        tz = pytz.timezone(timezone)
+        # Get current time in that timezone
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"Local time in {city}: {local_time}"
+        return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
-        return f"Error fetching time: {e}"
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 # Tool to fetch top 5 restaurants using DuckDuckGo Search
 @tool
@@ -105,7 +103,7 @@ with open("prompts.yaml", 'r') as stream:
 # Setting up the CodeAgent with the defined tools
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_weather, get_air_quality, get_local_time, get_top_restaurants],
+    tools=[final_answer, get_weather, get_air_quality, get_current_time_in_timezone, get_top_restaurants],
     max_steps=6,
     verbosity_level=1,
     grammar=None,