Update app.py

app.py

@@ -2,9 +2,9 @@ import logging
 import hashlib
 import json
 import os
-from smolagents import CodeAgent, HfApiModel, tool
+from smolagents import CodeAgent, ApiModel, tool
 
-# Set up logging to see what's happening (logs appear in the Space logs)
+# Set up logging (appears in Space logs)
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
@@ -27,7 +27,7 @@ def calculator(expression: str) -> str:
     Safely evaluate a mathematical expression.
     Input should be a simple arithmetic expression (e.g., '2 + 2').
     """
-    # Use a safe evaluator that only allows numbers and operators
+    # Allow only numbers, basic operators, parentheses, and spaces
     allowed_chars = set("0123456789+-*/(). ")
     if not all(c in allowed_chars for c in expression):
         return "Error: Expression contains disallowed characters."
@@ -42,7 +42,7 @@ def calculator(expression: str) -> str:
 def web_search(query: str) -> str:
     """
     Search the web for up-to-date information.
-    Uses DuckDuckGo (free, no API key). Install duckduckgo-search first.
+    Uses DuckDuckGo (free, no API key). Requires duckduckgo-search package.
     """
     try:
         from duckduckgo_search import DDGS
@@ -50,7 +50,6 @@ def web_search(query: str) -> str:
             results = list(ddgs.text(query, max_results=3))
         if not results:
             return "No results found."
-        # Format results as a simple string
         snippets = []
         for r in results:
             snippets.append(f"Title: {r['title']}\nBody: {r['body']}\nURL: {r['href']}")
@@ -61,16 +60,18 @@ def web_search(query: str) -> str:
         return f"Search error: {e}"
 
 # ---------- Agent ----------
+# Use ApiModel (the correct class name in current smolagents)
 # You can specify a model explicitly, e.g.:
-# model = HfApiModel("meta-llama/Llama-2-7b-chat-hf")
-# Or use the default (zephyr-7b-beta)
-model = HfApiModel()  # Uses your HF token if set, otherwise public inference
+# model = ApiModel("meta-llama/Llama-2-7b-chat-hf")
+# The default uses Hugging Face's free inference API.
+model = ApiModel()
 
-# Add tools you need. Remove web_search if you don't have the package.
+# Only include web_search if the package is installed
 tools = [calculator]
 try:
     import duckduckgo_search
     tools.append(web_search)
+    logger.info("Web search tool enabled.")
 except ImportError:
     logger.warning("duckduckgo-search not installed, web_search disabled.")