Update app.py

app.py

@@ -1,21 +1,12 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
-import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Example custom tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that does nothing yet.
-    Args:
-        arg1: The first argument.
-        arg2: The second argument.
-    """
-    return "What magic will you build?"
-
-# Tool to get current time in a specified timezone
+# Define the tool correctly using @tool decorator
+@tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """Fetches the current local time in a specified timezone.
     Args:
@@ -46,7 +37,7 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the Agent with the defined tools
+# Create the Agent with properly defined tools
 agent = CodeAgent(
     model=model,
     tools=[final_answer, DuckDuckGoSearchTool(), get_current_time_in_timezone, image_generation_tool],  
@@ -61,3 +52,4 @@ agent = CodeAgent(
 
 # Launch UI
 GradioUI(agent).launch()
+