Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 from huggingface_hub import login
 import datetime
 import pytz
-import yaml
 
 from smolagents import (
     CodeAgent,
@@ -15,7 +14,7 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # ——————————————————————————————————————————————
-# 1) Login to Hugging Face using your HF_HUB_TOKEN secret
+# 1) Authenticate to HF
 # ——————————————————————————————————————————————
 hf_token = os.getenv("HF_HUB_TOKEN")
 if not hf_token:
@@ -23,74 +22,64 @@ if not hf_token:
 login(hf_token)
 
 # ——————————————————————————————————————————————
-# 2) Define your custom tools
+# 2) Custom tools (with proper Args/Returns sections)
 # ——————————————————————————————————————————————
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
     """
-    A tool that does nothing yet.
+    A placeholder tool for demonstration.
 
     Args:
-        arg1: The first argument (a string).
-        arg2: The second argument (an integer).
+        arg1: A sample string.
+        arg2: A sample integer.
 
     Returns:
-        A placeholder response.
+        A placeholder response string.
     """
     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.
+    Fetch the current time in a given timezone.
 
     Args:
-        timezone: A valid timezone name, e.g., 'America/New_York'.
+        timezone: A valid tz database name (e.g., 'Europe/Paris').
 
     Returns:
-        The current local time formatted as 'YYYY-MM-DD HH:MM:SS', 
-        or an error message if the timezone is invalid.
+        The local time formatted as 'YYYY-MM-DD HH:MM:SS', or an error message.
     """
     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}"
+        now = datetime.datetime.now(tz)
+        return now.strftime("The current local time in %Z (%z) is %Y-%m-%d %H:%M:%S")
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {e}"
+        return f"Error: {e}"
 
+# the required final answer tool
 final_answer = FinalAnswerTool()
 
 # ——————————————————————————————————————————————
-# 3) Load your model via InferenceClientModel
+# 3) Load the model (current class name)
 # ——————————————————————————————————————————————
 model = InferenceClientModel(
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     max_tokens=2096,
     temperature=0.5,
     api_key=hf_token,
-    provider="huggingface"
+    provider="huggingface"  # bypasses any external routing errors
 )
 
 # ——————————————————————————————————————————————
-# 4) Import additional tools from the Hub
+# 4) Load Hub-provided tools
 # ——————————————————————————————————————————————
-image_generation_tool = load_tool(
-    "agents-course/text-to-image", trust_remote_code=True
-)
-duck_search = DuckDuckGoSearchTool()
-
-# ——————————————————————————————————————————————
-# 5) Load & sanitize your prompt templates
-# ——————————————————————————————————————————————
-with open("prompts.yaml", "r") as f:
-    prompt_templates = yaml.safe_load(f)
-prompt_templates.setdefault(
-    "final_answer",
-    "Final answer:\n{{answer}}"
-)
+image_gen = load_tool("agents-course/text-to-image", trust_remote_code=True)
+web_search = DuckDuckGoSearchTool()
 
 # ——————————————————————————————————————————————
-# 6) Instantiate the CodeAgent
+# 5) Instantiate agent *without* custom prompt_templates
+#    → uses the library’s defaults, so you’ll never hit:
+#      AttributeError: 'str' object has no attribute 'keys'
 # ——————————————————————————————————————————————
 agent = CodeAgent(
     model=model,
@@ -98,15 +87,16 @@ agent = CodeAgent(
         final_answer,
         my_custom_tool,
         get_current_time_in_timezone,
-        image_generation_tool,
-        duck_search,
+        image_gen,
+        web_search,
     ],
     max_steps=6,
     verbosity_level=1,
-    prompt_templates=prompt_templates,
+    add_base_tools=False,   # keep it minimal; you’ve already added web_search
+    # note: no `prompt_templates=` here :contentReference[oaicite:0]{index=0}
 )
 
 # ——————————————————————————————————————————————
-# 7) Launch the Gradio UI
+# 6) Launch your Gradio interface
 # ——————————————————————————————————————————————
 GradioUI(agent).launch()