Update app.py

app.py

@@ -8,7 +8,8 @@ import requests
 from typing import List, Dict, Union
 import pandas as pd
 import wikipediaapi
-from duckduckgo_search import ddg
+from serpapi import GoogleSearch
+from typing import List, Dict, Optional
 
 load_dotenv()
 
@@ -20,36 +21,81 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 
 class BasicAgent:
-    def __init__(self):
-        print("BasicAgent initialized.")
+import os
+from serpapi import GoogleSearch
+from typing import List, Dict, Optional
+
+class BasicAgent:
+    def __init__(self, api_key: str = None):
+        self.api_key = api_key or os.getenv("SERP_API_KEY")
+        if not self.api_key:
+            raise ValueError("Missing SERPAPI_API_KEY. Get one at https://serpapi.com/")
+        print("SerpAPI Agent initialized")
 
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question: {question[:50]}...")
-        answer = self.basic_search(question)
-        print(f"Answer: {answer}")
-        return answer
+    def search(self, query: str, num_results: int = 3) -> List[Dict]:
+        """Execute search and return structured results"""
+        params = {
+            "q": query,
+            "api_key": self.api_key,
+            "num": num_results,
+            "hl": "en",  # Language: English
+            "gl": "us"   # Country: United States
+        }
 
-    def basic_search(self, query):
         try:
-            # Get search results from DuckDuckGo
-            results = ddg(query, max_results=3)
-            
-            # Format the results
-            if not results:
-                return "No results found"
-                
-            formatted_results = []
-            for i, result in enumerate(results, 1):
-                formatted_results.append(
-                    f"{i}. {result['title']}\n"
-                    f"   {result['link']}\n"
-                    f"   {result['body']}"
-                )
-            
-            return "\n\n".join(formatted_results)
-            
+            search = GoogleSearch(params)
+            results = search.get_dict()
+            return self._format_results(results)
         except Exception as e:
-            return f"Search error: {str(e)}"
+            print(f"Search failed: {str(e)}")
+            return []
+
+    def _format_results(self, raw_results: Dict) -> List[Dict]:
+        """Extract and format organic results"""
+        formatted = []
+        for result in raw_results.get("organic_results", []):
+            formatted.append({
+                "position": result.get("position"),
+                "title": result.get("title"),
+                "link": result.get("link"),
+                "snippet": result.get("snippet"),
+                "source": result.get("source")
+            })
+        return formatted
+
+    def __call__(self, query: str) -> str:
+        """Callable interface that returns a string"""
+        results = self.search(query)
+        if not results:
+            return "No results found"
+        
+        output = []
+        for res in results:
+            output.append(
+                f"{res['position']}. {res['title']}\n"
+                f"   {res['link']}\n"
+                f"   {res['snippet']}\n"
+                f"   Source: {res['source']}"
+            )
+        return "\n\n".join(output)
+
+
+# Usage Example
+if __name__ == "__main__":
+    # Initialize with API key (or set SERPAPI_API_KEY environment variable)
+    agent = BasicAgent()
+    
+    # Perform search
+    query = "What is Python programming language?"
+    print(f"Searching for: {query}")
+    
+    # Option 1: Get structured data
+    structured_results = agent.search(query)
+    print("\nStructured Results:", structured_results[0])  # Print first result
+    
+    # Option 2: Get printable string
+    printable_results = agent(query)
+    print("\nFormatted Results:\n", printable_results)
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):