import os

from huggingface_hub import Repository, login, logout

from core.Agent import QAAgent

if not os.path.isdir('core'):
    login(token=os.environ['HF_TOKEN'])

    repo = Repository(
        local_dir="core",
        repo_type="dataset",
        clone_from=os.environ['REPO_URL'],
        token=True
    )
    repo.git_pull()
    logout()

QAAgent().chat_ui()