pocket_tts_cli/server/
mod.rs

1//! HTTP API Server
2//!
3//! Axum-based server providing TTS generation endpoints.
4
5use anyhow::Result;
6use pocket_tts::TTSModel;
7
8use crate::commands::serve::{ServeArgs, print_endpoints};
9use crate::voice::resolve_voice;
10
11pub mod handlers;
12pub mod routes;
13pub mod state;
14
15pub async fn start_server(args: ServeArgs) -> Result<()> {
16    // Initialize tracing
17    let _ = tracing_subscriber::fmt::try_init();
18
19    // Load model with configured parameters
20    let model = if args.quantized {
21        #[cfg(feature = "quantized")]
22        {
23            TTSModel::load_quantized_with_params(
24                &args.variant,
25                args.temperature,
26                args.lsd_decode_steps,
27                args.eos_threshold,
28            )?
29        }
30        #[cfg(not(feature = "quantized"))]
31        {
32            anyhow::bail!("Quantization feature not enabled. Rebuild with --features quantized");
33        }
34    } else {
35        TTSModel::load_with_params(
36            &args.variant,
37            args.temperature,
38            args.lsd_decode_steps,
39            args.eos_threshold,
40        )?
41    };
42
43    println!(
44        "  {} Model loaded (sample rate: {}Hz)",
45        "✓".to_string(),
46        model.sample_rate
47    );
48
49    // Pre-load default voice
50    println!("  Loading default voice: {}...", args.voice);
51    let default_voice_state = resolve_voice(&model, Some(&args.voice))?;
52    println!("  {} Default voice ready", "✓");
53
54    let state = state::AppState::new(model, default_voice_state);
55    let app = routes::create_router(state);
56
57    let addr = format!("{}:{}", args.host, args.port);
58
59    print_endpoints(&args.host, args.port);
60
61    let listener = tokio::net::TcpListener::bind(&addr).await?;
62    axum::serve(listener, app).await?;
63
64    Ok(())
65}
pocket_tts_cli/server/mod.rs

pocket_tts_cli/server/
mod.rs
