import os
import sys
from pathlib import Path
from huggingface_hub import HfApi, upload_folder


def main():
    token = os.environ.get("HF_TOKEN")
    repo_id = os.environ.get("HF_SPACE_ID")
    if not token or not repo_id:
        print("Missing HF_TOKEN or HF_SPACE_ID", file=sys.stderr)
        sys.exit(1)
    api = HfApi(token=token)
    src_dir = Path(__file__).parent
    print(f"Uploading folder {src_dir} to {repo_id}...")
    upload_folder(
        repo_id=repo_id,
        repo_type="space",
        folder_path=str(src_dir),
        token=token,
        commit_message="Bulk upload all files",
        ignore_patterns=["__pycache__", "*.pyc", ".venv", "deploy_hf_space.py"],
    )
    print("Uploaded. Restarting Space...")
    try:
        api.restart_space(repo_id=repo_id)
    except Exception as e:
        print(f"Warning: restart failed: {e}")


if __name__ == "__main__":
    main()