import {
  // 註冊表函數
  loadRegistry,
  resolveVad,
  resolveWakeword,
  resolveWhisper,
  
  // VAD 函數
  loadVadSession,
  createVadState,
  processVad,
  
  // 喚醒詞函數
  loadWakewordResources,
  createWakewordState,
  processWakewordChunk,
  
  // Whisper 函數
  loadWhisperResources,
  transcribe,
  
  // 配置管理
  ConfigManager,
  
  // 型別
  DEFAULT_VAD_PARAMS,
  DEFAULT_WAKEWORD_PARAMS,
} from 'web-asr-core';

// 載入模型註冊表
const registry = await loadRegistry('./models/global_registry.json');

// 初始化服務...

// 1. 載入 VAD 模型
const vadInfo = resolveVad(registry);
const vadSession = await loadVadSession(vadInfo.modelUrl);

// 2. 建立初始狀態
let vadState = createVadState();

// 3. 處理音訊塊（16kHz，Float32Array）
const audioChunk = new Float32Array(512); // 32ms at 16kHz
const vadResult = await processVad(
  vadSession,
  vadState,
  audioChunk,
  DEFAULT_VAD_PARAMS
);

// 4. 更新狀態以供下次迭代
vadState = vadResult.state;

// 5. 檢查是否檢測到語音
if (vadResult.detected) {
  console.log('檢測到語音！', vadResult.score);
}

// 1. 載入喚醒詞模型（使用新的 API）
const config = new ConfigManager();
const wwResources = await loadWakewordResources('hey_jarvis', config);

// 2. 建立初始狀態
let wwState = createWakewordState(wwResources.dims);

// 3. 處理音訊塊（16kHz，Float32Array）
const audioChunk = new Float32Array(1280); // 80ms at 16kHz
const wwResult = await processWakewordChunk(
  wwResources,
  wwState,
  audioChunk,
  { threshold: config.wakeword.hey_jarvis.threshold }
);

// 4. 更新狀態以供下次迭代
wwState = wwResult.state;

// 5. 檢查是否檢測到喚醒詞
if (wwResult.triggered) {
  console.log('檢測到喚醒詞！', wwResult.score);
  // 檢測後重設狀態
  wwState = resetWakewordState(wwResources.dims);
}

// 1. 載入 Whisper 模型
const whisperInfo = resolveWhisper(registry, 'whisper-base');
const whisperResources = await loadWhisperResources(
  whisperInfo.path,
  { quantized: whisperInfo.quantized }
);

// 2. 轉錄音訊（16kHz，Float32Array）
const audioData = new Float32Array(16000 * 5); // 5 秒音訊
const result = await transcribe(
  whisperResources,
  audioData,
  {
    language: 'zh',  // 支援中文
    task: 'transcribe',
    returnSegments: true,
  }
);

console.log('轉錄結果：', result.text);
if (result.segments) {
  result.segments.forEach(segment => {
    console.log(`[${segment.start}-${segment.end}]: ${segment.text}`);
  });
}

import * as WebASRCore from 'web-asr-core';

async function createVoiceAssistant() {
  // 載入註冊表和模型
  const registry = await WebASRCore.loadRegistry('./models/global_registry.json');
  const config = new WebASRCore.ConfigManager();
  
  // 初始化 VAD
  const vadInfo = WebASRCore.resolveVad(registry);
  const vadSession = await WebASRCore.loadVadSession(vadInfo.modelUrl);
  let vadState = WebASRCore.createVadState();
  
  // 初始化喚醒詞
  const wwResources = await WebASRCore.loadWakewordResources('hey_jarvis', config);
  let wwState = WebASRCore.createWakewordState(wwResources.dims);
  
  // 初始化 Whisper
  const whisperInfo = WebASRCore.resolveWhisper(registry, 'whisper-base');
  const whisperResources = await WebASRCore.loadWhisperResources(
    whisperInfo.path,
    { quantized: true }
  );
  
  // 音訊收集緩衝區
  const audioBuffer: Float32Array[] = [];
  let isListening = false;
  
  // 處理音訊流（每 80ms 處理新的音訊塊）
  async function processAudioChunk(chunk: Float32Array) {
    // 未監聽時檢查喚醒詞
    if (!isListening) {
      const wwResult = await WebASRCore.processWakewordChunk(
        wwResources,
        wwState,
        chunk,
        { threshold: config.wakeword.hey_jarvis.threshold }
      );
      wwState = wwResult.state;
      
      if (wwResult.triggered) {
        console.log('檢測到喚醒詞！開始監聽...');
        isListening = true;
        audioBuffer.length = 0;
        wwState = WebASRCore.resetWakewordState(wwResources.dims);
      }
      return;
    }
    
    // 使用 VAD 檢測語音
    const vadResult = await WebASRCore.processVad(
      vadSession,
      vadState,
      chunk,
      WebASRCore.DEFAULT_VAD_PARAMS
    );
    vadState = vadResult.state;
    
    // 語音活動時收集音訊
    if (vadResult.detected || vadState.isSpeechActive) {
      audioBuffer.push(chunk);
    }
    
    // 語音結束時進行轉錄
    if (!vadState.isSpeechActive && audioBuffer.length > 0) {
      // 合併音訊塊
      const totalLength = audioBuffer.reduce((sum, chunk) => sum + chunk.length, 0);
      const combinedAudio = new Float32Array(totalLength);
      let offset = 0;
      for (const chunk of audioBuffer) {
        combinedAudio.set(chunk, offset);
        offset += chunk.length;
      }
      
      // 轉錄
      const result = await WebASRCore.transcribe(
        whisperResources,
        combinedAudio,
        { language: 'zh' }
      );
      
      console.log('您說：', result.text);
      
      // 重設以進行下次互動
      audioBuffer.length = 0;
      isListening = false;
    }
  }
  
  return { processAudioChunk };
}

// 與 Web Audio API 配合使用
async function startRecording() {
  const assistant = await createVoiceAssistant();
  
  // 取得麥克風權限（關閉音訊處理以獲得原始音訊）
  const stream = await navigator.mediaDevices.getUserMedia({ 
    audio: {
      channelCount: 1,
      sampleRate: 16000,
      echoCancellation: false,
      noiseSuppression: false,
      autoGainControl: false
    }
  });
  
  const audioContext = new AudioContext({ sampleRate: 16000 });
  const source = audioContext.createMediaStreamSource(stream);
  
  // 建立處理器處理 80ms 塊
  const processor = audioContext.createScriptProcessor(1280, 1, 1);
  
  processor.onaudioprocess = async (e) => {
    const inputData = e.inputBuffer.getChannelData(0);
    await assistant.processAudioChunk(new Float32Array(inputData));
  };
  
  source.connect(processor);
  processor.connect(audioContext.destination);
}

import { ConfigManager } from 'web-asr-core';

const config = new ConfigManager();

// 自訂 VAD 參數
config.vad.threshold = 0.6;
config.vad.minSilenceDuration = 1000;

// 自訂喚醒詞參數
config.wakeword.hey_jarvis.threshold = 0.5;
config.wakeword.common.melFramesPerChunk = 5;

// 自訂 Whisper 參數
config.whisper.temperature = 0.2;
config.whisper.maxLength = 448;