Added some tools

app.py

@@ -1,22 +1,118 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-import datetime
 import requests
 import pytz
 import yaml
+from datetime import datetime, timedelta
 from tools.final_answer import FinalAnswerTool
 
 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: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def get_time_difference(location: str) -> str:
+    """
+    Calculates and returns the time difference between a specified location and U.S. Eastern Time.
+
+    This tool fetches the current time for the given location and for the U.S. Eastern Time zone (America/New_York),
+    then presents a fun, easy-to-read comparison.
+
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        location: The city or timezone to get the local time for (e.g., "London", "Asia/Tokyo", "Europe/Paris").
     """
-    return "What magic will you build ?"
+    
+    def get_time_data(timezone: str):
+        """Helper function to fetch time data from the World Time API."""
+        api_url = f"http://worldtimeapi.org/api/timezone/{timezone}"
+        try:
+            response = requests.get(api_url)
+            # Raise an exception for bad status codes (4xx or 5xx)
+            response.raise_for_status()
+            return response.json()
+        except requests.exceptions.RequestException as e:
+            # This handles network errors, bad status codes, etc.
+            print(f"API request failed for {timezone}: {e}")
+            return None
+
+    # Step 1: Get U.S. Eastern Time
+    et_data = get_time_data("America/New_York")
+    if not et_data:
+        return "Oops! I couldn't connect to get the U.S. Eastern Time right now. Please try again later."
+
+    # Step 2: Get time for the user-specified location
+    # The API can be picky. We'll try the user's input directly first.
+    # If it fails, we'll try to find a matching timezone.
+    location_data = get_time_data(location)
+    
+    if not location_data:
+        # Try to find a timezone that contains the location string
+        try:
+            all_timezones_response = requests.get("http://worldtimeapi.org/api/timezone")
+            all_timezones_response.raise_for_status()
+            all_timezones = all_timezones_response.json()
+            
+            # Find the first timezone that ends with the location name
+            found_timezone = next((tz for tz in all_timezones if tz.lower().endswith(f"/{location.lower()}")), None)
+
+            if found_timezone:
+                print(f"Found matching timezone: {found_timezone}")
+                location_data = get_time_data(found_timezone)
+            else:
+                 return f"Sorry, I couldn't find a timezone for '{location}'. Try a major city or a standard timezone format like 'Europe/London'."
+        except requests.exceptions.RequestException:
+             return "Sorry, I'm having trouble searching for that location right now."
+
+
+    if not location_data:
+        return f"I tried my best, but I still couldn't find the time for '{location}'. Are you sure it's spelled correctly?"
+
+    # Step 3: Parse the datetime objects from the API responses
+    try:
+        et_time = datetime.fromisoformat(et_data['datetime'])
+        location_time = datetime.fromisoformat(location_data['datetime'])
+        location_timezone_name = location.replace('_', ' ').title()
+        et_timezone_name = "U.S. Eastern Time"
+    except (ValueError, KeyError):
+        return "There was an issue processing the time data. It might be a temporary problem."
+
+    # Step 4: Calculate the difference
+    # We compare based on UTC offset to get the pure time difference
+    et_offset_str = et_data['utc_offset']
+    location_offset_str = location_data['utc_offset']
+    
+    et_offset = timedelta(hours=int(et_offset_str[1:3]), minutes=int(et_offset_str[4:6]))
+    if et_offset_str[0] == '-':
+        et_offset = -et_offset
+
+    location_offset = timedelta(hours=int(location_offset_str[1:3]), minutes=int(location_offset_str[4:6]))
+    if location_offset_str[0] == '-':
+        location_offset = -location_offset
+
+    time_difference = location_offset - et_offset
+    
+    # Step 5: Format the output string in a fun way
+    diff_total_seconds = time_difference.total_seconds()
+    diff_hours = int(diff_total_seconds / 3600)
+    diff_minutes = int((diff_total_seconds % 3600) / 60)
+
+    comparison = ""
+    if diff_total_seconds == 0:
+        comparison = f"{location_timezone_name} is in the same timezone as {et_timezone_name}!"
+    else:
+        ahead_or_behind = "ahead of" if diff_total_seconds > 0 else "behind"
+        # Format the hours and minutes for readability
+        hour_str = f"{abs(diff_hours)} hour{'s' if abs(diff_hours) != 1 else ''}"
+        minute_str = f"{abs(diff_minutes)} minute{'s' if abs(diff_minutes) != 1 else ''}"
+        
+        if diff_hours != 0 and diff_minutes != 0:
+            comparison = f"{location_timezone_name} is {hour_str} and {minute_str} {ahead_or_behind} {et_timezone_name}."
+        elif diff_hours != 0:
+            comparison = f"{location_timezone_name} is {hour_str} {ahead_or_behind} {et_timezone_name}."
+        else: # Only minutes are different
+            comparison = f"{location_timezone_name} is {minute_str} {ahead_or_behind} {et_timezone_name}."
+
+    return (f"Time check! Right now, it's {et_time.strftime('%-I:%M %p')} in {et_timezone_name}, "
+            f"which means it's {location_time.strftime('%-I:%M %p')} in {location_timezone_name}. "
+            f"{comparison}")
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -28,7 +124,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         # Create timezone object
         tz = pytz.timezone(timezone)
         # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        local_time = datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
@@ -55,7 +151,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, DuckDuckGoSearchTool, image_generation_tool, get_time_difference], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,