Update app.py

app.py

@@ -21,45 +21,44 @@ def my_custom_tool(
     """
     return "What magic will you build ?"
 
-
-import datetime
-import pytz
-from smolagents import tool
-
 @tool
 def time_difference_tool(timezone1: str, timezone2: str) -> str:
     """Calculates the time difference between two given timezones.
 
-    This tool returns the current time in both timezones and their time difference.
+    This tool returns the current time in both timezones and their time difference in hours.
 
     Args:
-        timezone1 (str): The first timezone (e.g., 'America/New_York').
-        timezone2 (str): The second timezone (e.g., 'Europe/Istanbul').
+        timezone1 (str): The name of the first timezone (e.g., 'America/New_York').
+        timezone2 (str): The name of the second timezone (e.g., 'Europe/Istanbul').
 
     Returns:
-        str: A message displaying the current time in both timezones and their time difference.
+        str: A message displaying the current time in both timezones and their time difference in hours.
     """
     try:
+        # Load timezone objects
         tz1 = pytz.timezone(timezone1)
         tz2 = pytz.timezone(timezone2)
 
+        # Get current times in both timezones
         now = datetime.datetime.utcnow()
-
         time1 = datetime.datetime.now(tz1).strftime("%Y-%m-%d %H:%M:%S")
         time2 = datetime.datetime.now(tz2).strftime("%Y-%m-%d %H:%M:%S")
 
-        diff_hours = (datetime.datetime.now(tz1) - datetime.datetime.now(tz2)).total_seconds() / 3600
+        # Calculate time difference in hours
+        diff_seconds = (datetime.datetime.now(tz1) - datetime.datetime.now(tz2)).total_seconds()
+        diff_hours = abs(diff_seconds / 3600)
 
         return (
             f"Current time in {timezone1}: {time1}\n"
             f"Current time in {timezone2}: {time2}\n"
-            f"Time difference: {abs(diff_hours)} hours"
+            f"Time difference: {diff_hours:.1f} hours"
         )
     except Exception as e:
         return f"Error: Invalid timezone input. Please provide a valid timezone. ({str(e)})"
 
 
 
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.