piper-rs 0.2.0

Use piper TTS models in Rust
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

/*
git submodule update --init

wget https://huggingface.co/rhasspy/piper-voices/resolve/main/en/en_US/libritts_r/medium/en_US-libritts_r-medium.onnx
wget https://huggingface.co/rhasspy/piper-voices/resolve/main/en/en_US/libritts_r/medium/en_US-libritts_r-medium.onnx.json
cargo run --example usage en_US-libritts_r-medium.onnx.json 80
*/

use piper_rs::Piper;
use rodio::buffer::SamplesBuffer;
use std::path::Path;

fn main() {
    let config_path = std::env::args().nth(1).expect("Please specify config path");
    let speaker_id: Option<i64> = std::env::args()
        .nth(2)
        .map(|s| s.parse().expect("Speaker ID must be a number"));

    let onnx_path = config_path.replace(".onnx.json", ".onnx");
    let mut piper = Piper::new(Path::new(&onnx_path), Path::new(&config_path)).unwrap();

    let text = "Hello! I'm playing audio from memory directly with piper-rs.";
    let (samples, sample_rate) = piper
        .create(text, false, speaker_id, None, None, None)
        .unwrap();

    let (_stream, handle) = rodio::OutputStream::try_default().unwrap();
    let sink = rodio::Sink::try_new(&handle).unwrap();
    sink.append(SamplesBuffer::new(1, sample_rate, samples));
    sink.sleep_until_end();
}