From 97c025f04d45f714c5b7bb033a754a11952dcf66 Mon Sep 17 00:00:00 2001
From: bel <bel@bel.bel>
Date: Mon, 1 Jan 2024 19:06:14 -0700
Subject: [PATCH] rust-whisper-baked works with WAV

---
 rust-whisper-baked/src/main.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/rust-whisper-baked/src/main.rs b/rust-whisper-baked/src/main.rs
index e5c7d2e..6fea6f4 100644
--- a/rust-whisper-baked/src/main.rs
+++ b/rust-whisper-baked/src/main.rs
@@ -6,6 +6,24 @@ use std::thread;
 
 fn main() {
    let flags = rust_whisper_lib::Flags::parse();
+   match flags.wav.clone() {
+      Some(path) => wav(flags, path),
+      None => channel(flags),
+   };
+}
+
+fn wav(flags: rust_whisper_lib::Flags, _path: String) {
+   rust_whisper_baked_lib::wav(flags,
+      |result: Result<rust_whisper_lib::Transcribed, String>| {
+         match result {
+            Ok(transcribed) => { println!("{}", transcribed.to_string()); },
+            Err(msg) => { eprintln!("error: {}", msg); },
+         };
+      },
+   );
+}
+
+fn channel(flags: rust_whisper_lib::Flags) {
    let (send, recv) = std::sync::mpsc::sync_channel(100);
 
    eprintln!("rust whisper baked lib channel...");