import os

from huggingface_hub import HfApi

# Info to change for your repository
# ----------------------------------
TOKEN = os.environ.get("HF_TOKEN")  # A read/write token for your org
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
ANTHROPIC_API_KEY = os.environ.get("ANTHROPIC_API_KEY")
COHERE_API_KEY = os.environ.get("COHERE_API_KEY")

OWNER = (
    "umdclip"  # Change to your org - don't forget to create a results and request dataset, with the correct format!
)
# ----------------------------------

REPO_ID = f"{OWNER}/advcal-leaderboard"
QUEUE_REPO = f"{OWNER}/advcal-requests"
RESULTS_REPO = f"{OWNER}/advcal-results"

PLAYGROUND_DATASET_NAMES = {
    "tossup": "umdclip/acf-co24-tossups",
    "bonus": "umdclip/acf-co24-bonuses",
}

# If you setup a cache later, just change HF_HOME
CACHE_PATH = os.getenv("HF_HOME", ".")

# Local caches
EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "eval-queue")
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")
EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk")
EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")

THEME = "gstaff/xkcd"
UNSELECTED_VAR_NAME = "Select Variable..."
UNSELECTED_MODEL_NAME = "Select Model..."
AVAILABLE_MODELS = {
    "OpenAI/gpt-4o": {
        "model": "gpt-4o-2024-11-20",
    },
    "OpenAI/gpt-4o-mini": {
        "model": "gpt-4o-mini-2024-07-18",
    },
    "OpenAI/gpt-3.5-turbo": {
        "model": "gpt-3.5-turbo-0125",
    },
    "Anthropic/claude-3-7-sonnet": {
        "model": "claude-3-7-sonnet-20250219",
    },
    "Anthropic/claude-3-5-sonnet": {
        "model": "claude-3-5-sonnet-20241022",
    },
    "Anthropic/claude-3-5-haiku": {
        "model": "claude-3-5-haiku-20241022",
    },
    "Cohere/command-r": {
        "model": "command-r-08-2024",
    },
    "Cohere/command-r-plus": {
        "model": "command-r-plus-08-2024",
    },
    "Cohere/command-r7b": {
        "model": "command-r7b-12-2024",
    },
}

DEFAULT_SELECTIONS = {
    "tossup": {
        "simple_workflow": False,
        "model": "OpenAI/gpt-4o-mini",
        "temperature": 0.2,
        "buzz_threshold": 0.85,
        "early_stop": True,
    },
    "bonus": {
        "simple_workflow": False,
        "model": "OpenAI/gpt-4o-mini",
        "temperature": 0.2,
        "buzz_threshold": 0.85,
        "early_stop": True,
    },
}

DAILY_SUBMISSION_LIMIT_PER_USER = 5
API = HfApi(token=TOKEN)