Update api/config.py

api/config.py

@@ -3,10 +3,123 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-APP_SECRET = os.getenv("APP_SECRET", "1")  # Replace with a secure default or handle appropriately
+APP_SECRET = os.getenv("APP_SECRET")
 
-# GizAI specific configurations
-GIZAI_BASE_URL = "https://app.giz.ai"
-GIZAI_API_ENDPOINT = "https://app.giz.ai/api/data/users/inferenceServer.infer"
+# Base URLs
+BASE_URL = "https://www.blackbox.ai"
 
-# Additional configurations can be added here
+# Headers
+headers = {
+    'accept': '*/*',
+    'accept-language': 'en-US,en;q=0.9',
+    'origin': 'https://www.blackbox.ai',
+    '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',
+}
+
+# Allowed models
+ALLOWED_MODELS = [
+    {"id": "blackboxai", "name": "blackboxai"},
+    {"id": "blackboxai-pro", "name": "blackboxai-pro"},
+    # Add all other models...
+    # GizAI models
+    {"id": "chat-gemini-flash", "name": "chat-gemini-flash"},
+    {"id": "chat-gemini-pro", "name": "chat-gemini-pro"},
+    {"id": "chat-gpt4m", "name": "chat-gpt4m"},
+    {"id": "chat-gpt4", "name": "chat-gpt4"},
+    {"id": "claude-sonnet", "name": "claude-sonnet"},
+    {"id": "claude-haiku", "name": "claude-haiku"},
+    {"id": "llama-3-70b", "name": "llama-3-70b"},
+    {"id": "llama-3-8b", "name": "llama-3-8b"},
+    {"id": "mistral-large", "name": "mistral-large"},
+    {"id": "chat-o1-mini", "name": "chat-o1-mini"},
+    {"id": "flux1", "name": "flux1"},
+    {"id": "sdxl", "name": "sdxl"},
+    {"id": "sd", "name": "sd"},
+    {"id": "sd35", "name": "sd35"},
+]
+
+# Model to provider mapping
+MODEL_PROVIDER_MAPPING = {
+    # Blackbox.ai models
+    "blackboxai": "blackboxai",
+    "blackboxai-pro": "blackboxai",
+    # ... other blackbox.ai models
+    "Niansuh": "blackboxai",
+
+    # GizAI models
+    "chat-gemini-flash": "gizai",
+    "chat-gemini-pro": "gizai",
+    "chat-gpt4m": "gizai",
+    "chat-gpt4": "gizai",
+    "claude-sonnet": "gizai",
+    "claude-haiku": "gizai",
+    "llama-3-70b": "gizai",
+    "llama-3-8b": "gizai",
+    "mistral-large": "gizai",
+    "chat-o1-mini": "gizai",
+    "flux1": "gizai",
+    "sdxl": "gizai",
+    "sd": "gizai",
+    "sd35": "gizai",
+    # Aliases
+    "gemini-flash": "gizai",
+    "gemini-pro": "gizai",
+    "gpt-4o-mini": "gizai",
+    "gpt-4o": "gizai",
+    "claude-3.5-sonnet": "gizai",
+    "claude-3-haiku": "gizai",
+    "llama-3.1-70b": "gizai",
+    "llama-3.1-8b": "gizai",
+    "o1-mini": "gizai",
+    "sd-1.5": "gizai",
+    "sd-3.5": "gizai",
+    "flux-schnell": "gizai",
+}
+
+# Model mapping (if any aliasing is needed)
+MODEL_MAPPING = {
+    # Blackbox.ai models
+    "blackboxai": "blackboxai",
+    "blackboxai-pro": "blackboxai-pro",
+    # ... other blackbox.ai models
+
+    # GizAI models
+    "chat-gemini-flash": "chat-gemini-flash",
+    "chat-gemini-pro": "chat-gemini-pro",
+    "chat-gpt4m": "chat-gpt4m",
+    "chat-gpt4": "chat-gpt4",
+    "claude-sonnet": "claude-sonnet",
+    "claude-haiku": "claude-haiku",
+    "llama-3-70b": "llama-3-70b",
+    "llama-3-8b": "llama-3-8b",
+    "mistral-large": "mistral-large",
+    "chat-o1-mini": "chat-o1-mini",
+    "flux1": "flux1",
+    "sdxl": "sdxl",
+    "sd": "sd",
+    "sd35": "sd35",
+    # Aliases
+    "gemini-flash": "chat-gemini-flash",
+    "gemini-pro": "chat-gemini-pro",
+    "gpt-4o-mini": "chat-gpt4m",
+    "gpt-4o": "chat-gpt4",
+    "claude-3.5-sonnet": "claude-sonnet",
+    "claude-3-haiku": "claude-haiku",
+    "llama-3.1-70b": "llama-3-70b",
+    "llama-3.1-8b": "llama-3-8b",
+    "o1-mini": "chat-o1-mini",
+    "sd-1.5": "sd",
+    "sd-3.5": "sd35",
+    "flux-schnell": "flux1",
+}
+
+# Any additional configurations (e.g., MODEL_PREFIXES, AGENT_MODE) can be added as needed.