Update app.py

app.py

@@ -19,40 +19,42 @@ def my_custom_tool(
     """
     return "What magic will you build ?"
 
-import datetime
-from smolagents import tool
-
 @tool
-def new_year_countdown_tool() -> str:
+def new_year_countdown_tool() -> dict:
     """Calculates the remaining time until the next New Year.
 
     This tool determines how many days, hours, and minutes are left 
     until January 1st of the next year.
 
     Returns:
-        str: A message displaying the remaining time until the New Year.
+        dict: A structured response with the remaining time until New Year.
     """
     try:
-        # 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"Time remaining until New Year {next_new_year.year}:\n"
-            f"{days} days, {hours} hours, {minutes} minutes."
-        )
+
+        return {
+            "days": diff.days,
+            "hours": diff.seconds // 3600,
+            "minutes": (diff.seconds % 3600) // 60
+        }
     except Exception as e:
-        return f"Error calculating New Year countdown: {str(e)}"
+        return {"error": f"Error calculating New Year countdown: {str(e)}"}
 
 @tool
+def extract_days_from_countdown() -> int:
+    """Extracts the number of days remaining until New Year from the structured countdown data.
+
+    Returns:
+        int: The number of days remaining until the next New Year.
+    """
+    remaining_time = new_year_countdown_tool()
+    if "days" in remaining_time:
+        return remaining_time["days"]
+    else:
+        raise ValueError("Failed to retrieve countdown data.")
+@tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
     Args:
@@ -93,6 +95,7 @@ agent = CodeAgent(
         final_answer,
         new_year_countdown_tool,
         get_current_time_in_timezone,
+        extract_days_from_countdown,
     ],  ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,