Deploy GAIA agent

app.py

@@ -11,9 +11,10 @@ from PIL import Image
 import matplotlib.pyplot as plt
 import numpy as np
 from pathlib import Path
+from duckduckgo_search import DDGS
 
 # SmolaAgent imports
-from smolagents import CodeAgent, tool, DuckDuckGoSearchTool, PythonInterpreterTool
+from smolagents import CodeAgent, tool, PythonInterpreterTool
 from smolagents.models import LiteLLMModel
 
 # --- Constants ---
@@ -21,7 +22,6 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Enhanced Tools for GAIA ---
 
-# Replace the web_search_tool implementation with this updated version
 @tool
 def web_search_tool(query: str) -> str:
     """
@@ -33,11 +33,10 @@ def web_search_tool(query: str) -> str:
     """
     try:
         # Use duckduckgo_search directly with the current API
-        from duckduckgo_search import DDGS
-        
         results = []
         with DDGS() as ddgs:
-            for r in ddgs.text(query, max_results=3):
+            search_results = ddgs.text(query, max_results=5)
+            for r in search_results:
                 results.append(f"Title: {r['title']}\nURL: {r['href']}\nSnippet: {r['body']}")
         
         if not results:
@@ -46,6 +45,7 @@ def web_search_tool(query: str) -> str:
         return "\n\n".join(results)
     except Exception as e:
         return f"Search failed: {str(e)}"
+
 @tool
 def calculator_tool(expression: str) -> str:
     """
@@ -469,7 +469,7 @@ with gr.Blocks() as demo:
         **Enhanced GAIA Agent with SmolaAgent Framework**
         
         This agent is equipped with:
-        - 🔍 Web search capabilities (DuckDuckGo)
+        - 🔍 Web search capabilities (DuckDuckGo) - **FIXED**
         - 🧮 Mathematical calculator
         - 🖼️ Image analysis
         - 📁 File processing (CSV, JSON, text files)