VoiceAI Service Portal

class VoiceAIStreamingClient {
  constructor({ apiKey, host = window.location.host, language = "vi", onPartial, onFinal, onError }) {
    this.wsUrl = `ws://${host}/v1/audio/transcriptions/stream?api_key=${encodeURIComponent(apiKey)}&language=${language}`;
    this.onPartial = onPartial || (() => {});
    this.onFinal = onFinal || (() => {});
    this.onError = onError || console.error;
  }

  async startMicrophone() {
    this.ws = new WebSocket(this.wsUrl);
    this.ws.binaryType = "arraybuffer";

    this.ws.onopen = () => {
      this.ws.send(JSON.stringify({
        action: "start",
        sample_rate: 16000,
        channels: 1,
        format: "pcm16",
        language: "vi",
        interim_results: true
      }));
    };

    this.ws.onmessage = (event) => {
      const msg = JSON.parse(event.data);
      if (msg.type === "transcript") {
        if (msg.is_final) this.onFinal(msg);
        else this.onPartial(msg);
      } else if (msg.type === "error") {
        this.onError(msg);
      }
    };

    this.stream = await navigator.mediaDevices.getUserMedia({ audio: { channelCount: 1, sampleRate: 16000 } });
    this.audioCtx = new (window.AudioContext || window.webkitAudioContext)({ sampleRate: 16000 });
    const source = this.audioCtx.createMediaStreamSource(this.stream);
    this.processor = this.audioCtx.createScriptProcessor(4096, 1, 1);

    this.processor.onaudioprocess = (e) => {
      if (!this.ws || this.ws.readyState !== WebSocket.OPEN) return;
      const input = e.inputBuffer.getChannelData(0);
      const pcm16 = new Int16Array(input.length);
      for (let i = 0; i < input.length; i++) {
        const s = Math.max(-1, Math.min(1, input[i]));
        pcm16[i] = s < 0 ? s * 0x8000 : s * 0x7FFF;
      }
      this.ws.send(pcm16.buffer);
    };

    source.connect(this.processor);
    this.processor.connect(this.audioCtx.destination);
  }

  stop() {
    if (this.processor) this.processor.disconnect();
    if (this.audioCtx) this.audioCtx.close();
    if (this.stream) this.stream.getTracks().forEach((t) => t.stop());
    if (this.ws && this.ws.readyState === WebSocket.OPEN) {
      this.ws.send(JSON.stringify({ action: "stop" }));
    }
  }
}