Spaces:
Sleeping
Sleeping
| from typing import Any, Optional | |
| from smolagents.tools import Tool | |
| import datetime | |
| import pytz | |
| class GetLocalTimeTool(Tool): | |
| name = "get_local_time" | |
| description = "Gets the current local time at a travel destination." | |
| inputs = {'destination': {'type': 'string', 'description': 'A city or location name (e.g., "Paris", "Tokyo", "New York").'}} | |
| output_type = "string" | |
| def __init__(self): | |
| super().__init__() | |
| try: | |
| import pytz | |
| except ImportError as e: | |
| raise ImportError( | |
| "You must install package `pytz` to run this tool: for instance run `pip install pytz`." | |
| ) from e | |
| # Map of common tourist destinations to their timezones | |
| self.destination_timezones = { | |
| "london": "Europe/London", | |
| "paris": "Europe/Paris", | |
| "rome": "Europe/Rome", | |
| "madrid": "Europe/Madrid", | |
| "berlin": "Europe/Berlin", | |
| "amsterdam": "Europe/Amsterdam", | |
| "athens": "Europe/Athens", | |
| "istanbul": "Europe/Istanbul", | |
| "dubai": "Asia/Dubai", | |
| "new delhi": "Asia/Kolkata", | |
| "mumbai": "Asia/Kolkata", | |
| "bangkok": "Asia/Bangkok", | |
| "singapore": "Asia/Singapore", | |
| "tokyo": "Asia/Tokyo", | |
| "seoul": "Asia/Seoul", | |
| "beijing": "Asia/Shanghai", | |
| "shanghai": "Asia/Shanghai", | |
| "hong kong": "Asia/Hong_Kong", | |
| "sydney": "Australia/Sydney", | |
| "melbourne": "Australia/Melbourne", | |
| "auckland": "Pacific/Auckland", | |
| "fiji": "Pacific/Fiji", | |
| "honolulu": "Pacific/Honolulu", | |
| "anchorage": "America/Anchorage", | |
| "los angeles": "America/Los_Angeles", | |
| "san francisco": "America/Los_Angeles", | |
| "las vegas": "America/Los_Angeles", | |
| "denver": "America/Denver", | |
| "chicago": "America/Chicago", | |
| "houston": "America/Chicago", | |
| "new york": "America/New_York", | |
| "miami": "America/New_York", | |
| "toronto": "America/Toronto", | |
| "mexico city": "America/Mexico_City", | |
| "rio de janeiro": "America/Sao_Paulo", | |
| "sao paulo": "America/Sao_Paulo", | |
| "buenos aires": "America/Argentina/Buenos_Aires", | |
| "cairo": "Africa/Cairo", | |
| "cape town": "Africa/Johannesburg", | |
| "johannesburg": "Africa/Johannesburg", | |
| "nairobi": "Africa/Nairobi" | |
| } | |
| def forward(self, destination: str) -> str: | |
| try: | |
| # Normalize the destination name | |
| normalized_dest = destination.lower().strip() | |
| # Find the closest matching timezone | |
| timezone = None | |
| for city, tz in self.destination_timezones.items(): | |
| if city in normalized_dest or normalized_dest in city: | |
| timezone = tz | |
| break | |
| if not timezone: | |
| # If we don't have a direct match, try to find it through pytz | |
| try: | |
| # Try web search for timezone if available | |
| import importlib | |
| if importlib.util.find_spec("duckduckgo_search"): | |
| from duckduckgo_search import DDGS | |
| ddgs = DDGS() | |
| results = ddgs.text(f"{destination} timezone") | |
| if results: | |
| # Simple heuristic to extract timezone from search results | |
| for result in results: | |
| body = result.get('body', '').lower() | |
| if 'utc' in body or 'gmt' in body: | |
| timezone_pos = body.find('utc') if 'utc' in body else body.find('gmt') | |
| timezone_info = body[timezone_pos:timezone_pos+8] | |
| return f"Based on web search, the timezone in {destination} appears to be around {timezone_info.upper()}. Current time information is not available." | |
| except: | |
| pass | |
| return f"I don't have timezone information for {destination}. Please try a major city nearby." | |
| # Get current time in that timezone | |
| tz = pytz.timezone(timezone) | |
| local_time = datetime.datetime.now(tz) | |
| # Format the result | |
| formatted_time = local_time.strftime("%I:%M %p on %A, %B %d, %Y") | |
| time_diff = local_time.utcoffset().total_seconds() / 3600 | |
| sign = "+" if time_diff >= 0 else "" | |
| return f"The current local time in {destination} is {formatted_time} (UTC{sign}{int(time_diff)})" | |
| except Exception as e: | |
| return f"Error getting local time for {destination}: {str(e)}" | |