Update app.py

app.py

@@ -5,7 +5,7 @@ from smolagents import (
     DuckDuckGoSearchTool,
     load_tool,
     tool,
-    HfApiModel,
+    HfApiModel,   # <- replacement for the old InferenceClientModel
 )
 import datetime
 import re
@@ -13,10 +13,7 @@ import requests
 import pytz
 import yaml
 
-# In the template, FinalAnswerTool lives in a local module
 from tools.final_answer import FinalAnswerTool
-
-# Simple Gradio wrapper provided by the template Space
 from Gradio_UI import GradioUI
 
 
@@ -24,10 +21,7 @@ from Gradio_UI import GradioUI
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Fetch the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
+    """Fetch the current local time in a specified timezone, e.g. 'America/Los_Angeles'."""
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -38,14 +32,10 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 @tool
 def fetch_page_title(url: str) -> str:
-    """Fetch the HTML page at a URL and return its <title>.
-    Args:
-        url: The web page to fetch (http/https).
-    """
+    """Fetch the HTML page at a URL and return its <title>."""
     try:
         resp = requests.get(url, timeout=10)
         resp.raise_for_status()
-        # crude but robust title extraction
         m = re.search(r"<title[^>]*>(.*?)</title>", resp.text, flags=re.IGNORECASE | re.DOTALL)
         title = re.sub(r"\s+", " ", m.group(1)).strip() if m else "(no <title> found)"
         return f"Title: {title}"
@@ -55,10 +45,7 @@ def fetch_page_title(url: str) -> str:
 
 @tool
 def echo_and_length(text: str) -> str:
-    """Return the text back (trimmed) and its character length.
-    Args:
-        text: Any text.
-    """
+    """Return the text back (trimmed) and its character length."""
     t = text.strip()
     return f"Echo: {t}\nLength: {len(t)}"
 
@@ -73,23 +60,18 @@ model = HfApiModel(
     max_tokens=2096,
 )
 
-# Import a community tool from the Hub (text-to-image)
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-# Built-in search tool
 web_search = DuckDuckGoSearchTool()
 
-# Load system prompt templates from YAML
 with open("prompts.yaml", "r", encoding="utf-8") as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the Agent with ALL the tools wired in
 agent = CodeAgent(
     model=model,
     tools=[
-        final_answer,            # required: allows the agent to end with a final answer
-        web_search,              # web search capability
-        image_generation_tool,   # text-to-image generation
+        final_answer,
+        web_search,
+        image_generation_tool,
         get_current_time_in_timezone,
         fetch_page_title,
         echo_and_length,
@@ -98,10 +80,9 @@ agent = CodeAgent(
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name="My First smolagents Agent",
+    name="my_first_smolagents_agent",  # <- valid Python identifier (no spaces)
     description="A tiny agent that can search, fetch titles, tell time, and generate images.",
-    prompt_templates=prompt_templates,  # use the system prompts provided in the template
+    prompt_templates=prompt_templates,
 )
 
-# Launch the Gradio UI
 GradioUI(agent).launch()