import os
from dotenv import load_dotenv

load_dotenv()

BASE_URL = "https://blackboxaichat.onrender.com"
common_headers = {
    'accept': '*/*',
    'accept-language': 'en-US,en;q=0.9',
    'content-type': 'application/json',
    'origin': 'https://www.blackbox.ai',
    'priority': 'u=1, i',
    'sec-ch-ua': '"Google Chrome";v="135", "Not-A.Brand";v="8", "Chromium";v="135"',
    'sec-ch-ua-mobile': '?0',
    'sec-ch-ua-model': '""',
    '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/135.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',
        'next-action': next_action,
        'next-router-state-tree': next_router_state_tree,
        'next-url': '/',
    }

APP_SECRET = os.getenv("APP_SECRET")

ALLOWED_MODELS = [
    {"id": "blackboxai", "name": "blackboxai"},
    {"id": "niansuh-t1", "name": "niansuh-t1"},
    {"id": "o3-mini", "name": "o3-mini"},
    {"id": "snapzionai", "name": "snapzionai"},
    {"id": "blackboxai-pro", "name": "blackboxai-pro"},
    {"id": "flux", "name": "flux"},
    {"id": "Llama-4-Maverick-17B-128E", "name": "Llama-4-Maverick-17B-128E"},
    {"id": "llama-3.1-8b", "name": "llama-3.1-8b"},
    {"id": "llama-3.1-70b", "name": "llama-3.1-70b"},
    {"id": "llama-3.1-405b", "name": "llama-3.1-405"},
    {"id": "gpt-4o", "name": "gpt-4o"},
    {"id": "chatgpt-4o-latest", "name": "chatgpt-4o-latest"},
    {"id": "x-ai/grok-vision-beta", "name": "x-ai/grok-vision-beta"},
    {"id": "x-ai/grok-beta", "name": "x-ai/grok-beta"},
    {"id": "anthropic/claude-3.7-sonnet:thinking", "name": "anthropic/claude-3.7-sonnet:thinking"},
    {"id": "gpt-4o-2024-05-13", "name": "gpt-4o"},
    {"id": "gemini-pro", "name": "Gemini-PRO"},
    {"id": "gemini-flash-2.0", "name": "gemini-flash-2.0"},
    {"id": "gemini-1.5-flash", "name": "gemini-flash-2.0"},
    {"id": "claude-sonnet-3.5", "name": "claude-sonnet-3.5"},
    {"id": "deepseek-r1", "name": "deepseek-r1"},
    {"id": "deepseek-v3", "name": "deepseek-v3"},
    {"id": "Mistral-Small-24B-Instruct-2501", "name": "Mistral-Small-24B-Instruct-2501"},
    {"id": "Meta-Llama-3.3-70B-Instruct-Turbo", "name": "Meta-Llama-3.3-70B-Instruct-Turbo"},
    {"id": "Mistral-7B-Instruct-v0.2", "name": "Mistral-7B-Instruct-v0.2"},
    {"id": "deepseek-llm-67b-chat", "name": "deepseek-llm-67b-chat"},
    {"id": "dbrx-instruct", "name": "dbrx-instruct"},
    {"id": "Meta-Llama-3.1-405B-Instruct-Turbo", "name": "Meta-Llama-3.1-405B-Instruct-Turbo"},
    {"id": "Qwen-QwQ-32B-Preview", "name": "Qwen-QwQ-32B-Preview"},
    {"id": "Nous-Hermes-2-Mixtral-8x7B-DPO", "name": "Nous-Hermes-2-Mixtral-8x7B-DPO"},
    {"id": "PythonAgent", "name": "python"},
    {"id": "JavaAgent", "name": "java"},
    {"id": "JavaScriptAgent", "name": "javascript"},
    {"id": "HTMLAgent", "name": "html"},
    {"id": "GoogleCloudAgent", "name": "googlecloud"},
    {"id": "AndroidDeveloper", "name": "androiddeveloper"},
    {"id": "SwiftDeveloper", "name": "swiftdeveloper"},
    {"id": "Next.jsAgent", "name": "next.js"},
    {"id": "MongoDBAgent", "name": "mongodb"},
    {"id": "PyTorchAgent", "name": "pytorch"},
    {"id": "ReactAgent", "name": "react"},
    {"id": "XcodeAgent", "name": "xcode"},
    {"id": "AngularJSAgent", "name": "angularjs"},
    {"id": "HerokuAgent", "name": "heroku"},
    {"id": "GodotAgent", "name": "godot"},
    {"id": "GoAgent", "name": "go"},
    {"id": "GitlabAgent", "name": "gitlab"},
    {"id": "GitAgent", "name": "git"},
    {"id": "RepoMap", "name": "repomap"},
    {"id": "gemini-1.5-pro-latest", "name": "gemini-flash-2.0"},
    {"id": "gemini-1.5-pro", "name": "gemini-flash-2.0"},
    {"id": "claude-3-5-sonnet-20240620", "name": "claude-3-5-sonnet"},
    {"id": "claude-3-5-sonnet", "name": "claude-3-5-sonnet"},
    {"id": "Niansuh", "name": "niansuh"},
    {"id": "o1-preview", "name": "o1-preview"},
    {"id": "Claude-sonnet-3.7", "name": "Claude-sonnet-3.7"},
    {"id": "claude-3-5-sonnet-x", "name": "claude-3-5-sonnet-x"},
    {"id": "gpt-3.5-turbo", "name": "gpt-3.5-turbo"},
    {"id": "gpt-3.5-turbo-202201", "name": "gpt-3.5-turbo-202201"},

    # Added New Agents
    {"id": "FlaskAgent", "name": "flask"},
    {"id": "FirebaseAgent", "name": "firebase"},
    {"id": "FastAPIAgent", "name": "fastapi"},
    {"id": "ErlangAgent", "name": "erlang"},
    {"id": "ElectronAgent", "name": "electron"},
    {"id": "DockerAgent", "name": "docker"},
    {"id": "DigitalOceanAgent", "name": "digitalocean"},
    {"id": "BitbucketAgent", "name": "bitbucket"},
    {"id": "AzureAgent", "name": "azure"},
    {"id": "FlutterAgent", "name": "flutter"},
    {"id": "YoutubeAgent", "name": "youtube"},
    {"id": "builderAgent", "name": "builder"},
    {"id": "TirexAi", "name": "TirexAi"},
    {"id": "o1", "name": "o1"},



    # Added New models

    {"id": "openai/gpt-4.1", "name": "openai/gpt-4.1" },
{"id": "x-ai/grok-3-beta", "name": "x-ai/grok-3-beta" },
{"id": "anthropic/claude-3.7-sonnet", "name": "anthropic/claude-3.7-sonnet" },
{"id": "google/gemini-2.5-pro-exp-03-25:free", "name": "google/gemini-2.5-pro-exp-03-25:free" },
{"id": "deepseek/deepseek-r1", "name": "deepseek/deepseek-r1" },
{"id": "meta-llama/llama-4-maverick:free", "name": "meta-llama/llama-4-maverick:free" },
{"id": "mistralai/mistral-large", "name": "mistralai/mistral-large" },
{"id": "openai/chatgpt-4o-latest", "name": "openai/chatgpt-4o-latest" },
{"id": "qwen/qwen-2.5-coder-32b-instruct", "name": "qwen/qwen-2.5-coder-32b-instruct" },
{"id": "openai/gpt-4.1-mini", "name": "openai/gpt-4.1-mini" },
{"id": "openai/gpt-4.1-nano", "name": "openai/gpt-4.1-nano" },
{"id": "anthropic/claude-3.7-sonnet:thinking", "name": "anthropic/claude-3.7-sonnet:thinking" },
{"id": "anthropic/claude-3.7-sonnet:beta", "name": "anthropic/claude-3.7-sonnet:beta" },
{"id": "anthropic/claude-3.5-haiku:beta", "name": "anthropic/claude-3.5-haiku:beta" },
{"id": "anthropic/claude-3.5-haiku", "name": "anthropic/claude-3.5-haiku" },
{"id": "anthropic/claude-3.5-haiku-20241022:beta", "name": "anthropic/claude-3.5-haiku-20241022:beta" },
{"id": "anthropic/claude-3.5-haiku-20241022", "name": "anthropic/claude-3.5-haiku-20241022" },
{"id": "anthropic/claude-3.5-sonnet:beta", "name": "anthropic/claude-3.5-sonnet:beta" },
{"id": "anthropic/claude-3.5-sonnet", "name": "anthropic/claude-3.5-sonnet" },
{"id": "x-ai/grok-3-mini-beta", "name": "x-ai/grok-3-mini-beta" },
{"id": "google/gemini-2.0-flash-lite-001", "name": "google/gemini-2.0-flash-lite-001" },
{"id": "meta-llama/llama-4-maverick", "name": "meta-llama/llama-4-maverick" },
{"id": "meta-llama/llama-4-scout:free", "name": "meta-llama/llama-4-scout:free" },
{"id": "meta-llama/llama-4-scout", "name": "meta-llama/llama-4-scout" },
{"id": "nvidia/llama-3.1-nemotron-70b-instruct:free", "name": "nvidia/llama-3.1-nemotron-70b-instruct:free" },
{"id": "nvidia/llama-3.1-nemotron-70b-instruct", "name": "nvidia/llama-3.1-nemotron-70b-instruct" },
{"id": "x-ai/grok-2-vision-1212", "name": "x-ai/grok-2-vision-1212" },
{"id": "x-ai/grok-2-1212", "name": "x-ai/grok-2-1212" },
{"id": "eleutherai/llemma_7b", "name": "eleutherai/llemma_7b" },
{"id": "alfredpros/codellama-7b-instruct-solidity", "name": "alfredpros/codellama-7b-instruct-solidity" },
{"id": "arliai/qwq-32b-arliai-rpr-v1:free", "name": "arliai/qwq-32b-arliai-rpr-v1:free" },
{"id": "agentica-org/deepcoder-14b-preview:free", "name": "agentica-org/deepcoder-14b-preview:free" },
{"id": "moonshotai/kimi-vl-a3b-thinking:free", "name": "moonshotai/kimi-vl-a3b-thinking:free" },
{"id": "openrouter/optimus-alpha", "name": "openrouter/optimus-alpha" },
{"id": "nvidia/llama-3.1-nemotron-nano-8b-v1:free", "name": "nvidia/llama-3.1-nemotron-nano-8b-v1:free" },
{"id": "nvidia/llama-3.3-nemotron-super-49b-v1:free", "name": "nvidia/llama-3.3-nemotron-super-49b-v1:free" },
{"id": "nvidia/llama-3.1-nemotron-ultra-253b-v1:free", "name": "nvidia/llama-3.1-nemotron-ultra-253b-v1:free" },
{"id": "tokyotech-llm/llama-3.1-swallow-8b-instruct-v0.3", "name": "tokyotech-llm/llama-3.1-swallow-8b-instruct-v0.3" },
{"id": "openrouter/quasar-alpha", "name": "openrouter/quasar-alpha" },
{"id": "all-hands/openhands-lm-32b-v0.1", "name": "all-hands/openhands-lm-32b-v0.1" },
{"id": "mistral/ministral-8b", "name": "mistral/ministral-8b" },
{"id": "deepseek/deepseek-v3-base:free", "name": "deepseek/deepseek-v3-base:free" },
{"id": "scb10x/llama3.1-typhoon2-8b-instruct", "name": "scb10x/llama3.1-typhoon2-8b-instruct" },
{"id": "scb10x/llama3.1-typhoon2-70b-instruct", "name": "scb10x/llama3.1-typhoon2-70b-instruct" },
{"id": "allenai/molmo-7b-d:free", "name": "allenai/molmo-7b-d:free" },
{"id": "bytedance-research/ui-tars-72b:free", "name": "bytedance-research/ui-tars-72b:free" },
{"id": "qwen/qwen2.5-vl-3b-instruct:free", "name": "qwen/qwen2.5-vl-3b-instruct:free" },
{"id": "qwen/qwen2.5-vl-32b-instruct:free", "name": "qwen/qwen2.5-vl-32b-instruct:free" },
{"id": "qwen/qwen2.5-vl-32b-instruct", "name": "qwen/qwen2.5-vl-32b-instruct" },
{"id": "deepseek/deepseek-chat-v3-0324:free", "name": "deepseek/deepseek-chat-v3-0324:free" },
{"id": "deepseek/deepseek-chat-v3-0324", "name": "deepseek/deepseek-chat-v3-0324" },
{"id": "featherless/qwerky-72b:free", "name": "featherless/qwerky-72b:free" },
{"id": "openai/o1-pro", "name": "openai/o1-pro" },
{"id": "mistralai/mistral-small-3.1-24b-instruct:free", "name": "mistralai/mistral-small-3.1-24b-instruct:free" },
{"id": "mistralai/mistral-small-3.1-24b-instruct", "name": "mistralai/mistral-small-3.1-24b-instruct" },
{"id": "open-r1/olympiccoder-7b:free", "name": "open-r1/olympiccoder-7b:free" },
{"id": "open-r1/olympiccoder-32b:free", "name": "open-r1/olympiccoder-32b:free" },
{"id": "steelskull/l3.3-electra-r1-70b", "name": "steelskull/l3.3-electra-r1-70b" },
{"id": "allenai/olmo-2-0325-32b-instruct", "name": "allenai/olmo-2-0325-32b-instruct" },
{"id": "google/gemma-3-1b-it:free", "name": "google/gemma-3-1b-it:free" },
{"id": "google/gemma-3-4b-it:free", "name": "google/gemma-3-4b-it:free" },
{"id": "google/gemma-3-4b-it", "name": "google/gemma-3-4b-it" },
{"id": "ai21/jamba-1.6-large", "name": "ai21/jamba-1.6-large" },
{"id": "ai21/jamba-1.6-mini", "name": "ai21/jamba-1.6-mini" },
{"id": "google/gemma-3-12b-it:free", "name": "google/gemma-3-12b-it:free" },
{"id": "google/gemma-3-12b-it", "name": "google/gemma-3-12b-it" },
{"id": "cohere/command-a", "name": "cohere/command-a" },
{"id": "openai/gpt-4o-mini-search-preview", "name": "openai/gpt-4o-mini-search-preview" },
{"id": "openai/gpt-4o-search-preview", "name": "openai/gpt-4o-search-preview" },
{"id": "rekaai/reka-flash-3:free", "name": "rekaai/reka-flash-3:free" },
{"id": "thedrummer/anubis-pro-105b-v1", "name": "thedrummer/anubis-pro-105b-v1" },
{"id": "latitudegames/wayfarer-large-70b-llama-3.3", "name": "latitudegames/wayfarer-large-70b-llama-3.3" },
{"id": "thedrummer/skyfall-36b-v2", "name": "thedrummer/skyfall-36b-v2" },
{"id": "microsoft/phi-4-multimodal-instruct", "name": "microsoft/phi-4-multimodal-instruct" },
{"id": "deepseek/deepseek-r1-zero:free", "name": "deepseek/deepseek-r1-zero:free" },
{"id": "qwen/qwq-32b:free", "name": "qwen/qwq-32b:free" },
{"id": "qwen/qwq-32b", "name": "qwen/qwq-32b" },
{"id": "qwen/qwen2.5-32b-instruct", "name": "qwen/qwen2.5-32b-instruct" },
{"id": "moonshotai/moonlight-16b-a3b-instruct:free", "name": "moonshotai/moonlight-16b-a3b-instruct:free" },
{"id": "nousresearch/deephermes-3-llama-3-8b-preview:free", "name": "nousresearch/deephermes-3-llama-3-8b-preview:free" },
{"id": "openai/gpt-4.5-preview", "name": "openai/gpt-4.5-preview" },
{"id": "mistralai/mistral-saba", "name": "mistralai/mistral-saba" },
{"id": "cognitivecomputations/dolphin3.0-r1-mistral-24b:free", "name": "cognitivecomputations/dolphin3.0-r1-mistral-24b:free" },
{"id": "cognitivecomputations/dolphin3.0-mistral-24b:free", "name": "cognitivecomputations/dolphin3.0-mistral-24b:free" },
{"id": "meta-llama/llama-guard-3-8b", "name": "meta-llama/llama-guard-3-8b" },
{"id": "openai/o3-mini-high", "name": "openai/o3-mini-high" },
{"id": "allenai/llama-3.1-tulu-3-405b", "name": "allenai/llama-3.1-tulu-3-405b" },
{"id": "deepseek/deepseek-r1-distill-llama-8b", "name": "deepseek/deepseek-r1-distill-llama-8b" },
{"id": "google/gemini-2.0-flash-001", "name": "google/gemini-2.0-flash-001" },
{"id": "qwen/qwen-vl-plus", "name": "qwen/qwen-vl-plus" },
{"id": "openai/gpt-4o-mini-2024-07-18", "name": "openai/gpt-4o-mini-2024-07-18" },
{"id": "openai/gpt-4o-mini", "name": "openai/gpt-4o-mini" },
{"id": "qwen/qwen-2-7b-instruct", "name": "qwen/qwen-2-7b-instruct" },
{"id": "google/gemini-2-27b-it", "name": "google/gemini-2-27b-it" },
{"id": "alpindale/magnum-72b", "name": "alpindale/magnum-72b" },
{"id": "nousresearch/hermes-2-theta-llama-3-8b", "name": "nousresearch/hermes-2-theta-llama-3-8b" },
{"id": "google/gemini-2-9b-it:free", "name": "google/gemini-2-9b-it:free" },
{"id": "google/gemini-2-9b-it", "name": "google/gemini-2-9b-it" },
{"id": "sao10k/l3-stheno-8b", "name": "sao10k/l3-stheno-8b" },
{"id": "ai21/jamba-instruct", "name": "ai21/jamba-instruct" },
{"id": "01-ai/yi-large", "name": "01-ai/yi-large" },
{"id": "meta-llama/llama-3.1-405b", "name": "meta-llama/llama-3.1-405b" },
{"id": "nothingiisreal/mn-celeste-12b", "name": "nothingiisreal/mn-celeste-12b" },
{"id": "01-ai/yi-vision", "name": "01-ai/yi-vision" },
{"id": "01-ai/yi-large-turbo", "name": "01-ai/yi-large-turbo" },
{"id": "google/gemini-pro-1.5-exp", "name": "google/gemini-pro-1.5-exp" },
{"id": "meta-llama/llama-3.1-70b-instruct", "name": "meta-llama/llama-3.1-70b-instruct" },
{"id": "meta-llama/llama-3.1-8b-instruct:free", "name": "meta-llama/llama-3.1-8b-instruct:free" },
{"id": "meta-llama/llama-3.1-8b-instruct", "name": "meta-llama/llama-3.1-8b-instruct" },
{"id": "meta-llama/llama-3.1-405b-instruct", "name": "meta-llama/llama-3.1-405b-instruct" },
{"id": "cognitivecomputations/dolphin-llama-3-70b", "name": "cognitivecomputations/dolphin-llama-3-70b" },
{"id": "mistralai/mistral-nemo:free", "name": "mistralai/mistral-nemo:free" },
{"id": "mistralai/mistral-nemo", "name": "mistralai/mistral-nemo" },
{"id": "mistralai/codestral-mamba", "name": "mistralai/codestral-mamba" },
{"id": "x-ai/grok-vision-beta", "name": "x-ai/grok-vision-beta" },
]

MODEL_MAPPING = {
    "blackboxai": "blackboxai",
    "anthropic/claude-3.7-sonnet:thinking": "anthropic/claude-3.7-sonnet:thinking",
    "niansuh-t1": "niansuh-t1",
    "Gemini-PRO": "Gemini-PRO",
    "blackboxai-pro": "blackboxai-pro",
    "gpt-4o": "gpt-4o",
    "gemini-flash-2.0": "gemini-flash-2.0",
    "chatgpt-4o-latest": "chatgpt-4o-latest",
    "o1": "o1",
    "Llama-4-Maverick-17B-128E": "Llama-4-Maverick-17B-128E",
    "gemini-1.5-flash": "gemini-flash-2.0",
    "deepseek-r1": "deepseek-r1",
    "deepseek-v3": "deepseek-v3",
    "Mistral-Small-24B-Instruct-2501": "Mistral-Small-24B-Instruct-2501",
    "Meta-Llama-3.3-70B-Instruct-Turbo": "Meta-Llama-3.3-70B-Instruct-Turbo",
    "Mistral-7B-Instruct-v0.2": "Mistral-7B-Instruct-v0.2",
    "deepseek-llm-67b-chat": "deepseek-llm-67b-chat",
    "dbrx-instruct": "dbrx-instruct",
    "Meta-Llama-3.1-405B-Instruct-Turbo": "Meta-Llama-3.1-405B-Instruct-Turbo",
    "Qwen-QwQ-32B-Preview": "Qwen-QwQ-32B-Preview",
    "Nous-Hermes-2-Mixtral-8x7B-DPO": "Nous-Hermes-2-Mixtral-8x7B-DPO",
    # Additional mappings
    "gemini-flash": "gemini-flash-2.0",
    "claude-sonnet-3.5": "Claude-Sonnet-3.5",
    "gemini-1.5-pro-latest": "gemini-flash-2.0",
    "gemini-1.5-pro": "gemini-flash-2.0",
    "claude-3-5-sonnet-20240620": "Claude-Sonnet-3.5",
    "Niansuh": "niansuh",
    "o1-preview": "o1-preview",
    "Claude-sonnet-3.7": "Claude-sonnet-3.7",
    "claude-3-5-sonnet-x": "claude-3-5-sonnet-x",
    "gpt-3.5-turbo": "gpt-3.5-turbo",
    "gpt-3.5-turbo-202201": "gpt-3.5-turbo-202201",

    # Added New Agents
    "FlaskAgent": "flask",
    "FirebaseAgent": "firebase",
    "FastAPIAgent": "fastapi",
    "ErlangAgent": "erlang",
    "ElectronAgent": "electron",
    "DockerAgent": "docker",
    "DigitalOceanAgent": "digitalocean",
    "BitbucketAgent": "bitbucket",
    "AzureAgent": "azure",
    "FlutterAgent": "flutter",
    "YoutubeAgent": "youtube",
    "builderAgent": "builder",
    "TirexAi": "TirexAi",
    "o3-mini": "o3-mini",


    # Added new Models

    "openai/gpt-4.1": "openai/gpt-4.1",
    "x-ai/grok-3-beta": "x-ai/grok-3-beta",
    "anthropic/claude-3.7-sonnet": "anthropic/claude-3.7-sonnet",
    "google/gemini-2.5-pro-exp-03-25:free": "google/gemini-2.5-pro-exp-03-25:free",
    "deepseek/deepseek-r1": "deepseek/deepseek-r1",
    "meta-llama/llama-4-maverick:free": "meta-llama/llama-4-maverick:free",
    "mistralai/mistral-large": "mistralai/mistral-large",
    "openai/chatgpt-4o-latest": "openai/chatgpt-4o-latest",
    "qwen/qwen-2.5-coder-32b-instruct": "qwen/qwen-2.5-coder-32b-instruct",
    "openai/gpt-4.1-mini": "openai/gpt-4.1-mini",
    "openai/gpt-4.1-nano": "openai/gpt-4.1-nano",
    "anthropic/claude-3.7-sonnet:thinking": "anthropic/claude-3.7-sonnet:thinking",
    "anthropic/claude-3.7-sonnet:beta": "anthropic/claude-3.7-sonnet:beta",
    "anthropic/claude-3.5-haiku:beta": "anthropic/claude-3.5-haiku:beta",
    "anthropic/claude-3.5-haiku": "anthropic/claude-3.5-haiku",
    "anthropic/claude-3.5-haiku-20241022:beta": "anthropic/claude-3.5-haiku-20241022:beta",
    "anthropic/claude-3.5-haiku-20241022": "anthropic/claude-3.5-haiku-20241022",
    "anthropic/claude-3.5-sonnet:beta": "anthropic/claude-3.5-sonnet:beta",
    "anthropic/claude-3.5-sonnet": "anthropic/claude-3.5-sonnet",
    "x-ai/grok-3-mini-beta": "x-ai/grok-3-mini-beta",
    "google/gemini-2.0-flash-lite-001": "google/gemini-2.0-flash-lite-001",
    "meta-llama/llama-4-maverick": "meta-llama/llama-4-maverick",
    "meta-llama/llama-4-scout:free": "meta-llama/llama-4-scout:free",
    "meta-llama/llama-4-scout": "meta-llama/llama-4-scout",
    "nvidia/llama-3.1-nemotron-70b-instruct:free": "nvidia/llama-3.1-nemotron-70b-instruct:free",
    "nvidia/llama-3.1-nemotron-70b-instruct": "nvidia/llama-3.1-nemotron-70b-instruct",
    "x-ai/grok-2-vision-1212": "x-ai/grok-2-vision-1212",
    "x-ai/grok-2-1212": "x-ai/grok-2-1212",
    "eleutherai/llemma_7b": "eleutherai/llemma_7b",
    "alfredpros/codellama-7b-instruct-solidity": "alfredpros/codellama-7b-instruct-solidity",
    "arliai/qwq-32b-arliai-rpr-v1:free": "arliai/qwq-32b-arliai-rpr-v1:free",
    "agentica-org/deepcoder-14b-preview:free": "agentica-org/deepcoder-14b-preview:free",
    "moonshotai/kimi-vl-a3b-thinking:free": "moonshotai/kimi-vl-a3b-thinking:free",
    "openrouter/optimus-alpha": "openrouter/optimus-alpha",
    "nvidia/llama-3.1-nemotron-nano-8b-v1:free": "nvidia/llama-3.1-nemotron-nano-8b-v1:free",
    "nvidia/llama-3.3-nemotron-super-49b-v1:free": "nvidia/llama-3.3-nemotron-super-49b-v1:free",
    "nvidia/llama-3.1-nemotron-ultra-253b-v1:free": "nvidia/llama-3.1-nemotron-ultra-253b-v1:free",
    "tokyotech-llm/llama-3.1-swallow-8b-instruct-v0.3": "tokyotech-llm/llama-3.1-swallow-8b-instruct-v0.3",
    "openrouter/quasar-alpha": "openrouter/quasar-alpha",
    "all-hands/openhands-lm-32b-v0.1": "all-hands/openhands-lm-32b-v0.1",
    "mistral/ministral-8b": "mistral/ministral-8b",
    "deepseek/deepseek-v3-base:free": "deepseek/deepseek-v3-base:free",
    "scb10x/llama3.1-typhoon2-8b-instruct": "scb10x/llama3.1-typhoon2-8b-instruct",
    "scb10x/llama3.1-typhoon2-70b-instruct": "scb10x/llama3.1-typhoon2-70b-instruct",
    "allenai/molmo-7b-d:free": "allenai/molmo-7b-d:free",
    "bytedance-research/ui-tars-72b:free": "bytedance-research/ui-tars-72b:free",
    "qwen/qwen2.5-vl-3b-instruct:free": "qwen/qwen2.5-vl-3b-instruct:free",
    "qwen/qwen2.5-vl-32b-instruct:free": "qwen/qwen2.5-vl-32b-instruct:free",
    "qwen/qwen2.5-vl-32b-instruct": "qwen/qwen2.5-vl-32b-instruct",
    "deepseek/deepseek-chat-v3-0324:free": "deepseek/deepseek-chat-v3-0324:free",
    "deepseek/deepseek-chat-v3-0324": "deepseek/deepseek-chat-v3-0324",
    "featherless/qwerky-72b:free": "featherless/qwerky-72b:free",
    "openai/o1-pro": "openai/o1-pro",
    "mistralai/mistral-small-3.1-24b-instruct:free": "mistralai/mistral-small-3.1-24b-instruct:free",
    "mistralai/mistral-small-3.1-24b-instruct": "mistralai/mistral-small-3.1-24b-instruct",
    "open-r1/olympiccoder-7b:free": "open-r1/olympiccoder-7b:free",
    "open-r1/olympiccoder-32b:free": "open-r1/olympiccoder-32b:free",
    "steelskull/l3.3-electra-r1-70b": "steelskull/l3.3-electra-r1-70b",
    "allenai/olmo-2-0325-32b-instruct": "allenai/olmo-2-0325-32b-instruct",
    "google/gemma-3-1b-it:free": "google/gemma-3-1b-it:free",
    "google/gemma-3-4b-it:free": "google/gemma-3-4b-it:free",
    "google/gemma-3-4b-it": "google/gemma-3-4b-it",
    "ai21/jamba-1.6-large": "ai21/jamba-1.6-large",
    "ai21/jamba-1.6-mini": "ai21/jamba-1.6-mini",
    "google/gemma-3-12b-it:free": "google/gemma-3-12b-it:free",
    "google/gemma-3-12b-it": "google/gemma-3-12b-it",
    "cohere/command-a": "cohere/command-a",
    "openai/gpt-4o-mini-search-preview": "openai/gpt-4o-mini-search-preview",
    "openai/gpt-4o-search-preview": "openai/gpt-4o-search-preview",
    "rekaai/reka-flash-3:free": "rekaai/reka-flash-3:free",
    "thedrummer/anubis-pro-105b-v1": "thedrummer/anubis-pro-105b-v1",
    "latitudegames/wayfarer-large-70b-llama-3.3": "latitudegames/wayfarer-large-70b-llama-3.3",
    "thedrummer/skyfall-36b-v2": "thedrummer/skyfall-36b-v2",
    "microsoft/phi-4-multimodal-instruct": "microsoft/phi-4-multimodal-instruct",
    "deepseek/deepseek-r1-zero:free": "deepseek/deepseek-r1-zero:free",
    "qwen/qwq-32b:free": "qwen/qwq-32b:free",
    "qwen/qwq-32b": "qwen/qwq-32b",
    "qwen/qwen2.5-32b-instruct": "qwen/qwen2.5-32b-instruct",
    "moonshotai/moonlight-16b-a3b-instruct:free": "moonshotai/moonlight-16b-a3b-instruct:free",
    "nousresearch/deephermes-3-llama-3-8b-preview:free": "nousresearch/deephermes-3-llama-3-8b-preview:free",
    "openai/gpt-4.5-preview": "openai/gpt-4.5-preview",
    "mistralai/mistral-saba": "mistralai/mistral-saba",
    "cognitivecomputations/dolphin3.0-r1-mistral-24b:free": "cognitivecomputations/dolphin3.0-r1-mistral-24b:free",
    "cognitivecomputations/dolphin3.0-mistral-24b:free": "cognitivecomputations/dolphin3.0-mistral-24b:free",
    "meta-llama/llama-guard-3-8b": "meta-llama/llama-guard-3-8b",
    "openai/o3-mini-high": "openai/o3-mini-high",
    "allenai/llama-3.1-tulu-3-405b": "allenai/llama-3.1-tulu-3-405b",
    "deepseek/deepseek-r1-distill-llama-8b": "deepseek/deepseek-r1-distill-llama-8b",
    "google/gemini-2.0-flash-001": "google/gemini-2.0-flash-001",
    "qwen/qwen-vl-plus": "qwen/qwen-vl-plus",
    "openai/gpt-4o-mini-2024-07-18": "openai/gpt-4o-mini-2024-07-18",
    "openai/gpt-4o-mini": "openai/gpt-4o-mini",
    "qwen/qwen-2-7b-instruct": "qwen/qwen-2-7b-instruct",
    "google/gemini-2-27b-it": "google/gemini-2-27b-it",
    "alpindale/magnum-72b": "alpindale/magnum-72b",
    "nousresearch/hermes-2-theta-llama-3-8b": "nousresearch/hermes-2-theta-llama-3-8b",
    "google/gemini-2-9b-it:free": "google/gemini-2-9b-it:free",
    "google/gemini-2-9b-it": "google/gemini-2-9b-it",
    "sao10k/l3-stheno-8b": "sao10k/l3-stheno-8b",
    "ai21/jamba-instruct": "ai21/jamba-instruct",
    "01-ai/yi-large": "01-ai/yi-large",
    "meta-llama/llama-3.1-405b": "meta-llama/llama-3.1-405b",
    "nothingiisreal/mn-celeste-12b": "nothingiisreal/mn-celeste-12b",
    "01-ai/yi-vision": "01-ai/yi-vision",
    "01-ai/yi-large-turbo": "01-ai/yi-large-turbo",
    "google/gemini-pro-1.5-exp": "google/gemini-pro-1.5-exp",
    "meta-llama/llama-3.1-70b-instruct": "meta-llama/llama-3.1-70b-instruct",
    "meta-llama/llama-3.1-8b-instruct:free": "meta-llama/llama-3.1-8b-instruct:free",
    "meta-llama/llama-3.1-8b-instruct": "meta-llama/llama-3.1-8b-instruct",
    "meta-llama/llama-3.1-405b-instruct": "meta-llama/llama-3.1-405b-instruct",
    "cognitivecomputations/dolphin-llama-3-70b": "cognitivecomputations/dolphin-llama-3-70b",
    "mistralai/mistral-nemo:free": "mistralai/mistral-nemo:free",
    "mistralai/mistral-nemo": "mistralai/mistral-nemo",
    "mistralai/codestral-mamba": "mistralai/codestral-mamba",
    "x-ai/grok-vision-beta": "x-ai/grok-vision-beta",

}

# Agent modes
AGENT_MODE = {
    'claude-sonnet-3.5': {'mode': True, 'id': "claude-sonnet-3.5", 'name': "claude-sonnet-3.5"},
    'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
    'gpt-4o': {'mode': True, 'id': "GPT-4o", 'name': "GPT-4o"},
    'Llama-4-Maverick-17B-128E': {'mode': True, 'id': "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8", 'name': "Llama-4-Maverick-17B-128E"},
    'o3-mini': {'mode': True, 'id': "o3-mini", 'name': "o3-mini"},
    'o1': {'mode': True, 'id': "o1", 'name': "o1"},
    'gemini-flash-2.0': {'mode': True, 'id': "Gemini/Gemini-Flash-2.0", 'name': "gemini-flash-2.0"},
    'Claude-Sonnet-3.7': {'mode': True, 'id': "Claude-Sonnet-3.7", 'name': "Claude-Sonnet-3.7"},
    'niansuh-t1': {'mode': True, 'id': "Niansuh-T1ybc2YUh", 'name': "niansuh-t1"},
    'chatgpt-4o-latest': {'mode': True, 'id': "GPT-4oyTDtGrw", 'name': "chatgpt-4o-latest"},
    'TirexAi': {'mode': True, 'id': "TierXxsRieQK", 'name': "TirexAi"},
    'Niansuh': {'mode': True, 'id': "NiansuhAIk1HgESy", 'name': "niansuh"},
    'snapzionai': {'mode': True, 'id': "Prompt-AnalyzerpINlz9W", 'name': "snapzionai"},
    'o1-preview': {'mode': True, 'id': "o1Dst8La8", 'name': "o1-preview"},
    'claude-3-5-sonnet-20241022': {'mode': True, 'id': "Claude-Sonnet-3.5zO2HZSF", 'name': "claude-3-5-sonnet-20241022"},
    'claude-3-5-sonnet-x': {'mode': True, 'id': "Claude-Sonnet-3.52022JE0UdQ3", 'name': "claude-3-5-sonnet-x"},
    'gpt-3.5-turbo': {'mode': True, 'id': "GPT-3.5-TurboYxtGz0H", 'name': "gpt-3.5-turbo"},
    'gpt-3.5-turbo-202201': {'mode': True, 'id': "GPT-3.5-Turbo-202201PNWREyV", 'name': "gpt-3.5-turbo-202201"},
    'deepseek-r1': {'mode': True, 'id': "deepseek-reasoner", 'name': "deepseek-r1"},
    'deepseek-v3': {'mode': True, 'id': "deepseek-chat", 'name': "deepseek-v3"},
    'Meta-Llama-3.3-70B-Instruct-Turbo': {'mode': True, 'id': "meta-llama/Llama-3.3-70B-Instruct-Turbo", 'name': "Meta-Llama-3.3-70B-Instruct-Turbo"},
    'Mistral-7B-Instruct-v0.2': {'mode': True, 'id': "mistralai/Mistral-7B-Instruct-v0.2", 'name': "Mistral-7B-Instruct-v0.2"},
    'Mistral-Small-24B-Instruct-2501': {'mode': True, 'id': "mistralai/Mistral-Small-24B-Instruct-2501", 'name': "Mistral-Small-24B-Instruct-2501"},
    'deepseek-llm-67b-chat': {'mode': True, 'id': "deepseek-ai/deepseek-llm-67b-chat", 'name': "deepseek-llm-67b-chat"},
    'dbrx-instruct': {'mode': True, 'id': "databricks/dbrx-instruct", 'name': "dbrx-instruct"},
    'Meta-Llama-3.1-405B-Instruct-Turbo': {'mode': True, 'id': "meta-llama/Meta-Llama-3.1-405B-Instruct-Lite-Pro", 'name': "Meta-Llama-3.1-405B-Instruct-Turbo"},
    'Qwen-QwQ-32B-Preview': {'mode': True, 'id': "Qwen/QwQ-32B-Preview", 'name': "Qwen-QwQ-32B-Preview"},
    'Nous-Hermes-2-Mixtral-8x7B-DPO': {'mode': True, 'id': "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO", 'name': "Nous-Hermes-2-Mixtral-8x7B-DPO"},


    # Added New Models

    'openai/gpt-4.1': {'mode': True, 'id': "openai/gpt-4.1", 'name': "openai/gpt-4.1"},
    'openai/gpt-4o-search-preview': {'mode': True, 'id': "openai/gpt-4o-search-preview", 'name': "openai/gpt-4o-search-preview"},
    'openai/gpt-4o-mini-search-preview': {'mode': True, 'id': "openai/gpt-4o-mini-search-preview", 'name': "openai/gpt-4o-mini-search-preview"},
'x-ai/grok-3-beta': {'mode': True, 'id': "x-ai/grok-3-beta", 'name': "x-ai/grok-3-beta"},
'anthropic/claude-3.7-sonnet': {'mode': True, 'id': "anthropic/claude-3.7-sonnet", 'name': "anthropic/claude-3.7-sonnet"},
'google/gemini-2.5-pro-exp-03-25:free': {'mode': True, 'id': "google/gemini-2.5-pro-exp-03-25:free", 'name': "google/gemini-2.5-pro-exp-03-25:free"},
'deepseek/deepseek-r1': {'mode': True, 'id': "deepseek/deepseek-r1", 'name': "deepseek/deepseek-r1"},
'meta-llama/llama-4-maverick:free': {'mode': True, 'id': "meta-llama/llama-4-maverick:free", 'name': "meta-llama/llama-4-maverick:free"},
'mistralai/mistral-large': {'mode': True, 'id': "mistralai/mistral-large", 'name': "mistralai/mistral-large"},
'openai/chatgpt-4o-latest': {'mode': True, 'id': "openai/chatgpt-4o-latest", 'name': "openai/chatgpt-4o-latest"},
'qwen/qwen-2.5-coder-32b-instruct': {'mode': True, 'id': "qwen/qwen-2.5-coder-32b-instruct", 'name': "qwen/qwen-2.5-coder-32b-instruct"},
'openai/gpt-4.1-mini': {'mode': True, 'id': "openai/gpt-4.1-mini", 'name': "openai/gpt-4.1-mini"},
'openai/gpt-4.1-nano': {'mode': True, 'id': "openai/gpt-4.1-nano", 'name': "openai/gpt-4.1-nano"},
'anthropic/claude-3.7-sonnet:thinking': {'mode': True, 'id': "anthropic/claude-3.7-sonnet:thinking", 'name': "anthropic/claude-3.7-sonnet:thinking"},
'anthropic/claude-3.7-sonnet:beta': {'mode': True, 'id': "anthropic/claude-3.7-sonnet:beta", 'name': "anthropic/claude-3.7-sonnet:beta"},
'anthropic/claude-3.5-haiku:beta': {'mode': True, 'id': "anthropic/claude-3.5-haiku:beta", 'name': "anthropic/claude-3.5-haiku:beta"},
'anthropic/claude-3.5-haiku': {'mode': True, 'id': "anthropic/claude-3.5-haiku", 'name': "anthropic/claude-3.5-haiku"},
'anthropic/claude-3.5-haiku-20241022:beta': {'mode': True, 'id': "anthropic/claude-3.5-haiku-20241022:beta", 'name': "anthropic/claude-3.5-haiku-20241022:beta"},
'anthropic/claude-3.5-haiku-20241022': {'mode': True, 'id': "anthropic/claude-3.5-haiku-20241022", 'name': "anthropic/claude-3.5-haiku-20241022"},
'anthropic/claude-3.5-sonnet:beta': {'mode': True, 'id': "anthropic/claude-3.5-sonnet:beta", 'name': "anthropic/claude-3.5-sonnet:beta"},
'anthropic/claude-3.5-sonnet': {'mode': True, 'id': "anthropic/claude-3.5-sonnet", 'name': "anthropic/claude-3.5-sonnet"},
'x-ai/grok-3-mini-beta': {'mode': True, 'id': "x-ai/grok-3-mini-beta", 'name': "x-ai/grok-3-mini-beta"},
'google/gemini-2.0-flash-lite-001': {'mode': True, 'id': "google/gemini-2.0-flash-lite-001", 'name': "google/gemini-2.0-flash-lite-001"},
'meta-llama/llama-4-maverick': {'mode': True, 'id': "meta-llama/llama-4-maverick", 'name': "meta-llama/llama-4-maverick"},
'meta-llama/llama-4-scout:free': {'mode': True, 'id': "meta-llama/llama-4-scout:free", 'name': "meta-llama/llama-4-scout:free"},
'meta-llama/llama-4-scout': {'mode': True, 'id': "meta-llama/llama-4-scout", 'name': "meta-llama/llama-4-scout"},
'nvidia/llama-3.1-nemotron-70b-instruct:free': {'mode': True, 'id': "nvidia/llama-3.1-nemotron-70b-instruct:free", 'name': "nvidia/llama-3.1-nemotron-70b-instruct:free"},
'nvidia/llama-3.1-nemotron-70b-instruct': {'mode': True, 'id': "nvidia/llama-3.1-nemotron-70b-instruct", 'name': "nvidia/llama-3.1-nemotron-70b-instruct"},
'x-ai/grok-2-vision-1212': {'mode': True, 'id': "x-ai/grok-2-vision-1212", 'name': "x-ai/grok-2-vision-1212"},
'x-ai/grok-2-1212': {'mode': True, 'id': "x-ai/grok-2-1212", 'name': "x-ai/grok-2-1212"},
'eleutherai/llemma_7b': {'mode': True, 'id': "eleutherai/llemma_7b", 'name': "eleutherai/llemma_7b"},
'alfredpros/codellama-7b-instruct-solidity': {'mode': True, 'id': "alfredpros/codellama-7b-instruct-solidity", 'name': "alfredpros/codellama-7b-instruct-solidity"},
'arliai/qwq-32b-arliai-rpr-v1:free': {'mode': True, 'id': "arliai/qwq-32b-arliai-rpr-v1:free", 'name': "arliai/qwq-32b-arliai-rpr-v1:free"},
'agentica-org/deepcoder-14b-preview:free': {'mode': True, 'id': "agentica-org/deepcoder-14b-preview:free", 'name': "agentica-org/deepcoder-14b-preview:free"},
'moonshotai/kimi-vl-a3b-thinking:free': {'mode': True, 'id': "moonshotai/kimi-vl-a3b-thinking:free", 'name': "moonshotai/kimi-vl-a3b-thinking:free"},
'openrouter/optimus-alpha': {'mode': True, 'id': "openrouter/optimus-alpha", 'name': "openrouter/optimus-alpha"},
'nvidia/llama-3.1-nemotron-nano-8b-v1:free': {'mode': True, 'id': "nvidia/llama-3.1-nemotron-nano-8b-v1:free", 'name': "nvidia/llama-3.1-nemotron-nano-8b-v1:free"},



}

TRENDING_AGENT_MODE = {
    "blackboxai": {},
    "gemini-1.5-flash": {'mode': True, 'id': 'gemini'},
    "llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
    'llama-3.1-70b': {'mode': True, 'id': "llama-3.1-70b"},
    'llama-3.1-405b': {'mode': True, 'id': "llama-3.1-405"},
    'blackboxai-pro': {'mode': True, 'id': "blackboxai-pro"},
    'PythonAgent': {'mode': True, 'id': "python"},
    'JavaAgent': {'mode': True, 'id': "java"},
    'JavaScriptAgent': {'mode': True, 'id': "javascript"},
    'HTMLAgent': {'mode': True, 'id': "html"},
    'GoogleCloudAgent': {'mode': True, 'id': "googlecloud"},
    'AndroidDeveloper': {'mode': True, 'id': "android"},
    'SwiftDeveloper': {'mode': True, 'id': "swift"},
    'Next.jsAgent': {'mode': True, 'id': "next.js"},
    'MongoDBAgent': {'mode': True, 'id': "mongodb"},
    'PyTorchAgent': {'mode': True, 'id': "pytorch"},
    'ReactAgent': {'mode': True, 'id': "react"},
    'XcodeAgent': {'mode': True, 'id': "xcode"},
    'AngularJSAgent': {'mode': True, 'id': "angularjs"},
    'HerokuAgent': {'mode': True, 'id': "heroku"},
    'GodotAgent': {'mode': True, 'id': "godot"},
    'GoAgent': {'mode': True, 'id': "go"},
    'GitlabAgent': {'mode': True, 'id': "gitlab"},
    'GitAgent': {'mode': True, 'id': "git"},
    'RepoMap': {'mode': True, 'id': "repomap"},
    'FlaskAgent': {'mode': True, 'id': "flask"},
    'FirebaseAgent': {'mode': True, 'id': "firebase"},
    'FastAPIAgent': {'mode': True, 'id': "fastapi"},
    'ErlangAgent': {'mode': True, 'id': "erlang"},
    'ElectronAgent': {'mode': True, 'id': "electron"},
    'DockerAgent': {'mode': True, 'id': "docker"},
    'DigitalOceanAgent': {'mode': True, 'id': "digitalocean"},
    'BitbucketAgent': {'mode': True, 'id': "bitbucket"},
    'AzureAgent': {'mode': True, 'id': "azure"},
    'FlutterAgent': {'mode': True, 'id': "flutter"},
    'YoutubeAgent': {'mode': True, 'id': "youtube"},
    'builderAgent': {'mode': True, 'id': "builder"},
}

# Model prefixes
MODEL_PREFIXES = {
    'PythonAgent': '@python',
    'JavaAgent': '@java',
    'JavaScriptAgent': '@javascript',
    'HTMLAgent': '@html',
    'GoogleCloudAgent': '@googlecloud',
    'AndroidDeveloper': '@android',
    'SwiftDeveloper': '@swift',
    'Next.jsAgent': '@next.js',
    'MongoDBAgent': '@mongodb',
    'PyTorchAgent': '@pytorch',
    'ReactAgent': '@react',
    'XcodeAgent': '@xcode',
    'AngularJSAgent': '@angularjs',
    'HerokuAgent': '@heroku',
    'GodotAgent': '@godot',
    'GoAgent': '@go',
    'GitlabAgent': '@gitlab',
    'GitAgent': '@git',
    'blackboxai-pro': '@blackboxai-pro',
    'FlaskAgent': '@flask',
    'FirebaseAgent': '@firebase',
    'FastAPIAgent': '@fastapi',
    'ErlangAgent': '@erlang',
    'ElectronAgent': '@electron',
    'DockerAgent': '@docker',
    'DigitalOceanAgent': '@digitalocean',
    'BitbucketAgent': '@bitbucket',
    'AzureAgent': '@azure',
    'FlutterAgent': '@flutter',
    'YoutubeAgent': '@youtube',
    'builderAgent': '@builder',
}

# Model referers
MODEL_REFERERS = {
    "blackboxai": "/?model=blackboxai",
    "blackboxai-pro": "/?model=blackboxai-pro",
}