wehub-resource-sync
82 lines
2.6 KiB
Rust
82 lines
2.6 KiB
Rust
//! Whisper transcription example demonstrating audio transcription.
|
|
//!
|
|
//! Run with:
|
|
//! ```bash
|
|
//! cargo run --example test_whisper --features whisper -- /path/to/audio
|
|
//! ```
|
|
|
|
#[cfg(feature = "whisper")]
|
|
use memvid_core::{WhisperConfig, WhisperTranscriber};
|
|
#[cfg(feature = "whisper")]
|
|
use std::env;
|
|
#[cfg(feature = "whisper")]
|
|
use std::path::PathBuf;
|
|
#[cfg(feature = "whisper")]
|
|
use std::time::Instant;
|
|
|
|
#[cfg(not(feature = "whisper"))]
|
|
fn main() {
|
|
eprintln!(
|
|
"This example requires the `whisper` feature.\n\
|
|
Re-run with:\n\
|
|
cargo run --example test_whisper --features whisper -- /path/to/audio"
|
|
);
|
|
}
|
|
|
|
#[cfg(feature = "whisper")]
|
|
fn main() {
|
|
// Get audio path from args
|
|
let args: Vec<String> = env::args().collect();
|
|
if args.len() < 2 {
|
|
eprintln!("Usage: cargo run --example test_whisper --features whisper -- /path/to/audio");
|
|
eprintln!("\nExample:");
|
|
eprintln!(
|
|
" cargo run --example test_whisper --features whisper -- examples/call_sale.mp3"
|
|
);
|
|
return;
|
|
}
|
|
|
|
let audio_path = PathBuf::from(&args[1]);
|
|
|
|
if !audio_path.exists() {
|
|
eprintln!("ERROR: Audio file not found at {:?}", audio_path);
|
|
eprintln!("Usage: cargo run --example test_whisper --features whisper -- /path/to/audio");
|
|
return;
|
|
}
|
|
|
|
println!("=== Whisper Transcription Example ===\n");
|
|
println!("Creating Whisper transcriber...");
|
|
let start = Instant::now();
|
|
let config = WhisperConfig::default();
|
|
println!("Model dir: {:?}", config.models_dir);
|
|
println!("Model name: {}", config.model_name);
|
|
|
|
let mut transcriber = WhisperTranscriber::new(&config).expect("Failed to create transcriber");
|
|
println!("Transcriber created in {:?}", start.elapsed());
|
|
|
|
println!("\nTranscribing audio file: {}", audio_path.display());
|
|
let start = Instant::now();
|
|
|
|
match transcriber.transcribe_file(&audio_path) {
|
|
Ok(result) => {
|
|
println!("Transcription completed in {:?}", start.elapsed());
|
|
println!("\n=== Transcription Result ===");
|
|
println!("Duration: {:.2} seconds", result.duration_secs);
|
|
println!("Language: {}", result.language);
|
|
println!("\nText:\n{}", result.text);
|
|
|
|
if !result.segments.is_empty() {
|
|
println!("\n=== Segments ===");
|
|
for seg in &result.segments {
|
|
println!("[{:.2}s - {:.2}s] {}", seg.start, seg.end, seg.text);
|
|
}
|
|
}
|
|
}
|
|
Err(e) => {
|
|
eprintln!("Transcription failed: {}", e);
|
|
}
|
|
}
|
|
|
|
println!("\n=== Transcription example completed! ===");
|
|
}
|