Spaces:
Running
Running
File size: 3,096 Bytes
a84b2a7 b0501bf a84b2a7 b0501bf a84b2a7 aa88294 a84b2a7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | #!/usr/bin/env python3
"""Deterministic deployment script to synchronize a green commit to Hugging Face Space."""
from __future__ import annotations
import argparse
import os
import subprocess
import sys
import tempfile
from pathlib import Path
def main():
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--space", default="Leon4gr45/openoperator", help="Target HF Space")
parser.add_argument("--sha", default="", help="Expected GitHub commit SHA")
args = parser.parse_args()
hf_token = os.environ.get("HF_TOKEN", "").strip()
if not hf_token:
print("Error: HF_TOKEN environment variable is required to push to Hugging Face Space.", file=sys.stderr)
return 1
root = Path(__file__).resolve().parents[1]
sha = args.sha.strip()
if not sha:
try:
sha = subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=root, text=True).strip()
except Exception:
sha = "HEAD"
print(f"Deploying commit SHA {sha} to Hugging Face Space {args.space}...")
with tempfile.TemporaryDirectory() as temp_dir:
snapshot = Path(temp_dir) / "hf_snapshot"
snapshot.mkdir(parents=True, exist_ok=True)
tar_process = subprocess.Popen(["git", "archive", "--format=tar", sha], cwd=root, stdout=subprocess.PIPE)
subprocess.check_call(["tar", "-xf", "-", "-C", str(snapshot)], stdin=tar_process.stdout)
tar_process.wait()
readme_path = snapshot / "README.md"
content = readme_path.read_text(encoding="utf-8") if readme_path.exists() else ""
metadata = """---
title: OpenOperator
emoji: 🤖
colorFrom: indigo
colorTo: blue
sdk: docker
app_port: 7860
pinned: false
---
"""
if not content.startswith("---"):
readme_path.write_text(metadata + content, encoding="utf-8")
build_json = snapshot / "openoperator-build.json"
build_json.write_text(f'{{\n "source_sha": "{sha}"\n}}\n', encoding="utf-8")
subprocess.check_call(["git", "init", "--initial-branch=main"], cwd=snapshot)
subprocess.check_call(["git", "config", "user.name", "OpenOperator Deployer"], cwd=snapshot)
subprocess.check_call(["git", "config", "user.email", "deploy@openoperator.local"], cwd=snapshot)
subprocess.check_call(["git", "add", "--all"], cwd=snapshot)
subprocess.check_call(["git", "commit", "-m", f"Deploy source revision {sha}"], cwd=snapshot)
remote_url = f"https://huggingface.co/spaces/{args.space}"
subprocess.check_call(["git", "remote", "add", "huggingface", remote_url], cwd=snapshot)
subprocess.check_call(["git", "config", "http.extraHeader", f"Authorization: Bearer {hf_token}"], cwd=snapshot)
print(f"Pushing snapshot to Hugging Face Space {args.space}...")
subprocess.check_call(
["git", "push", "--force", "huggingface", "main:main"],
cwd=snapshot,
)
print(f"Deployment push complete for {args.space} at SHA {sha}.")
return 0
if __name__ == "__main__":
raise SystemExit(main())
|