Update app.py

app.py

@@ -8,6 +8,7 @@ import json
 import time
 from requests_sse import EventSource
 from Gradio_UI import GradioUI
+from dateutil.relativedelta import relativedelta
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -71,6 +72,47 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+@tool
+def get_time_difference(time1: str, timezone1: str, time2: str, timezone2: str) -> str:
+    """A tool that calculates the time difference between two timestamps in different time zones.
+    Args:
+        time1: First timestamp in "YYYY-MM-DD HH:MM:SS" format.
+        timezone1: Time zone of the first timestamp (e.g., "America/New_York").
+        time2: Second timestamp in "YYYY-MM-DD HH:MM:SS" format.
+        timezone2: Time zone of the second timestamp (e.g., "Asia/Kolkata").
+    Returns:
+        str: Time difference as a human-readable string (e.g., "0 years, 0 months, 3 days, 4 hours, 2 minutes, 15 seconds").
+    """
+    try:
+        # Convert input strings to timezone-aware datetime objects
+        tz1 = pytz.timezone(timezone1)
+        tz2 = pytz.timezone(timezone2)
+
+        dt1 = tz1.localize(datetime.datetime.strptime(time1, "%Y-%m-%d %H:%M:%S"))
+        dt2 = tz2.localize(datetime.datetime.strptime(time2, "%Y-%m-%d %H:%M:%S"))
+
+        # Calculate absolute difference
+        if dt1 > dt2:
+            dt1, dt2 = dt2, dt1  # Ensure dt1 is earlier for proper calculations
+
+        diff = relativedelta(dt2, dt1)
+
+        # Convert difference to human-readable format
+        result = []
+        if diff.years: result.append(f"{diff.years} years")
+        if diff.months: result.append(f"{diff.months} months")
+        if diff.days: result.append(f"{diff.days} days")
+        if diff.hours: result.append(f"{diff.hours} hours")
+        if diff.minutes: result.append(f"{diff.minutes} minutes")
+        if diff.seconds: result.append(f"{diff.seconds} seconds")
+
+        return ", ".join(result) if result else "0 seconds"
+
+    except Exception as e:
+        return f"Error calculating time difference: {str(e)}"
+
+
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder: