Update app.py

app.py

@@ -11,27 +11,46 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ====================== CUSTOM TOOLS ======================
 # Add any tools you want here. The more useful ones you add, the better your score.
 
+# ====================== CUSTOM TOOLS ======================
+
 @tool
 def web_search(query: str) -> str:
-    """Perform a web search and return results. Very useful for GAIA questions."""
-    # You can implement with DuckDuckGo, Tavily, or even requests to a search API
-    # For simplicity, here's a placeholder using a free search (replace with better if you want)
+    """Perform a web search using DuckDuckGo and return the top results.
+    
+    Args:
+        query: The search query to look up on the web.
+    
+    Returns:
+        str: A string containing the title and snippet of the top search results.
+    """
     try:
         from duckduckgo_search import DDGS
         with DDGS() as ddgs:
             results = list(ddgs.text(query, max_results=5))
-            return "\n\n".join([f"{r['title']}: {r['body']}" for r in results])
+            if not results:
+                return "No search results found."
+            formatted = []
+            for r in results:
+                formatted.append(f"Title: {r.get('title', 'N/A')}\nSnippet: {r.get('body', r.get('snippet', 'N/A'))}")
+            return "\n\n".join(formatted)
     except Exception as e:
-        return f"Search failed: {str(e)}"
+        return f"Web search failed: {str(e)}"
 
 
 @tool
 def calculate(expression: str) -> str:
-    """Evaluate a mathematical expression safely."""
+    """Evaluate a simple mathematical expression.
+    
+    Args:
+        expression: The math expression to calculate (e.g. '2 + 2 * 3', 'sin(3.14)', etc.)
+    
+    Returns:
+        str: The result of the calculation as a string.
+    """
     try:
-        # You can use sympy for more complex math if you install it
         import math
-        return str(eval(expression, {"__builtins__": {}}, {"math": math}))
+        allowed_names = {"math": math}
+        return str(eval(expression, {"__builtins__": {}}, allowed_names))
     except Exception as e:
         return f"Calculation error: {str(e)}"