import os
from huggingface_hub import HfApi

ENDPOINTS = ["LogD",
             "KSol",
             "MLM CLint",
             "HLM CLint",
             "Caco-2 Permeability Efflux",
             "Caco-2 Permeability Papp A>B",
             "MPPB",
             "MBPB",
             "RLM CLint",
             "MGMB"]

LB_COLS0 = ["endpoint", 
            "user", 
            "MAE", 
            "R2", 
            "Spearman R",
            "Kendall's Tau", 
            "submission_time",
            "model_report"]
LB_COLS = ["user", "MAE", "R2", "Spearman R", "Kendall's Tau", "submission time", "model details"]
LB_DTYPES = ['markdown', 'number', 'number', 'number', 'number', 'str', 'markdown']

TOKEN = os.environ.get("HF_TOKEN")
CACHE_PATH=os.getenv("HF_HOME", ".")
API = HfApi(token=TOKEN)
organization="OpenADMET"
submissions_repo = f'{organization}/openadmet-challenge-submissions' # private
results_repo = f'{organization}/openadmet-challenge-results' # public
test_repo = f'{organization}/openadmet-challenge-test-data' # private