Node.js

const fs = require("node:fs");
const { WaveFile } = require("wavefile");
const { Model, ProcessorAsync, getVersion } = require("@ai-coustics/aic-sdk");

async function main() {
  const licenseKey = process.env.AIC_SDK_LICENSE;
  if (!licenseKey || licenseKey === "YOUR_SDK_KEY") {
    throw new Error("Set AIC_SDK_LICENSE to your SDK key, then retry.");
  }
  const input = new WaveFile(fs.readFileSync("input.wav"));
  if (input.fmt.numChannels !== 1 || input.fmt.sampleRate !== 16000 || input.bitDepth !== "16") {
    throw new Error("Export input.wav as mono, 16 kHz, 16-bit PCM WAV, then retry.");
  }
  if (input.data.samples.length !== input.data.chunkSize) {
    throw new Error("input.wav is truncated. Export the complete recording, then retry.");
  }
  input.toBitDepth("32f");
  const audio = input.getSamples(false, Float32Array);
  if (audio.length === 0) {
    throw new Error("input.wav is empty. Record speech, then retry.");
  }

  const modelPath = await Model.download("quail-vf-2.2-l-16khz", "./models");
  const model = Model.fromFile(modelPath);
  let processor;
  try {
    const blockSize = model.getOptimalBlockSize(16000);
    processor = new ProcessorAsync(model, licenseKey);
    await processor.initialize(16000, blockSize, false);
    const context = await processor.getContext();
    const delay = context.getAudioDelay();
    // Pad the last block and flush the delayed tail with silence.
    const paddedSize = Math.ceil((audio.length + delay) / blockSize) * blockSize;
    const padded = new Float32Array(paddedSize);
    padded.set(audio);
    const output = new Float32Array(paddedSize);
    for (let start = 0; start < paddedSize; start += blockSize) {
      // Await each block to preserve stream order. Async processing returns a new array.
      const enhanced = await processor.process(padded.subarray(start, start + blockSize));
      output.set(enhanced, start);
    }
    const enhanced = output.subarray(delay, delay + audio.length);
    if (!enhanced.every(Number.isFinite)) {
      throw new Error("Processing returned non-finite samples; no output was written.");
    }
    const pcm = Int16Array.from(enhanced, (sample) =>
      Math.trunc(Math.max(-1, Math.min(32767 / 32768, sample)) * 32768)
    );
    const result = new WaveFile();
    result.fromScratch(1, 16000, "16", pcm);
    fs.writeFileSync("enhanced.wav", result.toBuffer());
    console.log(`SDK ${getVersion()}, model ${model.getId()}`);
    console.log(`Processed ${audio.length} samples at 16000 Hz; wrote enhanced.wav`);
  } finally {
    // All processing promises above have settled before native resources are released.
    if (processor) processor.dispose();
    model.dispose();
  }
}

main().catch((error) => {
  console.error(`Error: ${error.message}`);
  process.exitCode = 1;
});