Update app.py

app.py

@@ -1,77 +1,98 @@
 # app.py
-import datetime, pytz, yaml, requests
+"""
+Fully‑working SmolAgents chat‑based agent for Hugging Face Spaces.
+Paste this file into the root of your duplicated Space.
+"""
+
 from smolagents import (
-    CodeAgent, HfApiModel, DuckDuckGoSearchTool,
-    load_tool, tool
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    HfApiModel,
+    load_tool,
+    tool,                      # decorator that auto‑wraps functions as Tool objects
 )
 from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI   # supplied by the template
+from Gradio_UI import GradioUI
+
+import datetime
+import pytz
+import yaml
+
 
 # ────────────────────────────────
-# 1. Custom tools
+# 1. Custom tools (decorated)
 # ────────────────────────────────
+@tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Return the current local time in a given timezone.
+    """Return the current local time in an IANA timezone.
     Args:
-        timezone: IANA timezone string, e.g. 'Asia/Karachi'.
+        timezone: e.g. 'Asia/Karachi'
     """
     try:
         tz = pytz.timezone(timezone)
-        return datetime.datetime.now(tz).strftime("%Y‑%m‑%d %H:%M:%S")
+        now = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current time in {timezone} is {now}"
     except Exception as e:
         return f"Error: {e}"
 
+
+@tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
     """A playful demo tool.
     Args:
         arg1: any text
         arg2: any integer
     """
-    return f"You sent {arg1!r} and {arg2}; now build something cooler 😉"
+    return f"You sent {arg1!r} and {arg2}. Now build something cooler 😉"
+
 
 # ────────────────────────────────
-# 2. External tools from the Hub
+# 2. Hub & built‑in tools
 # ────────────────────────────────
 image_generation_tool = load_tool(
     "agents-course/text-to-image",
     trust_remote_code=True
 )
-duck_search_tool = DuckDuckGoSearchTool()   # ← instantiate!
+duck_search_tool = DuckDuckGoSearchTool()       # simple web search
+
 
 # ────────────────────────────────
-# 3. LLM setup
+# 3. LLM backend
 # ────────────────────────────────
 model = HfApiModel(
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    max_tokens=2096,
+    max_tokens=2048,
     temperature=0.5,
 )
 
+
 # ────────────────────────────────
 # 4. Prompt templates
 # ────────────────────────────────
 with open("prompts.yaml") as f:
     prompt_templates = yaml.safe_load(f)
 
+
 # ────────────────────────────────
-# 5. Build the agent
+# 5. Assemble the agent
 # ────────────────────────────────
 agent = CodeAgent(
     model=model,
     tools=[
-        FinalAnswerTool(),      # must be first
+        FinalAnswerTool(),          # ← keep this first
         duck_search_tool,
         image_generation_tool,
-        get_current_time_in_timezone,   # functions are auto‑wrapped
-        my_custom_tool
+        get_current_time_in_timezone,
+        my_custom_tool,
     ],
     max_steps=6,
     verbosity_level=1,
-    prompt_templates=prompt_templates
+    prompt_templates=prompt_templates,
 )
 
+
 # ────────────────────────────────
-# 6. Launch the Gradio UI
+# 6. Launch Gradio
 # ────────────────────────────────
 if __name__ == "__main__":
     GradioUI(agent).launch()