Update api/config.py

api/config.py

@@ -1,32 +1,45 @@
-# config.py
-
 import os
 from dotenv import load_dotenv
 
 load_dotenv()
 
-# Base URL for the Blackbox AI API
+# Base URL and Common Headers
 BASE_URL = "https://www.blackbox.ai"
-
-# Simplified Common Headers
-COMMON_HEADERS = {
-    'Accept': '*/*',
-    'Accept-Language': 'en-US,en;q=0.9',
-    'Cache-Control': 'no-cache',
-    'Origin': BASE_URL,
-    'Pragma': 'no-cache',
-    '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'
-    ),
-    'Content-Type': 'application/json',
+common_headers = {
+    'accept': '*/*',
+    'accept-language': 'en-US,en;q=0.9',
+    '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',
 }
 
-# Application Secret (ensure this is set in your .env file)
+# 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")
 
-# Allowed Models
 ALLOWED_MODELS = [
     {"id": "blackboxai", "name": "blackboxai"},
     {"id": "blackboxai-pro", "name": "blackboxai-pro"},
@@ -79,12 +92,11 @@ ALLOWED_MODELS = [
     {"id": "builderAgent", "name": "builderAgent"},
 ]
 
-# Model Mapping for Aliases
 MODEL_MAPPING = {
     "blackboxai": "blackboxai",
     "blackboxai-pro": "blackboxai-pro",
     "flux": "flux",
-    "Image Generation": "flux",
+    "ImageGeneration": "flux",
     "llama-3.1-8b": "llama-3.1-8b",
     "llama-3.1-70b": "llama-3.1-70b",
     "llama-3.1-405b": "llama-3.1-405",
@@ -111,6 +123,7 @@ MODEL_MAPPING = {
     "GitlabAgent": "GitlabAgent",
     "GitAgent": "GitAgent",
     "RepoMap": "RepoMap",
+    # Additional mappings
     "gemini-flash": "gemini-1.5-flash",
     "claude-3.5-sonnet": "claude-sonnet-3.5",
     "flux": "flux",
@@ -136,14 +149,13 @@ MODEL_MAPPING = {
     "builderAgent": "builderAgent",
 }
 
-# Agent Modes
+# Agent modes
 AGENT_MODE = {
     'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
     'Niansuh': {'mode': True, 'id': "NiansuhAIk1HgESy", 'name': "Niansuh"},
     'o1-preview': {'mode': True, 'id': "o1Dst8La8", 'name': "o1-preview"},
 }
 
-# Trending Agent Modes
 TRENDING_AGENT_MODE = {
     "blackboxai": {},
     "gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
@@ -185,3 +197,17 @@ TRENDING_AGENT_MODE = {
     'YoutubeAgent': {'mode': True, 'id': "YoutubeAgent"},
     'builderAgent': {'mode': True, 'id': "builderAgent"},
 }
+
+# Model prefixes
+MODEL_PREFIXES = {
+
+}
+
+# Model referers
+MODEL_REFERERS = {
+    "blackboxai": "/?model=blackboxai",
+    "gpt-4o": "/?model=gpt-4o",
+    "gemini-pro": "/?model=gemini-pro",
+    "claude-sonnet-3.5": "/?model=claude-sonnet-3.5",
+    # Add any additional referers if necessary
+}