import os
from dotenv import load_dotenv

load_dotenv()

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

# APP_SECRET
APP_SECRET = os.getenv("APP_SECRET")

# Default Model
DEFAULT_MODEL = "blackboxai"

# Allowed Models (List of Strings)
ALLOWED_MODELS = [
    "blackboxai",
    "blackboxai-pro",
    "gpt-4o",
    "gemini-pro",
    "gemini-1.5-flash",
    "claude-sonnet-3.5",
    "Image Generation",
    "repomap",
    "Python Agent",
    "Java Agent",
    "JavaScript Agent",
    "HTML Agent",
    "Google Cloud Agent",
    "Android Developer",
    "Swift Developer",
    "Next.js Agent",
    "MongoDB Agent",
    "PyTorch Agent",
    "React Agent",
    "Xcode Agent",
    "AngularJS Agent",
    "Heroku Agent",
    "Godot Agent",
    "Go Agent",
    "Gitlab Agent",
    "Git Agent",
    "Flask Agent",
    "Firebase Agent",
    "FastAPI Agent",
    "Erlang Agent",
    "Electron Agent",
    "Docker Agent",
    "DigitalOcean Agent",
    "Bitbucket Agent",
    "Azure Agent",
    "Flutter Agent",
    "Youtube Agent",
    "builder Agent",
]

# Model Mapping
MODEL_MAPPING = {
    "blackboxai": "blackboxai",
    "blackboxai-pro": "blackboxai-pro",
    "gpt-4o": "gpt-4o",
    "gemini-pro": "gemini-pro",
    "gemini-1.5-flash": "gemini-1.5-flash",
    "claude-sonnet-3.5": "claude-sonnet-3.5",
    "gemini-flash": "gemini-1.5-flash",
    "claude-3.5-sonnet": "claude-sonnet-3.5",
    "flux": "Image Generation",
    # Agent Models
    "Image Generation": "Image Generation",
    "repomap": "repomap",
    "Python Agent": "Python Agent",
    "Java Agent": "Java Agent",
    "JavaScript Agent": "JavaScript Agent",
    "HTML Agent": "HTML Agent",
    "Google Cloud Agent": "Google Cloud Agent",
    "Android Developer": "Android Developer",
    "Swift Developer": "Swift Developer",
    "Next.js Agent": "Next.js Agent",
    "MongoDB Agent": "MongoDB Agent",
    "PyTorch Agent": "PyTorch Agent",
    "React Agent": "React Agent",
    "Xcode Agent": "Xcode Agent",
    "AngularJS Agent": "AngularJS Agent",
    "Heroku Agent": "Heroku Agent",
    "Godot Agent": "Godot Agent",
    "Go Agent": "Go Agent",
    "Gitlab Agent": "Gitlab Agent",
    "Git Agent": "Git Agent",
    "Flask Agent": "Flask Agent",
    "Firebase Agent": "Firebase Agent",
    "FastAPI Agent": "FastAPI Agent",
    "Erlang Agent": "Erlang Agent",
    "Electron Agent": "Electron Agent",
    "Docker Agent": "Docker Agent",
    "DigitalOcean Agent": "DigitalOcean Agent",
    "Bitbucket Agent": "Bitbucket Agent",
    "Azure Agent": "Azure Agent",
    "Flutter Agent": "Flutter Agent",
    "Youtube Agent": "Youtube Agent",
    "builder Agent": "builder Agent",
}

# Agent Modes
AGENT_MODE = {
    "Image Generation": {"mode": True, "id": "ImageGenerationLV45LJp", "name": "Image Generation"},
}

TRENDING_AGENT_MODE = {
    "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"},
    "Python Agent": {"mode": True, "id": "Python Agent"},
    "Java Agent": {"mode": True, "id": "Java Agent"},
    "JavaScript Agent": {"mode": True, "id": "JavaScript Agent"},
    "HTML Agent": {"mode": True, "id": "HTML Agent"},
    "Google Cloud Agent": {"mode": True, "id": "Google Cloud Agent"},
    "Android Developer": {"mode": True, "id": "Android Developer"},
    "Swift Developer": {"mode": True, "id": "Swift Developer"},
    "Next.js Agent": {"mode": True, "id": "Next.js Agent"},
    "MongoDB Agent": {"mode": True, "id": "MongoDB Agent"},
    "PyTorch Agent": {"mode": True, "id": "PyTorch Agent"},
    "React Agent": {"mode": True, "id": "React Agent"},
    "Xcode Agent": {"mode": True, "id": "Xcode Agent"},
    "AngularJS Agent": {"mode": True, "id": "AngularJS Agent"},
    "blackboxai-pro": {"mode": True, "id": "BLACKBOXAI-PRO"},
    "repomap": {"mode": True, "id": "repomap"},
    "Heroku Agent": {"mode": True, "id": "Heroku Agent"},
    "Godot Agent": {"mode": True, "id": "Godot Agent"},
    "Go Agent": {"mode": True, "id": "Go Agent"},
    "Gitlab Agent": {"mode": True, "id": "Gitlab Agent"},
    "Git Agent": {"mode": True, "id": "Git Agent"},
    "Flask Agent": {"mode": True, "id": "Flask Agent"},
    "Firebase Agent": {"mode": True, "id": "Firebase Agent"},
    "FastAPI Agent": {"mode": True, "id": "FastAPI Agent"},
    "Erlang Agent": {"mode": True, "id": "Erlang Agent"},
    "Electron Agent": {"mode": True, "id": "Electron Agent"},
    "Docker Agent": {"mode": True, "id": "Docker Agent"},
    "DigitalOcean Agent": {"mode": True, "id": "DigitalOcean Agent"},
    "Bitbucket Agent": {"mode": True, "id": "Bitbucket Agent"},
    "Azure Agent": {"mode": True, "id": "Azure Agent"},
    "Flutter Agent": {"mode": True, "id": "Flutter Agent"},
    "Youtube Agent": {"mode": True, "id": "Youtube Agent"},
    "builder Agent": {"mode": True, "id": "builder Agent"},
}

# Model Prefixes
MODEL_PREFIXES = {
    mode: f"@{value['id']}"
    for mode, value in TRENDING_AGENT_MODE.items()
    if mode not in ["gemini-1.5-flash", "llama-3.1-8b", "llama-3.1-70b", "llama-3.1-405b", "repomap"]
}