Update app.py

app.py

@@ -1,42 +1,63 @@
-import requests
 from flask import Flask, request, jsonify
+import requests
+import json
 
 app = Flask(__name__)
 
-# Your PythonAnywhere API endpoint
-PYTHONANYWHERE_URL = "https://omarnuwara.pythonanywhere.com/get-response"
-
-@app.route('/chat', methods=['POST'])
-def chat():
+@app.route('/duckchat', methods=['POST'])
+def duckchat():
     try:
-        # Ensure the request contains JSON and a "message" key
-        user_input = request.json.get("message", "") if request.is_json else None
-
-        if not user_input:
-            return jsonify({"error": "No message provided in the request."}), 400
-
-        # Create the payload to send to PythonAnywhere
-        data = {"message": user_input}
-
-        # Forward the request to PythonAnywhere
-        response = requests.post(PYTHONANYWHERE_URL, json=data)
-
-        if response.status_code == 200:
-            response_json = response.json()
-            ai_response = response_json.get("response", "No 'response' key found in the JSON.")
-            ai_response = ai_response.encode('latin1').decode('utf-8', 'ignore')
-            return (ai_response)
-        else:
-            # Handle errors from PythonAnywhere
-            return jsonify({
-                "error": "Error from PythonAnywhere",
-                "details": response.text
-            }), response.status_code
+        # Get the input data from the request
+        input_data = request.json
+
+        if not input_data or 'messages' not in input_data:
+            return jsonify({"error": "Invalid input. 'messages' field is required."}), 400
+
+        # Get the status endpoint to fetch x-vqd-4
+        status_url = "https://duckduckgo.com/duckchat/v1/status"
+        status_headers = {
+            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
+            "x-vqd-accept": "1"
+        }
+        status_response = requests.get(status_url, headers=status_headers)
+        x_vqd_4 = status_response.headers.get("x-vqd-4", "")
+
+        if not x_vqd_4:
+            return jsonify({"error": "Failed to retrieve x-vqd-4 token."}), 500
+
+        # Send the chat message
+        chat_url = "https://duckduckgo.com/duckchat/v1/chat"
+        chat_headers = {
+            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
+            "x-vqd-4": x_vqd_4
+        }
+        chat_payload = {
+            "messages": input_data['messages'],
+            "model": input_data.get('model', "gpt-4o-mini")
+        }
+        
+        chat_response = requests.post(chat_url, headers=chat_headers, json=chat_payload)
+
+        # Check if the response is successful
+        if chat_response.status_code != 200:
+            return jsonify({"error": "Failed to fetch response from DuckDuckGo Chat API."}), chat_response.status_code
+
+        # Process the response
+        response_lines = chat_response.text.splitlines()
+        final_response = ""
+
+        for line in response_lines:
+            try:
+                data = json.loads(line.replace("data: ", ""))
+                if "message" in data:
+                    final_response += data["message"]
+            except json.JSONDecodeError:
+                continue
+
+        return jsonify({"response": final_response})
 
     except Exception as e:
-        # Catch any unexpected errors
-        return jsonify({"error": "An error occurred", "details": str(e)}), 500
-
+        return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)