Run Space preparation in background

app.py

@@ -25,7 +25,9 @@ COMFY_HOST = "127.0.0.1"
 COMFY_PORT = "8188"
 
 COMFY_PROCESS: subprocess.Popen | None = None
+PREPARE_PROCESS: subprocess.Popen | None = None
 BOOTSTRAPPED = False
+BOOTSTRAP_LOG = Path("/tmp/voicegate_bootstrap.log")
 
 
 def gpu_status_lines() -> list[str]:
@@ -90,6 +92,13 @@ def run_bootstrap(lines: list[str], *, allow_heavy: bool = True) -> None:
 def ensure_comfy(lines: list[str], *, timeout: float = 240) -> dict[str, Any]:
     global COMFY_PROCESS
 
+    if PREPARE_PROCESS is not None:
+        returncode = PREPARE_PROCESS.poll()
+        if returncode is None:
+            raise RuntimeError("Runtime preparation is still running. Check Prepare Status first.")
+        if returncode != 0:
+            raise RuntimeError(f"Runtime preparation failed with return code {returncode}.")
+
     run_bootstrap(lines, allow_heavy=False)
 
     try:
@@ -261,14 +270,44 @@ def voxcpm_tts_workflow(prefix: str) -> dict[str, Any]:
 
 
 def prepare_runtime() -> str:
+    global PREPARE_PROCESS
+
     lines = ["VoiceGate runtime preparation"]
-    started = time.time()
-    try:
-        run_bootstrap(lines, allow_heavy=True)
-        lines.append(f"elapsed_sec={time.time() - started:.1f}")
-        lines.append(f"comfy_dir_exists={(COMFY_DIR / 'main.py').exists()}")
-    except Exception as exc:
-        lines.append(f"error={type(exc).__name__}: {exc}")
+    if PREPARE_PROCESS is not None and PREPARE_PROCESS.poll() is None:
+        lines.append(f"prepare=already_running pid={PREPARE_PROCESS.pid}")
+        return "\n".join(lines)
+    BOOTSTRAP_LOG.parent.mkdir(parents=True, exist_ok=True)
+    log = BOOTSTRAP_LOG.open("ab")
+    command = [sys.executable, str(ROOT / "scripts" / "bootstrap_comfy.py")]
+    PREPARE_PROCESS = subprocess.Popen(
+        command,
+        cwd=ROOT,
+        stdout=log,
+        stderr=subprocess.STDOUT,
+    )
+    lines.append(f"prepare=started pid={PREPARE_PROCESS.pid}")
+    lines.append(f"log={BOOTSTRAP_LOG}")
+    return "\n".join(lines)
+
+
+def prepare_status() -> str:
+    global BOOTSTRAPPED
+
+    lines = ["VoiceGate runtime preparation status"]
+    if PREPARE_PROCESS is None:
+        lines.append("prepare=not_started")
+    else:
+        returncode = PREPARE_PROCESS.poll()
+        if returncode is None:
+            lines.append(f"prepare=running pid={PREPARE_PROCESS.pid}")
+        else:
+            lines.append(f"prepare=finished returncode={returncode}")
+            if returncode == 0 and (COMFY_DIR / "main.py").exists():
+                BOOTSTRAPPED = True
+    lines.append(f"comfy_dir_exists={(COMFY_DIR / 'main.py').exists()}")
+    if BOOTSTRAP_LOG.exists():
+        lines.append("bootstrap_log_tail:")
+        lines.extend(BOOTSTRAP_LOG.read_text(encoding="utf-8", errors="replace").splitlines()[-80:])
     return "\n".join(lines)
 
 
@@ -344,12 +383,14 @@ with gr.Blocks(title="VoiceGate") as demo:
     gr.Markdown("# VoiceGate")
     with gr.Row():
         prepare_run = gr.Button("Prepare")
+        prepare_status_run = gr.Button("Prepare Status")
         gpu_run = gr.Button("GPU")
         comfy_run = gr.Button("ComfyUI")
         melband_run = gr.Button("MelBand")
         voxcpm_run = gr.Button("VoxCPM TTS")
     output = gr.Textbox(label="Status", lines=28)
     prepare_run.click(fn=prepare_runtime, outputs=output)
+    prepare_status_run.click(fn=prepare_status, outputs=output)
     gpu_run.click(fn=gpu_smoke_test, outputs=output)
     comfy_run.click(fn=comfy_runtime_test, outputs=output)
     melband_run.click(fn=melband_gpu_test, outputs=output)