Update app.py

app.py

@@ -272,9 +272,13 @@ def generate_chunked_response(prompt, model, max_tokens=10000, num_calls=3, temp
     return final_response
 
 class SimpleDDGSearch:
-    def search(self, query: str, num_results: int = 10):
-        results = DDGS().news(query, region='wt-wt', safesearch='off', max_results=num_results)
-        return [res["url"] for res in results]
+    def search(self, query: str, model: str = "claude-3-haiku") -> str:
+        result = DDGS().chat(
+            keywords=query,
+            model=model,
+            timeout=30
+        )
+        return result
 
 class TrafilaturaWebCrawler:
     def get_website_content_from_url(self, url: str) -> str: