Update app.py

app.py

@@ -27,7 +27,7 @@ def calculate_time_difference_to_helsinki(local_time:datetime.datetime)-> str: #
         return f"Error fetching the time difference to Helsinki: {str(e)}"
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> Union[str, datetime.datetime]:
+def get_current_time_in_timezone(timezone: 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').
@@ -36,10 +36,24 @@ def get_current_time_in_timezone(timezone: str) -> Union[str, datetime.datetime]
         # Create timezone object
         tz = pytz.timezone(timezone)
         # Get current time in that 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}", local_time
+        local_time = datetime.datetime.now(tz)
+        return local_time
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return f"Error fetching time for timezone '{timezone}': {str(e)}" 
+@tool
+def get_time_zone_string(local_time: datetime, timezone:str) -> str:
+    """A tool that converts the datetime obejct to a string format for a response to the user. 
+    Args:
+        local_time: A datetime object of the local time zone 
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+    """ 
+    try: 
+        local_time = local_time.strftime("%Y-%m-%d %H:%M:%S") 
+        return f"Local time in {timezone} is {local_time} 
+    except Exception as e: 
+        return f"Error converting '{local_time}' to string: {str(e)}"
+    
+    
 
 
 final_answer = FinalAnswerTool()