Deploy GAIA agent

app.py

@@ -25,63 +25,26 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 @tool
 def web_search_tool(query: str) -> str:
     """
-    Search the web for information using DuckDuckGo.
-    Args:
-        query: The search query string
-    Returns:
-        String containing search results
+    Perform a DuckDuckGo search and return formatted results.
     """
     try:
-        # Try multiple approaches for DDGS
-        search_results = []
-        
-        # Approach 1: Basic DDGS
-        try:
-            ddgs = DDGS()
-            search_results = list(ddgs.text(query, max_results=5))
-        except Exception as e1:
-            print(f"DDGS Approach 1 failed: {e1}")
-            
-            # Approach 2: Try with timeout parameter only
-            try:
-                ddgs = DDGS(timeout=10)
-                search_results = list(ddgs.text(query, max_results=5))
-            except Exception as e2:
-                print(f"DDGS Approach 2 failed: {e2}")
-                
-                # Approach 3: Minimal call without max_results
-                try:
-                    ddgs = DDGS()
-                    search_results = list(ddgs.text(query))[:5]  # Limit results manually
-                except Exception as e3:
-                    print(f"DDGS Approach 3 failed: {e3}")
-                    
-                    # Approach 4: Try older style initialization
-                    try:
-                        from duckduckgo_search import ddg
-                        search_results = ddg(query, max_results=5)
-                    except Exception as e4:
-                        print(f"DDGS Approach 4 failed: {e4}")
-                        return f"All DDGS approaches failed. Last error: {e4}"
-        
-        if not search_results:
-            return "No search results found."
-        
-        results = []
-        for r in search_results:
-            try:
-                title = r.get('title', 'No title')
-                url = r.get('href', 'No URL')
-                snippet = r.get('body', 'No snippet')
-                results.append(f"Title: {title}\nURL: {url}\nSnippet: {snippet}")
-            except Exception as parse_error:
-                print(f"Error parsing result: {parse_error}")
-                continue
-        
-        return "\n\n".join(results) if results else "No valid results could be parsed."
-        
+        from duckduckgo_search import DDGS
+        with DDGS() as ddgs:
+            results = ddgs.text(query, max_results=5)
+            if not results:
+                return "No search results found."
+
+            output = []
+            for result in results:
+                output.append(
+                    f"Title: {result.get('title', 'No title')}\n"
+                    f"URL: {result.get('href', 'No URL')}\n"
+                    f"Snippet: {result.get('body', 'No snippet')}\n"
+                )
+            return "\n\n".join(output)
     except Exception as e:
-        return f"Search tool error: {str(e)}"
+        return f"Web search failed: {str(e)}"
+
 
 @tool
 def calculator_tool(expression: str) -> str: