"""Publish the prepared Space using the user's Modal-held HF write token.

The token is returned to this local process in memory only; it is never printed
or saved. Existing repositories are never overwritten by this script.
"""
import os
from pathlib import Path
import modal

app = modal.App('beamboot-publish')

@app.function(image=modal.Image.debian_slim(python_version='3.12'), secrets=[modal.Secret.from_name('steerability-submission-write')])
def get_write_token():
    token = os.environ.get('HUGGINGFACE_TOKEN') or os.environ.get('HF_TOKEN')
    if not token:
        raise RuntimeError('No HF write token in configured secret')
    return token

@app.local_entrypoint()
def main():
    from huggingface_hub import HfApi
    token = get_write_token.remote()
    api = HfApi(token=token)
    name = api.whoami()['name']
    repo_id = f'{name}/beam-boot'
    if not api.repo_exists(repo_id=repo_id, repo_type='space'):
        api.create_repo(repo_id=repo_id, repo_type='space', space_sdk='static', private=False)
    print(f'Uploading to Space created for this task: https://huggingface.co/spaces/{repo_id}', flush=True)
    api.upload_folder(repo_id=repo_id, repo_type='space', folder_path=str(Path('static-space').resolve()),
                      commit_message='Add Super Beam: Beam plays a game that runs inside Beam',
                      ignore_patterns=['**/__pycache__/**', '**/*.pyc'])
    print(f'Uploaded files: https://huggingface.co/spaces/{repo_id}', flush=True)