Update api/config.py

api/config.py

@@ -1,23 +1,32 @@
+# config.py
+
 import os
 from dotenv import load_dotenv
 
 load_dotenv()
 
-# Base URL and Simplified Common Headers
+# Base URL for the Blackbox AI API
 BASE_URL = "https://www.blackbox.ai"
-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',
+
+# 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',
 }
 
+# Application Secret (ensure this is set in your .env file)
 APP_SECRET = os.getenv("APP_SECRET")
 
+# Allowed Models
 ALLOWED_MODELS = [
     {"id": "blackboxai", "name": "blackboxai"},
     {"id": "blackboxai-pro", "name": "blackboxai-pro"},
@@ -70,11 +79,12 @@ ALLOWED_MODELS = [
     {"id": "builderAgent", "name": "builderAgent"},
 ]
 
+# Model Mapping for Aliases
 MODEL_MAPPING = {
     "blackboxai": "blackboxai",
     "blackboxai-pro": "blackboxai-pro",
     "flux": "flux",
-    "ImageGeneration": "flux",
+    "Image Generation": "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",
@@ -101,7 +111,6 @@ 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",
@@ -127,13 +136,14 @@ 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'},
@@ -175,7 +185,3 @@ TRENDING_AGENT_MODE = {
     'YoutubeAgent': {'mode': True, 'id': "YoutubeAgent"},
     'builderAgent': {'mode': True, 'id': "builderAgent"},
 }
-
-# Removed MODEL_PREFIXES and MODEL_REFERERS as per requirements
-
-# No header configuration functions are included since extra headers are to be removed