import os

from huggingface_hub import HfApi, login

from .get_environments import get_environments
from .get_files import get_mp4_paths


class HuggingFaceClient:
    def __init__(self) -> None:
        login(
            token=os.environ.get("HUGGINGFACE_TOKEN"),
            add_to_git_credential=True,
            new_session=False,
        )
        self.hf_api = HfApi()
        self.environments = get_environments(self.hf_api)
        self.mp4_paths = get_mp4_paths(environments=self.environments)