from pathlib import Path


ROOT_DIR = Path(__file__).resolve().parents[1]
CONF_DIR = ROOT_DIR / "conf"
DATASETS_DIR = CONF_DIR / "datasets"
FIGURES_DIR = CONF_DIR / "Figures"
WEIGHT_DIR = ROOT_DIR / "weight"
UNIKP_MODEL_DIR = WEIGHT_DIR / "UniKP_model"
PROT_T5_DIR = WEIGHT_DIR / "prot_t5_xl_uniref50"
PROT_T5_MODEL = str(PROT_T5_DIR if PROT_T5_DIR.exists() else "prot_t5_xl_uniref50")

VOCAB_PATH = WEIGHT_DIR / "vocab.pkl"
TRFM_PATH = WEIGHT_DIR / "trfm_12_23000.pkl"
KCAT_DATASET_PATH = DATASETS_DIR / "Kcat_combination_0918_wildtype_mutant.json"
KM_TEST_PATH = DATASETS_DIR / "Km_test_11722.pkl"
KCAT_KM_SAMPLES_PATH = DATASETS_DIR / "kcat_km_samples.xlsx"
PH_SMILES_PATH = DATASETS_DIR / "Generated_pH_unified_smiles_636.xlsx"
DEGREE_SMILES_PATH = DATASETS_DIR / "Generated_degree_unified_smiles_572.xlsx"


def output_path(*parts):
    return ROOT_DIR.joinpath(*parts)