import os

from huggingface_hub import HfApi

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

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


if PRIVATE_REPO:
    REPO_ID = f"{OWNER}/MEDIC-Benchmark-Private"
    QUEUE_REPO = f"{OWNER}/medic-requests-private"
    RESULTS_REPO = f"{OWNER}/medic-results-private"
    
else:
    REPO_ID = f"{OWNER}/MEDIC-Benchmark"
    QUEUE_REPO = f"{OWNER}/medic-harness-requests"
    RESULTS_REPO = f"{OWNER}/medic-harness-results"

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

# Local caches

if not PRIVATE_REPO:
    EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "medic-harness-requests")
    EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "medic-harness-results")
    EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "medic-harness-requests-bk")
    EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "medic-harness-results-bk")
else:
    EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "medic-requests-private")
    EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "medic-results-private")
    EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "medic-requests-bk")
    EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "medic-results-bk")

API = HfApi(token=TOKEN)