deploy: switch to chatterbox requirements @ 0ef36cd

tools_api/dramabox.py

@@ -22,6 +22,8 @@ import threading
 import time
 from pathlib import Path
 
+import spaces
+
 # Backend env knobs — kept compatible with the upstream Space.
 _LTX_DTYPE = os.environ.get("LTX_DTYPE", "bf16")
 
@@ -79,6 +81,33 @@ def _ensure_server():
         return _tts_server
 
 
+@spaces.GPU(duration=60)
+def _generate_scene_gpu(
+    *,
+    prompt: str,
+    out_dir: Path,
+    audio_ref: Path | None,
+    cfg: float,
+    stg: float,
+    dur_mult: float,
+    gen_dur: float,
+    ref_dur: float,
+    seed: int,
+) -> dict:
+    """Top-level ZeroGPU wrapper so HF detects Dramabox GPU usage at startup."""
+    return _generate_impl(
+        prompt=prompt,
+        out_dir=out_dir,
+        audio_ref=audio_ref,
+        cfg=cfg,
+        stg=stg,
+        dur_mult=dur_mult,
+        gen_dur=gen_dur,
+        ref_dur=ref_dur,
+        seed=seed,
+    )
+
+
 def generate_scene(
     *,
     prompt: str,
@@ -105,30 +134,17 @@ def generate_scene(
     if not prompt:
         raise ValueError("Prompt is empty.")
 
-    # Try to GPU-decorate at call time if `spaces` is available. On the
-    # ZeroGPU Space this maps weights onto the GPU for the duration of the
-    # call; on local dev (no `spaces`) it's a no-op pass-through.
-    try:
-        import spaces  # type: ignore[import-not-found]
-
-        @spaces.GPU(duration=60)
-        def _run():
-            return _generate_impl(
-                prompt=prompt,
-                out_dir=out_dir,
-                audio_ref=audio_ref,
-                cfg=cfg, stg=stg, dur_mult=dur_mult,
-                gen_dur=gen_dur, ref_dur=ref_dur, seed=seed,
-            )
-        return _run()
-    except ImportError:
-        return _generate_impl(
-            prompt=prompt,
-            out_dir=out_dir,
-            audio_ref=audio_ref,
-            cfg=cfg, stg=stg, dur_mult=dur_mult,
-            gen_dur=gen_dur, ref_dur=ref_dur, seed=seed,
-        )
+    return _generate_scene_gpu(
+        prompt=prompt,
+        out_dir=out_dir,
+        audio_ref=audio_ref,
+        cfg=cfg,
+        stg=stg,
+        dur_mult=dur_mult,
+        gen_dur=gen_dur,
+        ref_dur=ref_dur,
+        seed=seed,
+    )
 
 
 def _generate_impl(