import os
from dotenv import load_dotenv

load_dotenv()

# Remove Blackbox-specific configurations
# BASE_URL = "https://www.blackbox.ai"
# headers = { ... }

# Add DDG-specific configurations if any
DDG_API_ENDPOINT = "https://duckduckgo.com/duckchat/v1/chat"
DDG_STATUS_URL = "https://duckduckgo.com/duckchat/v1/status"

APP_SECRET = os.getenv("APP_SECRET")

# Update ALLOWED_MODELS to include DDG models
ALLOWED_MODELS = [
    {"id": "gpt-4o-mini", "name": "GPT-4o Mini"},
    {"id": "claude-3-haiku-20240307", "name": "Claude 3 Haiku"},
    {"id": "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo", "name": "Meta LLaMA 3.1-70B Instruct Turbo"},
    {"id": "mistralai/Mixtral-8x7B-Instruct-v0.1", "name": "Mistral Mixtral 8x7B Instruct v0.1"},
    # Add more DDG models as needed
]

MODEL_MAPPING = {
    "gpt-4o-mini": "gpt-4o-mini",
    "claude-3-haiku": "claude-3-haiku-20240307",
    "llama-3.1-70b": "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
    "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1",
    # Add more mappings as needed
}

# Remove or update AGENT_MODE, TRENDING_AGENT_MODE, MODEL_PREFIXES, MODEL_REFERERS as necessary
# For example:
AGENT_MODE = {
    'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
    'Niansuh': {'mode': True, 'id': "NiansuhAIk1HgESy", 'name': "Niansuh"},
    # Add DDG-specific agent modes if required
}

TRENDING_AGENT_MODE = {
    "gpt-4o-mini": {},
    "claude-3-haiku-20240307": {},
    "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo": {},
    "mistralai/Mixtral-8x7B-Instruct-v0.1": {},
    # Add more DDG-specific trending agent modes if required
}

MODEL_PREFIXES = {
    'gpt-4o-mini': '@GPT-4o Mini',
    'claude-3-haiku-20240307': '@Claude-3-Haiku',
    'meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo': '@Meta-LLaMA-3.1-70B',
    'mistralai/Mixtral-8x7B-Instruct-v0.1': '@Mistral-Mixtral-8x7B',
    # Add more DDG-specific prefixes as needed
}

MODEL_REFERERS = {
    "gpt-4o-mini": "/?model=gpt-4o-mini",
    "claude-3-haiku-20240307": "/?model=claude-3-haiku-20240307",
    "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo": "/?model=meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
    "mistralai/Mixtral-8x7B-Instruct-v0.1": "/?model=mistralai/Mixtral-8x7B-Instruct-v0.1",
    # Add more DDG-specific referers as needed
}