jam/stop fix

app.py

@@ -339,8 +339,13 @@ def jam_stop(session_id: str = Body(..., embed=True)):
         worker = jam_registry.get(session_id)
     if worker is None:
         raise HTTPException(status_code=404, detail="Session not found")
+
     worker.stop()
-    worker.join(timeout=2.0)
+    worker.join(timeout=5.0)
+    if worker.is_alive():
+        # It’s daemon=True, so it won’t block process exit, but report it
+        print(f"⚠️ JamWorker {session_id} did not stop within timeout")
+
     with jam_lock:
         jam_registry.pop(session_id, None)
     return {"stopped": True}

jam_worker.py

@@ -41,13 +41,13 @@ class JamWorker(threading.Thread):
         self.state = mrt.init_state()
         self.idx = 0
         self.outbox: list[JamChunk] = []
-        self._stop = threading.Event()
+        self._stop_event = threading.Event()
         self.last_chunk_started_at = None
         self.last_chunk_completed_at = None
         self._lock = threading.Lock()
 
     def stop(self):
-        self._stop.set()
+        self._stop_event.set()
 
     def update_style(self, new_style_vec: np.ndarray | None):
         with self._lock:
@@ -90,7 +90,7 @@ class JamWorker(threading.Thread):
 
         # Prime: set initial context on state (caller should have done this; safe to re-set here)
         # NOTE: We assume caller passed a style_vec computed from tail/whole/blend.
-        while not self._stop.is_set():
+        while not self._stop_event.is_set():
             # honor live knob updates atomically
             with self._lock:
                 style_vec = self.params.style_vec
@@ -103,13 +103,13 @@ class JamWorker(threading.Thread):
             need = chunk_secs
             chunks = []
             self.last_chunk_started_at = time.time()
-            while need > 0 and not self._stop.is_set():
+            while need > 0 and not self._stop_event.is_set():
                 wav, self.state = self.mrt.generate_chunk(state=self.state, style=style_vec)
                 chunks.append(wav)
                 # model chunk length (seconds) at model SR
                 need -= (wav.samples.shape[0] / float(self.mrt.sample_rate))
 
-            if self._stop.is_set():
+            if self._stop_event.is_set():
                 break
 
             # 2) stitch and trim to exact seconds at model SR