Update app.py

app.py

@@ -19,43 +19,38 @@ def my_custom_tool(
     """
     return "What magic will you build ?"
 
+import datetime
+from smolagents import tool
+
 @tool
-def time_difference_tool(timezone1: str, timezone2: str) -> str:
-    """Calculates the time difference between two given timezones.
+def new_year_countdown_tool() -> str:
+    """Calculates the remaining time until the next New Year.
 
-    This tool returns the current time in both timezones and their time difference in hours.
-
-    Args:
-        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').
+    This tool determines how many days, hours, and minutes are left 
+    until January 1st of the next year.
 
     Returns:
-        str: A message displaying the current time in both timezones and their time difference in hours.
+        str: A message displaying the remaining time until the New Year.
     """
     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")
-
-        # Calculate time difference in hours
-        diff_seconds = (datetime.datetime.now(tz1) - datetime.datetime.now(tz2)).total_seconds()
-        diff_hours = abs(diff_seconds / 3600)
+        # Get current date and time
+        now = datetime.datetime.now()
+        
+        # Calculate next New Year's date (January 1st of the next year)
+        next_new_year = datetime.datetime(year=now.year + 1, month=1, day=1, hour=0, minute=0, second=0)
+        
+        # Compute the difference
+        diff = next_new_year - now
+        days = diff.days
+        hours, remainder = divmod(diff.seconds, 3600)
+        minutes, _ = divmod(remainder, 60)
 
         return (
-            f"Current time in {timezone1}: {time1}\n"
-            f"Current time in {timezone2}: {time2}\n"
-            f"Time difference: {diff_hours:.1f} hours"
+            f"Time remaining until New Year {next_new_year.year}:\n"
+            f"{days} days, {hours} hours, {minutes} minutes."
         )
     except Exception as e:
-        return f"Error: Invalid timezone input. Please provide a valid timezone. ({str(e)})"
-
-
-
+        return f"Error calculating New Year countdown: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -96,7 +91,7 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        #time_difference_tool,
+        new_year_countdown_tool,
         get_current_time_in_timezone,
     ],  ## add your tools here (don't remove final answer)
     max_steps=6,