Add persistent Athena music mode switching

platform/docker/profile-controller/profile_controller.py

@@ -28,6 +28,8 @@ IMAGE_WORKER = os.environ.get("IMAGE_WORKER", "image")
 RESTORE_WORKER = os.environ.get("RESTORE_WORKER", "restore")
 TTS_LABEL_KEY = "com.mike-ai.tts-worker"
 TTS_WORKER = os.environ.get("TTS_WORKER", "qwen3")
+MUSIC_LABEL_KEY = "com.mike-ai.music-worker"
+MUSIC_WORKER = os.environ.get("MUSIC_WORKER", "").strip()
 LOCK = threading.Lock()
 log = logging.getLogger("profile-controller")
 
@@ -96,6 +98,22 @@ def tts_container() -> dict:
     return matches[0]
 
 
+def music_container() -> dict:
+    if not MUSIC_WORKER:
+        raise RuntimeError("music worker is not configured")
+    matches = [item for item in labelled_containers(MUSIC_LABEL_KEY)
+               if item.get("Labels", {}).get(MUSIC_LABEL_KEY) == MUSIC_WORKER]
+    if len(matches) != 1:
+        raise RuntimeError(
+            f"expected exactly one music worker {MUSIC_WORKER!r}, found {len(matches)}")
+    return matches[0]
+
+
+def stop_music_if_configured() -> None:
+    if MUSIC_WORKER:
+        stop_container(music_container(), timeout=30)
+
+
 def stop_container(item: dict, timeout: int = 120) -> None:
     if item.get("State") != "running":
         return
@@ -133,6 +151,7 @@ def set_image_worker(running: bool, kind: str = IMAGE_WORKER) -> dict:
             # The 9B beta text encoder temporarily borrows the RTX 3060 from
             # Qwen3-TTS. The gateway retains Piper as a fallback meanwhile.
             stop_container(tts_container(), timeout=30)
+            stop_music_if_configured()
             for other in image_containers():
                 if other["Id"] != item["Id"]:
                     stop_container(other, timeout=20)
@@ -148,6 +167,23 @@ def set_image_worker(running: bool, kind: str = IMAGE_WORKER) -> dict:
                 "state": "running" if running else "stopped"}
 
 
+def set_music_worker(running: bool) -> dict:
+    """Start ACE-Step exclusively, or stop it before LLM restoration."""
+    with LOCK:
+        item = music_container()
+        if running:
+            for profile_item in containers().values():
+                stop_container(profile_item)
+            for worker in image_containers():
+                stop_container(worker, timeout=20)
+            stop_container(tts_container(), timeout=30)
+            start_container(item)
+        else:
+            stop_container(item, timeout=30)
+        return {"music_worker": MUSIC_WORKER,
+                "state": "running" if running else "stopped"}
+
+
 def active_profile(items: dict[str, dict] | None = None) -> str | None:
     items = items or containers()
     active = [name for name, item in items.items() if item.get("State") == "running"]
@@ -163,6 +199,7 @@ def activate(profile: str) -> dict:
         # Defensive mutual exclusion even if a caller bypasses the router.
         for worker in image_containers():
             stop_container(worker)
+        stop_music_if_configured()
         start_container(tts_container())
         items = containers()
         missing = [name for name in ALLOWED if name not in items]
@@ -227,7 +264,16 @@ class Handler(BaseHTTPRequestHandler):
             return
         try:
             items = containers()
+            music = music_container() if MUSIC_WORKER else {}
+            music_status = music.get("Status", "")
+            music_health = ("disabled" if not MUSIC_WORKER else
+                            "healthy" if "(healthy)" in music_status else
+                            "unhealthy" if "(unhealthy)" in music_status else
+                            "starting" if music.get("State") == "running" else
+                            "stopped")
             self.reply(200, {"active_profile": active_profile(items),
+                             "music_worker": music.get("State", "disabled"),
+                             "music_health": music_health,
                              "profiles": {name: items.get(name, {}).get(
                                  "State", "missing") for name in ALLOWED}})
         except Exception as exc:
@@ -245,6 +291,13 @@ class Handler(BaseHTTPRequestHandler):
                 log.exception("stopping inference failed")
                 self.reply(503, {"error": str(exc)})
             return
+        if self.path in {"/workers/music/start", "/workers/music/stop"}:
+            try:
+                self.reply(200, set_music_worker(self.path.endswith("/start")))
+            except Exception as exc:
+                log.exception("music worker transition failed")
+                self.reply(503, {"error": str(exc)})
+            return
         worker_paths = {
             "/workers/image/start": (IMAGE_WORKER, True),
             "/workers/image/stop": (IMAGE_WORKER, False),