Update app.py

app.py

@@ -5,39 +5,34 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Example custom tool (modify or expand as you wish)
 @tool
-:contentReference[oaicite:1]{index=1}
-    :contentReference[oaicite:2]{index=2}
+def my_custom_tool(arg1: str, arg2: int) -> str:
+    """A tool that does something custom.
     Args:
-        :contentReference[oaicite:3]{index=3}
-        :contentReference[oaicite:4]{index=4}
+        arg1: the first input string
+        arg2: the second input integer
     """
-    :contentReference[oaicite:5]{index=5}
+    return f"Received {arg1} and {arg2}"
 
 @tool
-:contentReference[oaicite:6]{index=6}
-    :contentReference[oaicite:7]{index=7}
+def get_current_time_in_timezone(timezone: str) -> str:
+    """Returns current time for a given timezone.
+    Args:
+        timezone: the name of the timezone, e.g., 'America/New_York'
+    """
     try:
-        :contentReference[oaicite:8]{index=8}
-        :contentReference[oaicite:9]{index=9}
-    :contentReference[oaicite:10]{index=10}
-        :contentReference[oaicite:11]{index=11}
+        tz = pytz.timezone(timezone)
+        return datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+    except Exception as e:
+        return f"Timezone error: {e}"
 
-# Final answer aggregator tool
 final_answer = FinalAnswerTool()
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+model = "Qwen/Qwen2.5-Coder-32B-Instruct"
 
-# Load image generation tool from HF Hub
-:contentReference[oaicite:12]{index=12}
-
-# Model specified as string per current smolagents standards
-:contentReference[oaicite:13]{index=13}
-
-# Load system prompt templates
-:contentReference[oaicite:14]{index=14}
-    :contentReference[oaicite:15]{index=15}
+with open("prompts.yaml", "r") as stream:
+    prompt_templates = yaml.safe_load(stream)
 
-# Instantiate CodeAgent with all tools
 agent = CodeAgent(
     model=model,
     tools=[
@@ -54,5 +49,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch the Gradio interface
-:contentReference[oaicite:16]{index=16}
+GradioUI(agent).launch()