speechcore 0.1.0

Reusable Rust speech-to-text runtime with audio capture, VAD, backend selection, model provisioning, and transcript streaming.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use std::fs::OpenOptions;
use std::io::Write;

/// Stores statistics about transcription performance
#[derive(Default, Clone)]
pub struct TranscriptionStats {
    pub segments_processed: usize,
    pub total_audio_duration: f32,
    pub total_inference_time: f32,
    pub total_processing_time: f32,
    pub min_rtf: f32,
    pub max_rtf: f32,
    pub avg_rtf: f32,
    pub audio_channel_drops: u64,
    pub segment_channel_drops: u64,
}

impl TranscriptionStats {
    pub fn new() -> Self {
        Self {
            segments_processed: 0,
            total_audio_duration: 0.0,
            total_inference_time: 0.0,
            total_processing_time: 0.0,
            min_rtf: f32::MAX,
            max_rtf: 0.0,
            avg_rtf: 0.0,
            audio_channel_drops: 0,
            segment_channel_drops: 0,
        }
    }

    pub fn update(&mut self, segment_duration: f32, inference_time: f32, processing_time: f32) {
        let rtf = inference_time / segment_duration;

        self.segments_processed += 1;
        self.total_audio_duration += segment_duration;
        self.total_inference_time += inference_time;
        self.total_processing_time += processing_time;

        self.min_rtf = self.min_rtf.min(rtf);
        self.max_rtf = self.max_rtf.max(rtf);
        self.avg_rtf = self.total_inference_time / self.total_audio_duration;
    }

    pub fn record_audio_drop(&mut self, count: u64) -> u64 {
        self.audio_channel_drops += count;
        self.audio_channel_drops
    }

    pub fn record_segment_drop(&mut self, count: u64) -> u64 {
        self.segment_channel_drops += count;
        self.segment_channel_drops
    }

    pub fn report(&self) -> String {
        format!(
            "Transcription Statistics:\n\
             - Segments processed: {}\n\
             - Total audio duration: {:.2}s\n\
             - Total inference time: {:.2}s\n\
             - Total processing time: {:.2}s\n\
             - Average real-time factor (RTF): {:.2}x\n\
             - Min RTF: {:.2}x\n\
             - Max RTF: {:.2}x\n\
             - Audio channel drops: {}\n\
             - Segment channel drops: {}",
            self.segments_processed,
            self.total_audio_duration,
            self.total_inference_time,
            self.total_processing_time,
            self.avg_rtf,
            if self.min_rtf == f32::MAX {
                0.0
            } else {
                self.min_rtf
            },
            self.max_rtf,
            self.audio_channel_drops,
            self.segment_channel_drops
        )
    }

    /// Logs the statistics to a file
    pub fn log_to_file(&self, is_final: bool) {
        if self.segments_processed > 0 {
            let stats_report = self.report();

            // Write to file
            let timestamp = chrono::Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
            let report_type = if is_final {
                "Final Report"
            } else {
                "Periodic Report"
            };
            let file_content = format!(
                "\n--- {} ({}) ---\n{}\n",
                timestamp, report_type, stats_report
            );

            match OpenOptions::new()
                .append(true)
                .create(true)
                .open("transcription_stats.log")
            {
                Ok(mut file) => {
                    if let Err(e) = writeln!(file, "{}", file_content) {
                        tracing::warn!("Failed to write to stats file: {}", e);
                    }
                }
                Err(e) => tracing::warn!("Failed to open stats file: {}", e),
            }
        }
    }
}