Update app.py

app.py

@@ -280,10 +280,41 @@ def duckduckgo_search(query):
         results = ddgs.text(query, max_results=5)
     return results
 
-def duckduckgo_chat(keywords, model, timeout=30):
-    with DDGS() as ddgs:
-        response = ddgs.chat(keywords, model=model, timeout=timeout)
-    return response
+def chat(self, keywords: str, model: str = "gpt-4o-mini", timeout: int = 30) -> str:
+    """Initiates a chat session with DuckDuckGo AI.
+
+    Args:
+        keywords (str): The initial message or question to send to the AI.
+        model (str): The model to use: "gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b". Defaults to "gpt-4o-mini".
+        timeout (int): Timeout value for the HTTP client. Defaults to 30.
+
+    Returns:
+        str: The response from the AI.
+    """
+    if model == "gpt-4o-mini":
+        url = "https://api.duckduckgo.com/api/d2a/v1/chat/gpt-4o-mini"
+    elif model == "claude-3-haiku":
+        url = "https://api.duckduckgo.com/api/d2a/v1/chat/claude-3-haiku"
+    elif model == "llama-3.1-70b":
+        url = "https://api.duckduckgo.com/api/d2a/v1/chat/llama-3.1-70b"
+    elif model == "mixtral-8x7b":
+        url = "https://api.duckduckgo.com/api/d2a/v1/chat/mixtral-8x7b"
+    else:
+        raise ValueError(f"Invalid model: {model}")
+
+    payload = {"keywords": keywords}
+    headers = {
+        "Content-Type": "application/json",
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
+    }
+
+    try:
+        response = requests.post(url, json=payload, headers=headers, timeout=timeout)
+        response.raise_for_status()
+        return response.json()["response"]
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Error in DuckDuckGo chat: {str(e)}")
+        return "Error in DuckDuckGo chat. Please try again later."
 
 class CitingSources(BaseModel):
     sources: List[str] = Field(
@@ -358,10 +389,8 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
                     yield partial_response
             elif model in ["gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b"]:
                 # Use DuckDuckGo Chat API
-                for partial_response in get_response_from_duckduckgo(message, model, num_calls=num_calls, temperature=temperature):
-                    first_line = partial_response.split('\n')[0] if partial_response else ''
-                    logging.info(f"Generated Response (first line): {first_line}")
-                    yield partial_response
+                response = chat(message, model=model, timeout=30)
+                yield response
             else:
                 # Use Hugging Face API
                 for partial_response in get_response_from_pdf(message, model, selected_docs, num_calls=num_calls, temperature=temperature):
@@ -379,15 +408,6 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
 
 logging.basicConfig(level=logging.DEBUG)
 
-def get_response_from_duckduckgo(message, model, num_calls=3, temperature=0.2):
-    for i in range(num_calls):
-        try:
-            response = duckduckgo_chat(message, model=model, timeout=30)
-            yield response
-        except Exception as e:
-            logging.error(f"Error in generating response from DuckDuckGo: {str(e)}")
-            yield f"I apologize, but an error occurred: {str(e)}. Please try again later."
-
 def get_response_from_cloudflare(prompt, context, query, num_calls=3, temperature=0.2, search_type="pdf"):
     headers = {
         "Authorization": f"Bearer {API_TOKEN}",