auto-cache sd-cli binary in vendor/ after first compile

.gitattributes

@@ -0,0 +1 @@
+vendor/sd-cli filter=lfs diff=lfs merge=lfs -text

app.py

@@ -38,6 +38,8 @@ os.makedirs(BIN_DIR, exist_ok=True)
 SDCPP_REPO = "https://github.com/leejet/stable-diffusion.cpp"
 SDCPP_COMMIT = "67dda3f"
 _SDCLI_NAME = "sd-cli"
+_VENDOR_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "vendor")
+SPACE_REPO_ID = "Luminia/Anima-2B-CPU"
 
 
 def _ensure_file(repo_id, filename, dest_dir):
@@ -88,8 +90,37 @@ def _build_sdcli():
     print(f"[init] sd-cli compiled with OpenBLAS in {elapsed:.0f}s")
 
 
+def _upload_sdcli_to_repo(binary_path):
+    """Upload compiled sd-cli to the Space repo so next boot skips compilation."""
+    token = os.environ.get("HF_TOKEN")
+    if not token:
+        print("[init] No HF_TOKEN, skipping binary upload to repo")
+        return
+    try:
+        from huggingface_hub import HfApi
+        api = HfApi(token=token)
+        api.upload_file(
+            repo_id=SPACE_REPO_ID,
+            repo_type="space",
+            path_or_fileobj=binary_path,
+            path_in_repo="vendor/sd-cli",
+            commit_message="cache compiled sd-cli with OpenBLAS",
+        )
+        print("[init] Uploaded sd-cli to repo (next boot will skip compile)")
+    except Exception as e:
+        print(f"[init] Failed to upload sd-cli to repo: {e}")
+
+
 def _ensure_sdcli():
-    """Find sd-cli in cache/PATH, or compile from source with OpenBLAS."""
+    """Find sd-cli in vendor/, cache, PATH, or compile from source."""
+    # 1. Check vendor/ in the Space repo (pre-compiled, fastest)
+    vendor_bin = os.path.join(_VENDOR_DIR, _SDCLI_NAME)
+    if os.path.isfile(vendor_bin):
+        os.chmod(vendor_bin, os.stat(vendor_bin).st_mode | stat.S_IEXEC)
+        print(f"[init] Using cached sd-cli from vendor/")
+        return vendor_bin
+
+    # 2. Check /tmp cache or PATH
     for candidate in [
         os.path.join(BIN_DIR, _SDCLI_NAME),
         os.path.join(BIN_DIR, _SDCLI_NAME + ".exe"),
@@ -102,10 +133,14 @@ def _ensure_sdcli():
         print("[init] sd-cli not in PATH. On Windows, build from source or add to PATH.")
         sys.exit(1)
 
+    # 3. Compile from source
     _build_sdcli()
     cli_path = os.path.join(BIN_DIR, _SDCLI_NAME)
     if not os.path.isfile(cli_path):
         raise RuntimeError("sd-cli build failed")
+
+    # 4. Upload to repo for next boot
+    _upload_sdcli_to_repo(cli_path)
     return cli_path