Update app.py

app.py

@@ -36,9 +36,17 @@ class ConversationManager:
 
 conversation_manager = ConversationManager()
 
-def get_web_search_results(query: str, max_results: int = 10) -> List[Dict[str, str]]:
+def duckduckgo_search(query):
+    with DDGS() as ddgs:
+        results = ddgs.text(query, max_results=5)
+        return list(results)
+
+def get_web_search_results(query: str, max_results: int = 10, use_ddgs_text: bool = False) -> List[Dict[str, str]]:
     try:
-        results = list(DDGS().text(query, max_results=max_results))
+        if use_ddgs_text:
+            results = duckduckgo_search(query)
+        else:
+            results = list(DDGS().text(query, max_results=max_results))
         if not results:
             print(f"No results found for query: {query}")
         return results
@@ -87,10 +95,11 @@ def summarize_web_results(query: str, search_results: List[Dict[str, str]], conv
     except Exception as e:
         return f"An error occurred during summarization: {str(e)}"
 
-def respond(message, history, model, temperature, num_calls, use_web_search):
+def respond(message, history, model, temperature, num_calls, use_web_search, search_method):
     logging.info(f"User Query: {message}")
     logging.info(f"Model Used: {model}")
     logging.info(f"Use Web Search: {use_web_search}")
+    logging.info(f"Search Method: {search_method}")
 
     if use_web_search:
         original_query = message
@@ -100,7 +109,7 @@ def respond(message, history, model, temperature, num_calls, use_web_search):
 
         final_summary = ""
         for _ in range(num_calls):
-            search_results = get_web_search_results(rephrased_query)
+            search_results = get_web_search_results(rephrased_query, use_ddgs_text=(search_method == "DDGS.text"))
             if not search_results:
                 final_summary += f"No search results found for the query: {rephrased_query}\n\n"
             elif "error" in search_results[0]:
@@ -140,7 +149,7 @@ def initial_conversation():
         (None, "Welcome! I'm your AI assistant for web search. Here's how you can use me:\n\n"
                 "1. Make sure the 'Use Web Search' checkbox is enabled in the Additional Inputs section.\n"
                 "2. Ask me any question, and I'll search the web for the most relevant and up-to-date information.\n"
-                "3. You can adjust the model, temperature, and number of API calls in the Additional Inputs section to fine-tune your results.\n\n"
+                "3. You can adjust the model, temperature, number of API calls, and search method in the Additional Inputs section to fine-tune your results.\n\n"
                 "To get started, just ask me a question!")
     ]
 
@@ -151,7 +160,8 @@ demo = gr.ChatInterface(
         gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[0]),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature"),
         gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls"),
-        gr.Checkbox(label="Use Web Search", value=True)
+        gr.Checkbox(label="Use Web Search", value=True),
+        gr.Radio(["DDGS.chat", "DDGS.text"], label="Search Method", value="DDGS.chat")
     ],
     title="AI-powered Web Search Assistant",
     description="Ask questions and get answers from the latest web information.",