Update api/config.py

api/config.py

@@ -3,6 +3,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
+# Base URL and Common Headers
 BASE_URL = "https://www.blackbox.ai"
 common_headers = {
     'accept': '*/*',
@@ -10,9 +11,32 @@ common_headers = {
     'cache-control': 'no-cache',
     'origin': BASE_URL,
     'pragma': 'no-cache',
+    'priority': 'u=1, i',
+    'sec-ch-ua': '"Chromium";v="130", "Google Chrome";v="130", "Not?A_Brand";v="99"',
+    'sec-ch-ua-mobile': '?0',
+    'sec-ch-ua-platform': '"Windows"',
+    'sec-fetch-dest': 'empty',
+    'sec-fetch-mode': 'cors',
+    'sec-fetch-site': 'same-origin',
     'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
                   'AppleWebKit/537.36 (KHTML, like Gecko) '
                   'Chrome/130.0.0.0 Safari/537.36',
+}
+
+# Header Configurations for Specific API Calls
+def get_headers_api_chat(referer_url):
+    return {**common_headers, 'Content-Type': 'application/json', 'Referer': referer_url}
+
+def get_headers_chat(chat_url, next_action, next_router_state_tree):
+    return {
+        **common_headers,
+        'Accept': 'text/x-component',
+        'Content-Type': 'text/plain;charset=UTF-8',
+        'Referer': chat_url,
+        'next-action': next_action,
+        'next-router-state-tree': next_router_state_tree,
+        'next-url': '/',
+    }
 
 APP_SECRET = os.getenv("APP_SECRET")