import os
from dotenv import load_dotenv

load_dotenv()

APP_SECRET = os.getenv("APP_SECRET")

# Base URLs
BASE_URL = "https://www.blackbox.ai"

# 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.