Update app.py

app.py

@@ -10,7 +10,7 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def calculate_time_difference_to_helsinki(local_time:datetime)-> str: #it's import to specify the return type
+def calculate_time_difference_to_helsinki(local_time:datetime.datetime)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that calculates the time difference in hours of current local time in a  specified timezone to Helsinki
     Args:
@@ -27,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) -> Union[str, datetime]:
+def get_current_time_in_timezone(timezone: str) -> Union[str, datetime.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').