Update app.py

app.py

@@ -1,12 +1,16 @@
 import requests
 from flask import Flask, request, jsonify
 from datetime import datetime
+import json
 
 app = Flask(__name__)
 
 # URLs
 SEPTIMIUS_JSON_URL = "https://corvo-ai-septimius.hf.space/read_Septimius_json"
-PYTHONANYWHERE_URL = "https://omarnuwara.pythonanywhere.com/get-response"
+
+# Brain Inc API Details
+BRAININC_API_URL = "https://api.braininc.net/stream/bas-demo-v4/nlp/completions_generation"
+BRAININC_API_TOKEN = "token 72ec00483379076f580eb8126f29da802a5140c3"  # Replace if needed
 
 def is_token_valid(token):
     """
@@ -38,6 +42,48 @@ def is_token_valid(token):
         return False
 
 
+def generate_ai_response(message):
+    """
+    Generate AI response using Brain Inc API.
+    """
+    headers = {
+        "Authorization": f"{BRAININC_API_TOKEN}",
+        "Content-Type": "application/json;charset=UTF-8",
+    }
+
+    payload = {
+        "identifier_type": "object_id",
+        "identifier_value": "4775260",
+        "pubsub_topic": "/studios/516104/wsuid_4647155_new-edge-112_nodeid_editor-111/textGenStream/1722445975926",
+        "stream": True,
+        "variables": {
+            "Input": message,
+            "User  Input": message,
+            "input": message,
+        }
+    }
+
+    try:
+        response = requests.post(BRAININC_API_URL, headers=headers, json=payload)
+        combined_content = ""
+
+        for line in response.text.splitlines():
+            if line.startswith('data: '):
+                json_data = line[6:].strip()
+                try:
+                    data = json.loads(json_data)
+                    delta_content = data.get('choices')[0].get('delta').get('content', '')
+                    combined_content += delta_content
+                except json.JSONDecodeError:
+                    continue
+
+        combined_content = combined_content.replace("\n", "<br>")
+        return combined_content
+    except requests.exceptions.RequestException as e:
+        print(f"Error connecting to Brain Inc API: {e}")
+        return "Error generating response"
+
+
 @app.route('/chat', methods=['POST'])
 def chat():
     try:
@@ -55,21 +101,9 @@ def chat():
         if not is_token_valid(token):
             return jsonify({"error": "You must subscribe to continue."}), 403
 
-        # Forward the message to PythonAnywhere API
-        data = {"message": message}
-        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')  # Handle encoding issues
-            return jsonify({"response": ai_response})
-        else:
-            # Handle errors from PythonAnywhere
-            return jsonify({
-                "error": "Error from PythonAnywhere",
-                "details": response.text
-            }), response.status_code
+        # Generate AI response using Brain Inc API
+        ai_response = generate_ai_response(message)
+        return jsonify({"response": ai_response})
 
     except Exception as e:
         # Catch any unexpected errors