Update app.py

app.py

@@ -11,17 +11,29 @@ from Gradio_UI import GradioUI
 
 # Belo is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def time_left_for_event(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def time_left_for_event(event:str, date:str)-> str: #it's import to specify the return type
     """A tool that calculates the time left for a certain event, and returns in a very convenient way the time between today and that events date
     Args:
-        arg1: the event name, this can be wedding, birthday, trip, travel, work event, reunion, etc..
-        arg2: the day of that event in calendar time
+        event: the event name, this can be wedding, birthday, trip, travel, work event, reunion, etc..
+        sate: the day of that event in calendar time
+
+    Returns:
+        A string telling the user how many months, weeks or days are left before the event date
     """
 
-    local_time = datetime.datetime.now()
-    future_time = datetime.datetime(arg2)
-    time_between = local_time - future_time
-    return f"you have {time_between} before the {arg1}"
+    try:
+        # Calculate Todays date
+        today = datetime.datetime.now()
+        # calculate the events date difference
+        difference = event_date - today
+        # Convert the difference into months, weeks or days
+        if difference < 7:
+            ans = f"{difference} days"
+        elif difference > 7:
+            ans = f"{difference} weeks"
+        return f"you have the following time: {ans}"
+    except Exception as e:
+        return f"Error: {e}"