whisper-rs 0.13.0

Rust bindings for whisper.cpp
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

use log::{debug, error, info, warn};
use whisper_rs_sys::ggml_log_level;

unsafe extern "C" fn whisper_cpp_log_trampoline(
    level: ggml_log_level,
    text: *const std::os::raw::c_char,
    _: *mut std::os::raw::c_void, // user_data
) {
    if text.is_null() {
        error!("whisper_cpp_log_trampoline: text is nullptr");
    }

    // SAFETY: we must trust whisper.cpp that it will not pass us a string that does not satisfy
    // from_ptr's requirements.
    let log_str = unsafe { std::ffi::CStr::from_ptr(text) }.to_string_lossy();
    // whisper.cpp gives newlines at the end of its log messages, so we trim them
    let trimmed = log_str.trim();

    match level {
        whisper_rs_sys::ggml_log_level_GGML_LOG_LEVEL_DEBUG => debug!("{}", trimmed),
        whisper_rs_sys::ggml_log_level_GGML_LOG_LEVEL_INFO => info!("{}", trimmed),
        whisper_rs_sys::ggml_log_level_GGML_LOG_LEVEL_WARN => warn!("{}", trimmed),
        whisper_rs_sys::ggml_log_level_GGML_LOG_LEVEL_ERROR => error!("{}", trimmed),
        _ => {
            warn!(
                "whisper_cpp_log_trampoline: unknown log level {}: message: {}",
                level, trimmed
            )
        }
    }
}

/// Shortcut utility to redirect all whisper.cpp logging to the `log` crate.
///
/// Filter for logs from the `whisper-rs` crate to see all log output from whisper.cpp.
///
/// You should only call this once (subsequent calls have no ill effect).
pub fn install_whisper_log_trampoline() {
    crate::LOG_TRAMPOLINE_INSTALL.call_once(|| unsafe {
        whisper_rs_sys::whisper_log_set(Some(whisper_cpp_log_trampoline), std::ptr::null_mut());
    });
}