Update app.py

app.py

@@ -47,10 +47,14 @@ def duckduckgo_search(query: str, max_results: int = 10) -> List[Dict[str, str]]
         results = list(ddgs.text(query, max_results=max_results))
     return results
 
-def get_web_search_results(query: str, model: str, num_calls: int = 3, temperature: float = 0.2, max_results: int = 10) -> Dict[str, any]:
+def get_web_search_results(query: str, model: str, num_calls: int = 3, temperature: float = 0.2, max_results: int = 10, search_method: str = "DDGS.chat") -> Dict[str, any]:
     try:
         # Perform web search
-        search_results = duckduckgo_search(query, max_results)
+        if search_method == "DDGS.text":
+            search_results = duckduckgo_search(query, max_results)
+        else:  # Default to DDGS.chat
+            with DDGS() as ddgs:
+                search_results = list(ddgs.text(query, max_results=max_results))
         
         if not search_results:
             return {"error": f"No results found for query: {query}"}
@@ -160,7 +164,7 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sea
 
         final_summary = ""
         for _ in range(num_calls):
-            search_results = get_web_search_results(rephrased_query, use_ddgs_text=(search_method == "DDGS.text"))
+            search_results = get_web_search_results(rephrased_query, model, num_calls, temperature, max_results=10, search_method=search_method)
             if not search_results:
                 final_summary += f"No search results found for the query: {rephrased_query}\n\n"
             elif "error" in search_results[0]: