Update app.py

app.py

@@ -9,19 +9,29 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that combines a string and an integer to create a custom message.
+def date_difference(date1: str, date2: str) -> str:
+    """A tool that calculates the difference in days between two dates.
     
     Args:
-        arg1: A string that will be part of the output message.
-        arg2: An integer that will be used to repeat the string.
+        date1: The first date in 'YYYY-MM-DD' format.
+        date2: The second date in 'YYYY-MM-DD' format.
     
     Returns:
-        A string that combines the input arguments in a meaningful way.
+        A string describing the difference in days between the two dates.
     """
-    # Example functionality: Repeat the string `arg1` `arg2` times
-    result = arg1 * arg2
-    return f"Custom tool result: {result}"
+    try:
+        # Convert the input strings to datetime objects
+        date_format = "%Y-%m-%d"
+        date1_obj = datetime.strptime(date1, date_format)
+        date2_obj = datetime.strptime(date2, date_format)
+        
+        # Calculate the difference in days
+        delta = abs((date2_obj - date1_obj).days)
+        
+        # Return the result as a string
+        return f"The difference between {date1} and {date2} is {delta} days."
+    except ValueError as e:
+        return f"Error: Invalid date format. Please use 'YYYY-MM-DD'. Details: {e}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: