Add tools to have the countries that are in the same time zone

app.py

@@ -9,14 +9,50 @@ 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 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
+def countries_with_same_current_offset(timezone : str, return_country_names : bool = True) -> str:
+    """
+    Return a list of all countries that have the same current UTC offset
+    as the given timezone_name.
+    Args :
+        timezone: A string, e.g. 'America/New_York'
+        return_country_names: If True, return a list of country names;
+                                If False, return ISO country codes.
+    :return: A list of matching country names or ISO country codes.
     """
-    return "What magic will you build ?"
+    # Current UTC time
+    now_utc = datetime.datetime.utcnow().replace(tzinfo=pytz.utc)
+    
+    # Get the current offset of the specified timezone
+    try:
+        ref_tz = pytz.timezone(timezone)
+    except pytz.UnknownTimeZoneError:
+        raise ValueError(f"Unknown timezone: {timezone}")
+    
+    ref_offset = now_utc.astimezone(ref_tz).utcoffset()
+    
+    matching_countries = []
+
+    # Traverse all known countries in pytz
+    for country_code, timezones in pytz.country_timezones.items():
+        for tz_name in timezones:
+            tz = pytz.timezone(tz_name)
+            current_offset = now_utc.astimezone(tz).utcoffset()
+            
+            if current_offset == ref_offset:
+                # We found a matching offset; record this country
+                if return_country_names:
+                    # Get the human-readable name from pytz.country_names
+                    country_name = pytz.country_names.get(country_code, country_code)
+                    matching_countries.append(country_name)
+                else:
+                    matching_countries.append(country_code)
+                # No need to check other timezones for this country
+                break
+
+    return sorted(matching_countries)
+
+
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: