Update app.py

app.py

@@ -1,24 +1,54 @@
-from smolagents import CodeAgent
+import datetime
+import pytz
 import yaml
+from smolagents import CodeAgent, HfApiModel
+from smolagents.tools import tool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
-from tool import my_custom_tool, get_current_time_in_timezone
 
-final_answer = FinalAnswerTool()
+# -----------------------------
+# ✅ Define tools directly here
+# -----------------------------
+
+@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
+def get_current_time_in_timezone(timezone: str) -> str:
+    """Fetch the current local time in a specified timezone.
+    Args:
+        timezone: A valid timezone string (e.g., 'America/New_York').
+    """
+    try:
+        tz = pytz.timezone(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: {str(e)}"
 
-# Load simplified prompt templates
+# -----------------------------
+# ✅ Load simplified prompt template
+# -----------------------------
 with open("simple_prompts.yaml") as f:
     prompt_templates = yaml.safe_load(f)
 
-# Set up the model
-from smolagents import HfApiModel
+# -----------------------------
+# ✅ Set up model and agent
+# -----------------------------
 model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    max_tokens=2048,
+    temperature=0.5
 )
 
-# Define agent with simplified prompt YAML
+final_answer = FinalAnswerTool()
+
 agent = CodeAgent(
     model=model,
     tools=[final_answer, my_custom_tool, get_current_time_in_timezone],
@@ -27,4 +57,7 @@ agent = CodeAgent(
     verbosity_level=1,
 )
 
+# -----------------------------
+# ✅ Launch Gradio interface
+# -----------------------------
 GradioUI(agent).launch()