transcribe-rs 0.3.9

A simple library to help you transcribe audio
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

use super::SAMPLE_RATE;
use crate::{
    ModelCapabilities, SpeechModel, TranscribeError, TranscribeOptions, TranscriptionResult,
    TranscriptionSegment,
};

/// Mock model that returns the sample count as text.
pub(crate) struct MockModel;

impl SpeechModel for MockModel {
    fn capabilities(&self) -> ModelCapabilities {
        ModelCapabilities {
            name: "mock",
            engine_id: "mock",
            sample_rate: 16000,
            languages: &[],
            supports_timestamps: false,
            supports_translation: false,
            supports_streaming: false,
        }
    }

    fn transcribe_raw(
        &mut self,
        samples: &[f32],
        _options: &TranscribeOptions,
    ) -> Result<TranscriptionResult, TranscribeError> {
        Ok(TranscriptionResult {
            text: format!("chunk_{}", samples.len()),
            segments: Some(vec![TranscriptionSegment {
                start: 0.0,
                end: samples.len() as f32 / SAMPLE_RATE,
                text: format!("chunk_{}", samples.len()),
            }]),
        })
    }
}

/// Mock model that fails on the Nth call.
pub(crate) struct FailOnNthModel {
    call_count: usize,
    fail_on: usize,
}

impl FailOnNthModel {
    pub(crate) fn new(fail_on: usize) -> Self {
        Self {
            call_count: 0,
            fail_on,
        }
    }
}

impl SpeechModel for FailOnNthModel {
    fn capabilities(&self) -> ModelCapabilities {
        ModelCapabilities {
            name: "mock",
            engine_id: "mock",
            sample_rate: 16000,
            languages: &[],
            supports_timestamps: false,
            supports_translation: false,
            supports_streaming: false,
        }
    }

    fn transcribe_raw(
        &mut self,
        _samples: &[f32],
        _options: &TranscribeOptions,
    ) -> Result<TranscriptionResult, TranscribeError> {
        self.call_count += 1;
        if self.call_count == self.fail_on {
            return Err(TranscribeError::Inference("mock failure".into()));
        }
        Ok(TranscriptionResult {
            text: format!("chunk_{}", self.call_count),
            segments: None,
        })
    }
}

pub(crate) fn make_speech(frame_size: usize, num_frames: usize) -> Vec<f32> {
    vec![1.0f32; frame_size * num_frames]
}

pub(crate) fn make_silence(frame_size: usize, num_frames: usize) -> Vec<f32> {
    vec![0.0f32; frame_size * num_frames]
}