fix so each buffer page isnt stream_step from last BUT stream_step of content

This commit is contained in:
Bel LaPointe
2023-12-20 23:43:06 -05:00
parent 6bb29c106f
commit e74a5ac0ca
2 changed files with 12 additions and 15 deletions

View File

@@ -1,5 +1,4 @@
use whisper_rs::{WhisperContext, FullParams, SamplingStrategy, WhisperError};
use std::time::{Duration, Instant};
use clap::Parser;
use std::thread;
use std::fs::File;
@@ -63,16 +62,14 @@ pub fn channel<F>(flags: Flags, handler_fn: F, stream: std::sync::mpsc::Receiver
handler_fn,
).unwrap();
let stream_retain = (flags.stream_retain * 16_000.0) as usize;
let stream_step = Duration::new(flags.stream_step, 0);
match &flags.debug {
true => { File::create("/tmp/page.rawf32audio").unwrap(); },
false => {},
};
let mut buffer = vec![];
let mut last = Instant::now();
for data in stream.iter() {
data.iter().for_each(|x| buffer.push(*x));
if Instant::now() - last > stream_step {
if buffer.len() >= (flags.stream_step * 16_000) as usize {
w.transcribe_async(&buffer).unwrap();
match &flags.debug {
@@ -93,7 +90,6 @@ pub fn channel<F>(flags: Flags, handler_fn: F, stream: std::sync::mpsc::Receiver
buffer[i] = buffer[buffer.len() - stream_retain + i];
}
buffer.truncate(stream_retain);
last = Instant::now();
}
}
}