scripts/02_update_keys.py

# https://huggingface.co/docs/huggingface_hub/en/guides/manage-spaces
from pathlib import Path
from dotenv import dotenv_values
from huggingface_hub import HfApi


dotenv_path = Path(__file__).resolve().parent.parent / '.env'
print(dotenv_path)


env_vars = dotenv_values(dotenv_path)


SPACE_NAME: str = env_vars.pop('SPACE_NAME', '')
HF_TOKEN: str   = env_vars['HF_TOKEN']
if SPACE_NAME:
    api = HfApi(token = HF_TOKEN)
    for k,v in env_vars.items():
        api.add_space_secret(
            repo_id=SPACE_NAME,
            key=k,
            value=v,
        )
else:
    print('You should set "SPACE_NAME" in .env files')

print('Update Keys Successfully!')