feat: replace `--pinyin` flag with generic `--post="+pinyin"` option

1 files changed, 4 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index e3a5b84..73b4e06 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,5 @@
+pub mod post_proc;
+
 mod args;
 pub use args::*;
 
@@ -252,7 +254,7 @@ async fn main() {
         match rx.recv().await {
             Ok(event) => match event {
                 vtx_engine::EngineEvent::TranscriptionComplete(result) => {
-                    let raw = maybe_to_pinyin(&result.text, args.use_pinyin);
+                    let raw = run_post_process(&result.text, args.post.as_deref());
                     let formatted = format_output(&pattern, &raw, 0.0, last_volume_db);
 
                     for ch in &channels {
@@ -260,7 +262,7 @@ async fn main() {
                     }
                 }
                 vtx_engine::EngineEvent::TranscriptionSegment(segment) => {
-                    let raw = maybe_to_pinyin(&segment.text, args.use_pinyin);
+                    let raw = run_post_process(&segment.text, args.post.as_deref());
                     let formatted = format_output(&pattern, &raw, 0.0, last_volume_db);
 
                     for ch in &channels {