Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, WebSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, WebSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -16,87 +16,71 @@ 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
-"""
-# return 
+# def my_custom_tool(arg1:str, arg2:int): str #it's important 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
+#     """
+#     # return
 
 @tool
-def get_location_tool(ip: str): str
+def get_location_tool(ip: str):
+    response = DbIpCity.get(ip, api_key='free')
 
-return {
-            'city': response.city,
-            'region': response.region,
-            'country': response.country,
-            'latitude': response.latitude,
-            'longitude': response.longitude
-        }
+    return {
+        'city': response.city,
+        'region': response.region,
+        'country': response.country,
+        'latitude': response.latitude,
+        'longitude': response.longitude
+    }
 
-"""This is a tool that gets the exact location of a user including city, region, country, latitude, and longitude.
+    """
+    This is a tool that gets the exact location of a user including city, region, country, latitude, and longitude.
     Args: 
         ip: A string representing a valid ip address (e.g., '192.168.10.1')
-
-    Args:
-        city: A string representing a valid city (e.g., 'Dubai, Singapore')
-
-    Args:
-        region: A string representing a valid region (e.g, 'Africa, Asia')
-
-    Args:
-        country: A string representing a valid country (e.g., 'Ethiopia, United Arab Emirates')
-
-    Args:
-        latitude: A string representing valid latitude coordinates (e.g., '40.741895')
-
-    Args:
-        longitude: A string representing valid longitude coordinates (e.g, '73.989308')
-
+    
     Returns: ip address, city, region, country, and geo coordinates
-"""
+    """
         
-response = DbIpCity.get(ip, api_key='free')
-
-
-ip_address = 'YOUR_IP_ADDRESS'  # Replace with the actual IP address
-location = get_location(ip_address)
+response = DbIpCity.get('YOUR_IP_ADDRESS')  # Replace with the actual IP address
+location = get_location('ip_address')
 print(location)
 
 
 @tool
-def get_current_time_in_timezone(timezone: str): str
+def get_current_time_in_timezone(timezone: str):
     
-"""A tool that fetches the current local time in a specified timezone.
+    """A tool that fetches the current local time in a specified timezone.
     Args:
         timezone: A string representing a valid timezone (e.g., 'America/New_York')
-
+    
     Returns: The current local time in the following format ("%Y-%m-%d %H:%M:%S")
-"""
-try:
+    """
+    try:
         # Create timezone object
         tz = pytz.timezone(timezone)
-        
-    # Get current time in that timezone
+    
+        # Get current time in that timezone
         local_time = datetime.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)}"
+    except Exception as e:
+        return f"Error fetching time for timezone '{timezone}': {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:
-# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
+# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
 
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct', # it is possible that this model may be overloaded
+    custom_role_conversions=None,
 )
 
 @tool
@@ -110,17 +94,23 @@ def get_weather_data(city: str):
         A dictionary containing weather data
     """
     # Get weather for multiple cities
-cities = ["Tokyo", "New York", "London"]
-data = {}
+    cities = ["Tokyo", "New York", "London"]
+    data = {}
+
+    for city in cities:
+        data[city] = get_weather_data(city)
+    
+    # Calculate average temperatures
+    avg_temps = {city: sum(info["temperatures"])/len(info["temperatures"]) 
+                for city, info in data.items()}
+    
+    # Find the warmest city
+    warmest_city = max(avg_temps, key=avg_temps.get)
+    print(f"The warmest city is {warmest_city} with an average of {avg_temps[warmest_city]:.1f}°C")  
+
+ 
+
 
-for city in cities:
-    data[city] = get_weather_data(city)
-# Calculate average temperatures
-avg_temps = {city: sum(info["temperatures"])/len(info["temperatures"]) 
-             for city, info in data.items()}
-# Find the warmest city
-warmest_city = max(avg_temps, key=avg_temps.get)
-print(f"The warmest city is {warmest_city} with an average of {avg_temps[warmest_city]:.1f}°C")
 
 @tool
 def hotel_recommendation_tool(city: str,price: str): str