| # 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', '') | |
| if SPACE_NAME: | |
| api = HfApi() | |
| 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!') | |