Update app.py

app.py

@@ -9,33 +9,37 @@ from Gradio_UI import GradioUI
 # --- FERRAMENTAS PERSONALIZADAS ---
 
 @tool
-def get_time_diff_to_rio(timezone: str) -> str:
-    """Calcula a diferença de horas entre uma timezone específica e o Rio de Janeiro (Brasil).
+def get_time_diff_between_timezones(timezone_a: str, timezone_b: str) -> str:
+    """Calcula a diferença de horas entre duas timezones quaisquer.
     Args:
-        timezone: Uma string de timezone válida (ex: 'Europe/Lisbon', 'Europe/London').
+        timezone_a: A primeira timezone (ex: 'Europe/Lisbon').
+        timezone_b: A segunda timezone para comparar (ex: 'Asia/Tokyo').
     """
     try:
         # Definir as timezones
-        target_tz = pytz.timezone(timezone)
-        rio_tz = pytz.timezone('America/Sao_Paulo') # Rio de Janeiro usa este standard
+        tz_a = pytz.timezone(timezone_a)
+        tz_b = pytz.timezone(timezone_b)
         
-        # Obter a data atual para garantir que o horário de verão é considerado
+        # Obter a data atual
         now = datetime.datetime.now()
         
-        # Calcular o offset em relação ao UTC (em horas)
-        target_offset = target_tz.localize(now).utcoffset().total_seconds() / 3600
-        rio_offset = rio_tz.localize(now).utcoffset().total_seconds() / 3600
+        # Calcular os offsets em relação ao UTC
+        offset_a = tz_a.localize(now).utcoffset().total_seconds() / 3600
+        offset_b = tz_b.localize(now).utcoffset().total_seconds() / 3600
         
-        diff = target_offset - rio_offset
+        diff = offset_a - offset_b
         
         if diff == 0:
-            return f"Não há diferença horária entre {timezone} e o Rio de Janeiro."
+            return f"Não há diferença horária entre {timezone_a} e {timezone_b}."
         
-        sentido = "à frente" if diff > 0 else "atrás"
-        return f"A zona {timezone} está {abs(diff)} horas {sentido} em relação ao Rio de Janeiro."
+        # Lógica para explicar a diferença de forma humana
+        if diff > 0:
+            return f"{timezone_a} está {abs(diff)} horas à frente de {timezone_b}."
+        else:
+            return f"{timezone_a} está {abs(diff)} horas atrás de {timezone_b}."
 
     except Exception as e:
-        return f"Erro ao calcular a diferença: {str(e)}"
+        return f"Erro ao calcular a diferença entre {timezone_a} e {timezone_b}: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -77,7 +81,7 @@ agent = CodeAgent(
     tools=[
         final_answer, 
         search_tool,      
-        get_football_info, 
+        get_time_diff_between_timezones, # Substitui a antiga por esta
         get_current_time_in_timezone
     ], 
     max_steps=10,