Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from typing import Union
 
 from Gradio_UI import GradioUI
 
@@ -26,7 +27,7 @@ def calculate_time_difference_to_helsinki(local_time:datetime)-> str: #it's impo
         return f"Error fetching the time difference to Helsinki: {str(e)}"
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
+def get_current_time_in_timezone(timezone: str) -> Union[str, datetime]:
     """A tool that fetches the current local time in a specified timezone.
     Args:
         timezone: A string representing a valid timezone (e.g., 'America/New_York').