Update app.py

app.py

@@ -7,25 +7,28 @@ 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 get_nairobi_weather() -> str:
-    """Fetches and returns current weather data for Nairobi"""
+def get_nairobi_weather(units: Optional[str] = "metric") -> str:
+    """A tool that fetches current weather data for Nairobi, Kenya.
+    Args:
+        units: (Optional) Temperature units - 'metric' (Celsius), 'imperial' (Fahrenheit), or 'standard' (Kelvin). Defaults to 'metric'.
+    """
     # Nairobi coordinates
     lat, lon = -1.286389, 36.817223
     api_key = OPEN_WEATHER_TOKEN  # Replace with your own key
     
     try:
-        # Fetch weather data
-        url = f"https://api.openweathermap.org/data/2.5/weather?lat={lat}&lon={lon}&appid={api_key}&units=metric"
+        url = f"https://api.openweathermap.org/data/2.5/weather?lat={lat}&lon={lon}&appid={api_key}&units={units}"
         response = requests.get(url)
         response.raise_for_status()
         data = response.json()
         
-        # Format the weather report
+        # Get current time in Nairobi
+        nairobi_time = datetime.datetime.now(pytz.timezone('Africa/Nairobi')).strftime("%Y-%m-%d %H:%M:%S")
+        
         report = (
-            f"🌍 Location: Nairobi, Kenya\n"
-            f"🌡️ Temperature: {data['main']['temp']}°C (Feels like {data['main']['feels_like']}°C)\n"
+            f"🌤️ Current Weather in Nairobi, Kenya ({nairobi_time}):\n"
+            f"🌡️ Temperature: {data['main']['temp']}°{'C' if units == 'metric' else 'F' if units == 'imperial' else 'K'}\n"
             f"☁️ Conditions: {data['weather'][0]['description'].title()}\n"
             f"💧 Humidity: {data['main']['humidity']}%\n"
             f"🌬️ Wind: {data['wind']['speed']} m/s\n"
@@ -34,53 +37,20 @@ def get_nairobi_weather() -> str:
         return report
         
     except requests.exceptions.RequestException as e:
-        return f"⚠️ Error fetching data: {str(e)}"
+        return f"⚠️ Error fetching weather data: {str(e)}"
     except KeyError:
         return "⚠️ Error processing weather data"
 
-# Create the interface
-with gr.Blocks(title="Nairobi Weather") as demo:
-    gr.Markdown("# 🌤️ Current Weather in Nairobi")
-    with gr.Row():
-        weather_output = gr.Textbox(label="Weather Report", interactive=False)
-        refresh_btn = gr.Button("Refresh Data")
-    refresh_btn.click(fn=get_nairobi_weather, outputs=weather_output)
-    
-    # Display initial data
-    demo.load(fn=get_nairobi_weather, outputs=weather_output)
-
-demo.launch()
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'Kenya/Nairobi').
-    """
-    try:
-        # 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")
-        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)}"
-
-
+# Your existing setup
 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 = 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',
+    custom_role_conversions=None,
 )
 
-
-# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
@@ -88,7 +58,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, get_current_time_in_timezone, get_nairobi_weather, image_generation_tool],  # Added weather tool
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -98,5 +68,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()