perf: improve logger

ckaznable · May 25, 2023 · 926596f · 926596f
1 parent f5521fc
commit 926596f
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -16,6 +16,7 @@ clap = { version = "4.2.7", features = ["derive"] }
 ringbuf = "0.3.3"
 tract-onnx = "0.17.9"
 rubato = "0.12.0"
+owo-colors = "3.5.0"
 
 [profile.release]
 opt-level = 'z'     # Optimize for size

diff --git a/src/main.rs b/src/main.rs
@@ -1,5 +1,6 @@
 use audio::{resample_to_16k, YOUTUBE_TS_SAMPLE_RATE};
 use clap::Parser;
+use owo_colors::OwoColorize;
 use ringbuf::{Consumer, HeapRb, LocalRb, Producer, Rb, SharedRb};
 use speech::{SpeechConfig, WhisperPayload};
 use std::{
@@ -241,7 +242,7 @@ fn evoke_whisper_thread(
                 speech::process(&mut state, &mut payload, &mut |segment, start| {
                     println!(
                         "[{}] {}",
-                        util::format_timestamp_to_time(segment_time + start),
+                        util::format_timestamp_to_time(segment_time + start).bright_yellow(),
                         segment
                     );
                 });

diff --git a/src/util.rs b/src/util.rs
@@ -1,3 +1,5 @@
+use owo_colors::OwoColorize;
+
 #[derive(Clone, Default)]
 pub struct Log {
     enable: bool,
@@ -10,13 +12,13 @@ impl Log {
 
     pub fn verbose<S: AsRef<str>>(&self, msg: S) {
         if self.enable {
-            println!("[verbose] {}", msg.as_ref());
+            println!("{} {}", "[verbose]".green(), msg.as_ref().green());
         }
     }
 
     pub fn error<S: AsRef<str>>(&self, msg: S) {
         if self.enable {
-            println!("[error] {}", msg.as_ref());
+            println!("{} {}", "[error]".red(), msg.as_ref().red().bold());
         }
     }
 }