Update app.py

app.py

@@ -22,6 +22,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 
+import requests
+from bs4 import BeautifulSoup
+
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
@@ -29,30 +32,85 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         print(f"Agent received question: {question[:50]}...")
         answer = self.basic_search(question)
-        print(f"Answer: {answer}")
+        print(f"Answer: {answer[:200]}...")  # Truncate long answers
         return answer
 
     def basic_search(self, query):
         try:
-            # Get search results from DuckDuckGo
-            results = ddg(query, max_results=3)
+            # DuckDuckGo search URL
+            url = "https://html.duckduckgo.com/html/"
+            
+            # Headers to mimic a browser request
+            headers = {
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0',
+                'Accept': 'text/html,application/xhtml+xml',
+                'Accept-Language': 'en-US,en;q=0.5',
+            }
+            
+            # Search parameters
+            params = {
+                'q': query,
+                'kl': 'us-en'
+            }
+            
+            # Make the POST request
+            response = requests.post(url, headers=headers, data=params)
+            response.raise_for_status()  # Raise exception for bad status codes
+            
+            # Parse the HTML response
+            soup = BeautifulSoup(response.text, 'html.parser')
+            
+            # Find all search results
+            results = soup.find_all('div', class_='result')
             
-            # Format the results
             if not results:
-                return "No results found"
+                return "No results found for your query."
+                
+            # Prepare the answer with top 3 results
+            answer = "Here are the top search results:\n\n"
+            for i, result in enumerate(results[:3], 1):  # Limit to 3 results
+                title = result.find('a', class_='result__a').get_text(strip=True)
+                link = result.find('a', class_='result__a')['href']
+                snippet = result.find('a', class_='result__snippet').get_text(strip=True) if result.find('a', class_='result__snippet') else "No description available"
                 
-            formatted_results = []
-            for i, result in enumerate(results, 1):
-                formatted_results.append(
-                    f"{i}. {result['title']}\n"
-                    f"   {result['link']}\n"
-                    f"   {result['body']}"
-                )
+                answer += f"{i}. {title}\n   URL: {link}\n   Description: {snippet}\n\n"
             
-            return "\n\n".join(formatted_results)
+            return answer
             
+        except requests.exceptions.RequestException as e:
+            return f"Failed to complete the search request: {str(e)}"
         except Exception as e:
-            return f"Search error: {str(e)}"
+            return f"An unexpected error occurred: {str(e)}"
+
+# Example usage
+if __name__ == "__main__":
+    agent = BasicAgent()
+    
+    # Test queries
+    test_queries = [
+        "What is Python programming?",
+        "Latest Python version",
+        "How to learn machine learning"
+    ]
+    
+    for query in test_queries:
+        print("\n" + "="*50)
+        print(f"Testing query: {query}")
+        result = agent(query)
+        print("\nSearch Result:")
+        print(result)
+    
+    # Interactive mode
+    print("\n" + "="*50)
+    print("Interactive mode (type 'quit' to exit)")
+    while True:
+        user_query = input("\nEnter your search query: ")
+        if user_query.lower() == 'quit':
+            break
+        print("\n" + "="*50)
+        result = agent(user_query)
+        print("\nSearch Result:")
+        print(result)