Update app.py

app.py

@@ -8,17 +8,61 @@ from Gradio_UI import GradioUI
 
 # Custom Tools
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
+@tool
+def convert_time_between_timezones(
+    time_str: str,
+    from_tz: str,
+    to_tz: str,
+    time_format: str = "%Y-%m-%d %H:%M:%S"
+) -> str:
+    """Converts a time between two timezones, handling DST automatically.
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        time_str: The time string to convert (e.g., '2023-12-25 16:00:00')
+        from_tz: Source timezone (e.g., 'Europe/Madrid', 'UTC+2', 'America/Mexico_City')
+        to_tz: Target timezone (e.g., 'America/Sao_Paulo', 'UTC-3')
+        time_format: Format of input/output time strings
     """
     try:
-        tz = pytz.timezone(timezone)
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
+        # Handle UTC offsets (like 'UTC+2')
+        if from_tz.startswith('UTC'):
+            from_tz = f'Etc/GMT{-int(from_tz[3:])}' if '+' in from_tz else f'Etc/GMT{+int(from_tz[3:])}'
+        if to_tz.startswith('UTC'):
+            to_tz = f'Etc/GMT{-int(to_tz[3:])}' if '+' in to_tz else f'Etc/GMT{+int(to_tz[3:])}'
+
+        # Get timezone objects
+        from_zone = pytz.timezone(from_tz)
+        to_zone = pytz.timezone(to_tz)
+        
+        # Parse time (naive) and localize to source timezone
+        naive_time = datetime.datetime.strptime(time_str, time_format)
+        localized_time = from_zone.localize(naive_time)
+        
+        # Convert to target timezone
+        converted_time = localized_time.astimezone(to_zone)
+        
+        # Get current UTC offsets to show the difference
+        from_offset = localized_time.strftime('%z')
+        to_offset = converted_time.strftime('%z')
+        
+        return (f"{localized_time.strftime(time_format)} {from_tz} ({from_offset}) "
+                f"is {converted_time.strftime(time_format)} {to_tz} ({to_offset})")
+    
+    except pytz.UnknownTimeZoneError as e:
+        common_tz = {
+            'Spain': 'Europe/Madrid',
+            'São Paulo': 'America/Sao_Paulo',
+            'SP': 'America/Sao_Paulo',
+            'Paulo, Mexico': 'America/Mexico_City',
+            'CEST': 'Europe/Madrid',  # Central European Summer Time
+            'CET': 'Europe/Paris',    # Central European Time
+            'BRT': 'America/Sao_Paulo'
+        }
+        if e.args[0] in common_tz:
+            return f"Note: Using '{common_tz[e.args[0]]}' for '{e.args[0]}'.\n" + \
+                   convert_time_between_timezones(time_str, common_tz[e.args[0]], to_tz, time_format)
+        return f"Error: Unknown timezone - {str(e)}. Common timezones: Europe/Madrid, America/Sao_Paulo, etc."
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return f"Error converting time: {str(e)}"
 
 @tool
 def weather_lookup(city: str) -> str:
@@ -74,14 +118,7 @@ with open("prompts.yaml", 'r') as stream:
 # Create agent with all tools
 agent = CodeAgent(
     model=model,
-    tools=[
-        final_answer,  # Required
-        duckduckgo_search,
-        image_generation_tool,
-        get_current_time_in_timezone,
-        weather_lookup,
-        unit_converter
-    ],
+    tools=[final_answer, duckduckgo_search, image_generation_tool, convert_time_between_timezones],
     max_steps=6,
     verbosity_level=1,
     grammar=None,