jarvis/app/src/stt.rs

mod vosk;

use crate::config;
use once_cell::sync::OnceCell;

use crate::config::structs::SpeechToTextEngine;

static STT_TYPE: OnceCell<SpeechToTextEngine> = OnceCell::new();

pub fn init() -> Result<(), ()> {
    if !STT_TYPE.get().is_none() {
        return Ok(());
    } // already initialized

    // set default stt type
    // @TODO. Make it configurable?
    STT_TYPE.set(config::DEFAULT_SPEECH_TO_TEXT_ENGINE).unwrap();

    // load given recorder
    match STT_TYPE.get().unwrap() {
        SpeechToTextEngine::Vosk => {
            // Init Vosk
            info!("Initializing Vosk STT backend.");
            vosk::init_vosk();
            info!("STT backend initialized.");
        }
    }

    Ok(())
}

pub fn recognize(data: &[i16], partial: bool) -> Option<String> {
    match STT_TYPE.get().unwrap() {
        SpeechToTextEngine::Vosk => vosk::recognize(data, partial),
    }
}