console.log("Initializing Kokoro TTS Worker");

import { KokoroTTS } from "https://cdn.jsdelivr.net/npm/kokoro-js@1.2.0/+esm";
async function detectWebGPU() {
  try {
    const adapter = await navigator.gpu.requestAdapter();
    return !!adapter;
  } catch (e) {
    return false;
  }
}
// Device detection
const device = (await detectWebGPU()) ? "webgpu" : "wasm";
self.postMessage({ status: "device", device });

console.log(`Detected device: ${device}`);

// Load the model
const model_id = "onnx-community/Kokoro-82M-v1.0-ONNX";
const tts = await KokoroTTS.from_pretrained(model_id, {
  dtype: device === "wasm" ? "q8" : "fp32",
  device,
});

console.log("Kokoro TTS model loaded successfully");

self.postMessage({ status: "ready", voices: tts.voices, device });

console.log("Available voices:", tts.voices);

// Listen for messages from the main thread
self.addEventListener("message", async (e) => {
  const { text, voice } = e.data;

  try {
    // Generate speech
    const audio = await tts.generate(text, { voice });

    // Send the audio file back to the main thread
    const blob = audio.toBlob();
    self.postMessage({
      status: "complete",
      audio: URL.createObjectURL(blob),
      text,
    });
  } catch (error) {
    self.postMessage({ status: "error", error: error.message });
  }
});