// Copyright (c)  2025  Xiaomi Corporation (authors: Fangjun Kuang)

const sherpa_onnx = require('sherpa-onnx');

const silenceScale = 0.2;

function createOfflineTts() {
  let offlineTtsMatchaModelConfig = {
    acousticModel: './matcha-icefall-en_US-ljspeech/model-steps-3.onnx',
    vocoder: './vocos-22khz-univ.onnx',
    lexicon: './lexicon-en-us.txt',
    tokens: './matcha-icefall-en_US-ljspeech/tokens.txt',

    noiseScale: 0.667,
    lengthScale: 1.0,
  };
  let offlineTtsModelConfig = {
    offlineTtsMatchaModelConfig: offlineTtsMatchaModelConfig,
    numThreads: 1,
    debug: 1,
    provider: 'cpu',
  };

  let offlineTtsConfig = {
    offlineTtsModelConfig: offlineTtsModelConfig,
    maxNumSentences: 1,
    silenceScale: silenceScale,
  };

  return sherpa_onnx.createOfflineTts(offlineTtsConfig);
}

const tts = createOfflineTts();
const text =
    'Today as always, men fall into two groups: slaves and free men. Whoever does not have two-thirds of his day for himself, is a slave, whatever he may be: a statesman, a businessman, an official, or a scholar.';

const generationConfig = {
  sid: 0,
  speed: 1.0,
  silenceScale: silenceScale,
};

const audio = tts.generateWithConfig(text, generationConfig);
tts.save('./test-matcha-en-lexicon.wav', audio);
console.log('Saved to test-matcha-en-lexicon.wav successfully.');
tts.free();