import os

from huggingface_hub import HfApi

# Run in local mode (no Hugging Face connection required)
# Set to True when developing locally without HF credentials
LOCAL_MODE = True

# Info to change for your repository
# ----------------------------------
# Get token from environment or use None in local mode
TOKEN = os.environ.get("HF_TOKEN") if not LOCAL_MODE else None

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

REPO_ID = f"{OWNER}/leaderboard"
QUEUE_REPO = f"{OWNER}/vulnerability-requests"
RESULTS_REPO = f"{OWNER}/vulnerability-assessments"

# 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, "assessment-queue")
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "assessment-results")
EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "assessment-queue-bk")
EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "assessment-results-bk")

# Initialize API with token if available
API = HfApi(token=TOKEN)