deepgram-rust-captions 0.1.2

Rust implementation of Deepgram's WebVTT and SRT formatting
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

use crate::{converters::DeepgramConverter, helpers::seconds_to_timestamp};

/// Convert a Deepgram response to an SRT string.
pub fn srt(converter: &DeepgramConverter, line_length: Option<u8>) -> String {
    let mut output = Vec::<String>::new();

    let line_length = line_length.unwrap_or(8);

    let lines = converter.get_lines(line_length);

    let mut entry = 1;

    let mut current_speaker = None;

    for words in lines {
        output.push(entry.to_string());
        entry += 1;

        let first_word = words.first().unwrap();
        let last_word = words.last().unwrap();

        let start_time = seconds_to_timestamp(first_word.start, "%H:%M:%S,%3f");
        let end_time = seconds_to_timestamp(last_word.end, "%H:%M:%S,%3f");

        output.push(format!("{} --> {}", start_time, end_time));

        if first_word.speaker.is_some()
            && (current_speaker.is_none() || current_speaker != first_word.speaker)
        {
            current_speaker = first_word.speaker;
            output.push(format!("[speaker {}]", current_speaker.unwrap()));
        }

        let punctuated_words = words
            .iter()
            .map(|word| {
                word.punctuated_word
                    .as_ref()
                    .unwrap_or(&word.word)
                    .to_string()
            })
            .collect::<Vec<String>>();

        output.push(punctuated_words.join(" "));
        output.push("".to_string());
    }

    output.join("\n")
}