Update app.py

app.py

@@ -3,11 +3,28 @@ import requests
 import pandas as pd
 import gradio as gr
 from huggingface_hub import InferenceClient
-from smolagents import CodeAgent, DuckDuckGoSearchTool
+from smolagents import CodeAgent, tool
+from duckduckgo_search import DDGS
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 MODEL_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
 
+@tool
+def web_search(query: str) -> str:
+    """Performs a web search to find current factual information.
+    
+    Args:
+        query: The search query string.
+    """
+    try:
+        with DDGS() as ddgs:
+            results = list(ddgs.text(query, max_results=3))
+            if not results:
+                return "No results found."
+            return "\n".join([f"- {r.get('title', '')}: {r.get('body', '')}" for r in results])
+    except Exception as e:
+        return f"Search error: {e}"
+
 class CustomHFModel:
     def __init__(self, model_id: str, token: str):
         self.model_id = model_id
@@ -26,7 +43,7 @@ class CustomHFModel:
 
             response = self.client.chat_completion(
                 messages=formatted_messages,
-                max_tokens=200,
+                max_tokens=300,
                 temperature=0.0
             )
             return response.choices[0].message
@@ -38,10 +55,9 @@ class CustomHFModel:
 
 def create_agent(token: str):
     model = CustomHFModel(model_id=MODEL_ID, token=token)
-    search_tool = DuckDuckGoSearchTool()
     
     agent = CodeAgent(
-        tools=[search_tool],
+        tools=[web_search],
         model=model,
         max_steps=3
     )
@@ -54,7 +70,7 @@ def solve_question(agent, question: str) -> str:
     prompt = (
         f"Question: {question}\n\n"
         "Instructions:\n"
-        "1. Use the search tool if you need external knowledge or facts.\n"
+        "1. Use the `web_search` tool if you need external knowledge or facts.\n"
         "2. Return ONLY the concise final answer (e.g., a single number, name, or short string).\n"
         "3. Do not include extra sentences, explanation, or quotes."
     )