Update app.py

app.py

@@ -4,9 +4,45 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+from organisation_tools import PlanningTool
 from Gradio_UI import GradioUI
 
+@tool
+class PlanningToolWrapper:
+    """A tool for managing and retrieving activities from a planning tool.
+    Args:
+        filename: The name of the JSON file to store the planning data.
+        activity: The activity to add.
+        timestamp: The timestamp for the activity.
+        date: The date to filter activities by.
+        month: The month to filter activities by.
+        year: The year to filter activities by.
+    """
+    def __init__(self, filename: str = "planning_data.json"):
+        self.planning_tool = PlanningTool(filename)
+
+    def add_activity(self, activity: str, timestamp: str) -> str:
+        """Adds an activity to the planning tool."""
+        return self.planning_tool.add_activity(activity, timestamp)
+
+    def get_activities(self) -> str:
+        """Retrieves all activities from the planning tool."""
+        return self.planning_tool.get_activities()
+
+    def get_activities_by_date(self, date: str) -> str:
+        """Retrieves activities by date from the planning tool."""
+        return self.planning_tool.get_activities_by_date(date)
+
+    def get_activities_by_month(self, month: str) -> str:
+        """Retrieves activities by month from the planning tool."""
+        return self.planning_tool.get_activities_by_month(month)
+
+    def get_activities_by_year(self, year: str) -> str:
+        """Retrieves activities by year from the planning tool."""
+        return self.planning_tool.get_activities_by_year(year)
+
+
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def simple_calculator(arg1:int, arg2:int)-> str: #it's import to specify the return type