#include <iostream>

#include <vector>

#include <cstdint>

#include <whisper.h>

#include "../stream/stream_components_service.h"

#include "../stream/stream_components.h"

#include "../common/utils.h"

#include "../common/common.h"

#include <speex/speex_preprocess.h>

using namespace stream_components;

int main() {

std::string wav_file_path = "../samples/jfk.wav"; // 替换为您的 WAV 文件路径

// audio arrays

std::vector<float> pcmf32; // mono-channel F32 PCM

std::vector<std::vector<float>> pcmf32s; // stereo-channel F32 PCM

::read_wav(wav_file_path, pcmf32, pcmf32s, false);

printf("size of samples %lu\n", pcmf32.size());

whisper_local_stream_params params;

struct whisper_context_params cparams{};

cparams.use_gpu = params.service.use_gpu;

//Instantiate the service

stream_components::WhisperService whisperService(params.service, params.audio, cparams);

//Simulate websokcet by adding 1500 data each time.

std::vector<float> audio_buffer;

int chunk_size = 160; // 适用于 16 kHz 采样率的 100 毫秒帧

SpeexPreprocessState *st = speex_preprocess_state_init(chunk_size, WHISPER_SAMPLE_RATE);

int vad = 1;

speex_preprocess_ctl(st, SPEEX_PREPROCESS_SET_VAD, &vad);

bool last_is_speech = false;

// 处理音频帧

for (size_t i = 0; i < pcmf32.size(); i += chunk_size) {

spx_int16_t frame[chunk_size];

for (int j = 0; j < chunk_size; ++j) {

if (i + j < pcmf32.size()) {

frame[j] = (spx_int16_t)(pcmf32[i + j] * 32768);

} else {

frame[j] = 0; // 对于超出范围的部分填充 0

}

}

int is_speech = speex_preprocess_run(st, frame);

// 将当前帧添加到 audio_buffer

audio_buffer.insert(audio_buffer.end(), pcmf32.begin() + i, pcmf32.begin() + std::min(i + chunk_size, pcmf32.size()));

printf("is_speech %d \n",is_speech);

if (!is_speech && last_is_speech) {

bool b = whisperService.process(pcmf32.data(), pcmf32.size());

const nlohmann::json &json_array = get_result(whisperService.ctx);

const std::basic_string<char, std::char_traits<char>, std::allocator<char>> &string = json_array.dump();

printf("%s\n",string.c_str());

return 0;

audio_buffer.clear();

}

last_is_speech = is_speech != 0;

}

speex_preprocess_state_destroy(st);

}