Update worker.js

Author: Rbell1216

worker.js

@@ -22,7 +22,10 @@ if (self.location.hostname === "localhost2") {
 }
 
 const tts = await KokoroTTS.from_pretrained(model_id, {
-  dtype: device === "wasm" ? "q8" : "fp32", device,
+  // --- THIS IS THE MODIFIED LINE ---
+  dtype: "fp16", // Use fp16 for better performance on mobile
+  // --- END MODIFIED LINE ---
+  device,
   progress_callback: (progress) => {
     self.postMessage({ status: "loading_model_progress", progress });
   }
@@ -33,8 +36,70 @@ const tts = await KokoroTTS.from_pretrained(model_id, {
 
 self.postMessage({ status: "loading_model_ready", voices: tts.voices, device });
 
-// Track how many buffers are currently in the queue
+// --- THIS IS THE MEMORY-SAFE QUEUE LOGIC ---
 let bufferQueueSize = 0;
+const MAX_QUEUE_SIZE = 6; // Work ahead by 6 chunks (This is already set correctly)
+let shouldStop = false;
+// --- END QUEUE LOGIC ---
+
+self.addEventListener("message", async (e) => {
+  const { type, text, voice, speed } = e.data; // <-- Get speed from the message
+  
+  if (type === "stop") {
+    bufferQueueSize = 0;
+    shouldStop = true;
+    console.log("Stop command received, stopping generation");
+    return;
+  }
+
+  // --- THIS IS THE MEMORY-SAFE QUEUE LOGIC ---
+  if (type === "buffer_processed") {
+    bufferQueueSize = Math.max(0, bufferQueueSize - 1); // Free up a slot
+    return;
+  }
+  // --- END QUEUE LOGIC ---
+
+  if (type === "generate" && text) { 
+    shouldStop = false;
+    let chunks = splitTextSmart(text, 300); 
+    
+    self.postMessage({ status: "chunk_count", count: chunks.length });
+
+    for (const chunk of chunks) {
+      if (shouldStop) {
+        console.log("Stopping audio generation");
+        self.postMessage({ status: "complete" });
+        break;
+      }
+      console.log(chunk);
+
+      // --- THIS IS THE MEMORY-SAFE QUEUE LOGIC ---
+      // Wait if the queue is full
+      while (bufferQueueSize >= MAX_QUEUE_SIZE && !shouldStop) {
+        console.log("Waiting for buffer space...");
+        await new Promise(resolve => setTimeout(resolve, 1000)); // Wait 1 sec
+        if (shouldStop) break;
+      }
+
+      if (shouldStop) {
+        console.log("Stopping after queue wait");
+        self.postMessage({ status: "complete" });
+        break;
+      }
+      // --- END QUEUE LOGIC ---
+      
+      const audio = await tts.generate(chunk, { voice, speed }); 
+      let ab = audio.audio.buffer;
+
+      bufferQueueSize++; // Increment the queue size
+      self.postMessage({ status: "stream_audio_data", audio: ab, text: chunk }, [ab]);
+    }
+
+    if (!shouldStop) {
+      self.postMessage({ status: "complete" });
+    }
+  }
+});let bufferQueueSize = 0;
 const MAX_QUEUE_SIZE = 6;
 let shouldStop = false;