src/envs.py

import os
from huggingface_hub import HfApi
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# Hugging Face configuration
TOKEN = os.environ.get("HF_TOKEN")  # A read/write token for your org
OWNER = os.environ.get("OWNER", "guard-bench")  # Change to your org
SUBMITTER_TOKEN = os.environ.get("SUBMITTER_TOKEN")
ADMIN_USERNAME = os.environ.get("ADMIN_USERNAME")
ADMIN_PASSWORD = os.environ.get("ADMIN_PASSWORD")

# Repository IDs
REPO_ID = f"{OWNER}/leaderboard"
RESULTS_DATASET_ID = os.environ.get("RESULTS_DATASET_ID", f"{OWNER}/guardbench-results")

# Cache paths
CACHE_PATH = os.getenv("HF_HOME", ".")
DATA_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "data")

# Local data paths
LEADERBOARD_FILE = os.path.join(DATA_PATH, "leaderboard.json")

# HF API instance
API = HfApi(token=TOKEN)