Update app.py

app.py

@@ -280,42 +280,6 @@ def duckduckgo_search(query):
         results = ddgs.text(query, max_results=5)
     return results
 
-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(
         ...,
@@ -406,6 +370,42 @@ def respond(message, history, model, temperature, num_calls, use_web_search, sel
         else:
             yield f"An error occurred with the {model} model: {str(e)}. Please try again or select a different model."
 
+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."
+
 logging.basicConfig(level=logging.DEBUG)
 
 def get_response_from_cloudflare(prompt, context, query, num_calls=3, temperature=0.2, search_type="pdf"):