code refactor

app.py

@@ -4,6 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.web_search import WebSearchTool
+import utils
 
 from Gradio_UI import GradioUI
 
@@ -50,39 +52,6 @@ def get_geo_coordinates(city):
     else:
         print("Error on fetching geo coordinates.")
 
-def get_weather_description(code):
-    weather_codes = {        
-        0: "Clear sky",
-        1: "Mainly clear",
-        2: "Partly cloudy",
-        3: "Overcast",
-        45: "Fog and depositing rime fog",
-        48: "Fog and depositing rime fog",
-        51: "Drizzle: Light intensity",
-        53: "Drizzle: Moderate intensity",
-        55: "Drizzle: Dense intensity",
-        56: "Freezing Drizzle: Light intensity",
-        57: "Freezing Drizzle: Dense intensity",
-        61: "Rain: Slight intensity",
-        63: "Rain: Moderate intensity",
-        65: "Rain: Heavy intensity",
-        66: "Freezing Rain: Light intensity",
-        67: "Freezing Rain: Heavy intensity",
-        71: "Snow fall: Slight intensity",
-        73: "Snow fall: Moderate intensity",
-        75: "Snow fall: Heavy intensity",
-        77: "Snow grains",
-        80: "Rain showers: Slight intensity",
-        81: "Rain showers: Moderate intensity",
-        82: "Rain showers: Violent intensity",
-        85: "Snow showers: Slight intensity",
-        86: "Snow showers: Heavy intensity",
-        95: "Thunderstorm: Slight or moderate",
-        96: "Thunderstorm with slight hail",
-        99: "Thunderstorm with heavy hail"
-    }
-    return weather_codes.get(code, "Code not found")
-    
 @tool
 def get_weather(city: str) -> str:
     """A tool that fetches the current weather information of a specified city.
@@ -98,7 +67,7 @@ def get_weather(city: str) -> str:
         temperature = weather_data['current_weather']['temperature']
         windspeed = weather_data['current_weather']['windspeed']
         wind_unit = weather_data['current_weather_units']['windspeed']
-        weather_description = get_weather_description(weather_data['current_weather']['weathercode'])
+        weather_description = utils.get_weather_description(weather_data['current_weather']['weathercode'])
         return f"This is the weather on {city}. Temperature on {city} is {temperature}{unit}, the windspeed is {windspeed}{wind_unit} and the condition is {weather_description}"
     else:
         return "Error on fetching weather data."
@@ -106,7 +75,7 @@ def get_weather(city: str) -> str:
     
         
 final_answer = FinalAnswerTool()
-
+web_search = WebSearchTool()
 # 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' 
 
@@ -126,7 +95,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_weather], ## add your tools here (don't remove final answer)
+    tools=[final_answer, web_search, get_current_time_in_timezone, get_weather], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -136,5 +105,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()

utils.py

@@ -0,0 +1,32 @@
+def get_weather_description(code):
+    weather_codes = {        
+        0: "Clear sky",
+        1: "Mainly clear",
+        2: "Partly cloudy",
+        3: "Overcast",
+        45: "Fog and depositing rime fog",
+        48: "Fog and depositing rime fog",
+        51: "Drizzle: Light intensity",
+        53: "Drizzle: Moderate intensity",
+        55: "Drizzle: Dense intensity",
+        56: "Freezing Drizzle: Light intensity",
+        57: "Freezing Drizzle: Dense intensity",
+        61: "Rain: Slight intensity",
+        63: "Rain: Moderate intensity",
+        65: "Rain: Heavy intensity",
+        66: "Freezing Rain: Light intensity",
+        67: "Freezing Rain: Heavy intensity",
+        71: "Snow fall: Slight intensity",
+        73: "Snow fall: Moderate intensity",
+        75: "Snow fall: Heavy intensity",
+        77: "Snow grains",
+        80: "Rain showers: Slight intensity",
+        81: "Rain showers: Moderate intensity",
+        82: "Rain showers: Violent intensity",
+        85: "Snow showers: Slight intensity",
+        86: "Snow showers: Heavy intensity",
+        95: "Thunderstorm: Slight or moderate",
+        96: "Thunderstorm with slight hail",
+        99: "Thunderstorm with heavy hail"
+    }
+    return weather_codes.get(code, "Code not found")