# prepare_env.py

import os

TMP_PATH = "/tmp"

os.environ["HF_HOME"] = os.path.join(TMP_PATH, "huggingface")
os.environ["TRANSFORMERS_CACHE"] = os.environ["HF_HOME"] + "/transformers"
os.environ["HF_DATASETS_CACHE"] = os.environ["HF_HOME"] + "/datasets"
os.environ["HF_METRICS_CACHE"] = os.environ["HF_HOME"] + "/metrics"
os.environ["MPLCONFIGDIR"] = os.path.join(TMP_PATH, "matplotlib")
os.environ["TORCH_HOME"] = os.path.join(TMP_PATH, "torch")
os.environ["XDG_CACHE_HOME"] = os.path.join(TMP_PATH, "xdg-cache")
os.environ["HOME"] = TMP_PATH

# Création des dossiers si nécessaires
for path in [
    os.environ["HF_HOME"],
    os.environ["MPLCONFIGDIR"],
    os.environ["TORCH_HOME"],
    os.environ["XDG_CACHE_HOME"],
    TMP_PATH
]:
    os.makedirs(path, exist_ok=True)

print("✅ Environnement temporaire prêt.")