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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
//! # Voice Toolkit - Rust 语音处理工具库
//!
//! `voice-toolkit` 是一个功能强大的 Rust 语音处理工具库,提供统一的接口来处理语音转文本(STT)、
//! 文本转语音(TTS)和音频处理任务。该库基于 Whisper 模型提供高质量的语音识别功能,
//! 支持多种音频格式转换和实时语音处理。
//!
//! ## 主要特性
//!
//! - **语音转文本 (STT)**: 基于 OpenAI Whisper 模型的高质量语音识别
//! - **文本转语音 (TTS)**: 基于 Index-TTS 的语音合成功能
//! - **音频处理**: 支持多种音频格式的转换和处理
//! - **实时流式处理**: 支持实时音频流的转录
//! - **语音活动检测 (VAD)**: 智能检测语音片段,提高处理效率
//! - **跨平台支持**: 支持 Windows、macOS 和 Linux
//! - **GPU 加速**: 可选的 CUDA、Vulkan 和 Metal 加速支持
//!
//! ## 快速开始
//!
//! ### 基本依赖
//!
//! 在 `Cargo.toml` 中添加依赖:
//!
//! ```toml
//! [dependencies]
//! voice-toolkit = { version = "0.15.0", features = ["stt", "tts", "audio"] }
//! ```
//!
//! ### 语音转文本示例
//!
//! ```rust
//! use voice_toolkit::transcribe_file_unified;
//!
//! #[tokio::main]
//! async fn main() -> Result<(), Box<dyn std::error::Error>> {
//! let model_path = "models/ggml-base.bin";
//! let audio_path = "audio/hello.wav";
//!
//! let result = transcribe_file_unified(model_path, audio_path).await?;
//! println!("转录结果: {}", result.text);
//!
//! Ok(())
//! }
//! ```
//!
//! ### 音频格式转换示例
//!
//! ```rust
//! use voice_toolkit::audio;
//!
//! #[tokio::main]
//! async fn main() -> Result<(), Box<dyn std::error::Error>> {
//! let input_path = "audio/input.mp3";
//! let output_path = "audio/output.wav";
//!
//! // 将 MP3 转换为 Whisper 兼容的 WAV 格式
//! audio::convert_to_whisper_format(input_path, output_path).await?;
//! println!("转换完成: {}", output_path);
//!
//! Ok(())
//! }
//! ```
//!
//! ### 文本转语音示例
//!
//! ```rust
//! use voice_toolkit::tts;
//!
//! #[tokio::main]
//! async fn main() -> Result<(), Box<dyn std::error::Error>> {
//! let text = "你好,世界!欢迎使用语音工具库。";
//! let output_path = "output/hello.wav";
//!
//! // 使用 Index-TTS 生成语音
//! tts::synthesize_text(text, output_path, None).await?;
//! println!("语音合成完成: {}", output_path);
//!
//! Ok(())
//! }
//! ```
//!
//! ## 功能特性
//!
//! ### STT 功能
//! - 支持多种音频格式:WAV、MP3、FLAC、M4A、OGG 等
//! - 自动音频格式转换和预处理
//! - 支持实时流式转录
//! - 集成语音活动检测 (VAD)
//! - 性能监控和基准测试
//!
//! ### TTS 功能
//! - 基于 Index-TTS 的高质量语音合成
//! - 支持多种输出格式
//! - 可扩展的引擎架构
//!
//! ### 音频处理功能
//! - 音频格式转换
//! - 音频重采样
//! - 元数据提取
//! - Whisper 兼容格式转换
//!
//! ## 特性标志
//!
//! 该库使用特性标志来控制功能模块的启用:
//!
//! - `stt`: 启用语音转文本功能(默认启用)
//! - `tts`: 启用文本转语音功能
//! - `audio`: 启用音频处理功能(默认启用)
//! - `streaming`: 启用实时流式处理(需要 `stt`)
//! - `cuda`: 启用 CUDA GPU 加速(需要 `stt`)
//! - `vulkan`: 启用 Vulkan GPU 加速(需要 `stt`)
//! - `metal`: 启用 Metal GPU 加速(需要 `stt`)
//!
//! ## 系统要求
//!
//! - **Rust**: 1.70 或更高版本
//! - **FFmpeg**: 用于音频处理
//! - macOS: `brew install ffmpeg`
//! - Ubuntu: `sudo apt-get install ffmpeg`
//! - Windows: 使用 vcpkg 安装
//! - **Whisper 模型**: 需要下载 Whisper 模型文件(.bin 格式)
//!
//! ## 错误处理
//!
//! 该库使用统一的错误处理机制,所有函数都返回 `Result<T, Error>` 类型,
//! 其中 `Error` 是一个枚举类型,包含了所有可能的错误情况。
//!
//! ## 性能考虑
//!
//! - 首次加载模型时会有一定的延迟
//! - 建议在长期运行的应用中复用模型实例
//! - 使用 GPU 加速可以显著提高处理速度
//! - 对于实时应用,建议使用流式处理功能
//!
//! ## 许可证
//!
//! 本项目采用 MIT 或 Apache 2.0 许可证。详情请参阅 [LICENSE](LICENSE) 文件。
//!
//! ## 贡献
//!
//! 欢迎提交 Issue 和 Pull Request!请参阅 [CONTRIBUTING.md](CONTRIBUTING.md) 了解详情。
//!
//! ## 更新日志
//!
//! 请参阅 [CHANGELOG.md](CHANGELOG.md) 了解版本更新详情。
// 导入错误处理模块
pub use ;
// 重新导出各个模块
/// 语音转文本 (STT) 模块
///
/// 该模块提供基于 OpenAI Whisper 模型的语音识别功能,包括:
/// - 文件转录:处理音频文件并转换为文本
/// - 流式转录:实时处理音频流
/// - 语音活动检测 (VAD):智能检测语音片段
/// - 性能监控:提供转录性能指标
pub use rs_voice_toolkit_stt as stt;
/// 音频处理模块
///
/// 该模块提供音频文件的格式转换、重采样和元数据提取功能,包括:
/// - 多格式支持:WAV、MP3、FLAC、M4A、OGG 等
/// - 音频重采样:支持多种采样率转换
/// - 元数据提取:获取音频文件的详细信息
/// - Whisper 兼容格式:自动转换为 Whisper 识别所需的格式
pub use rs_voice_toolkit_audio as audio;
/// 文本转语音 (TTS) 模块
///
/// 该模块提供文本到语音的转换功能,包括:
/// - Index-TTS 引擎:基于 Index-TTS 的高质量语音合成
/// - 多种输出格式:支持 WAV、MP3 等格式
/// - 可扩展架构:支持多种 TTS 引擎
/// - 灵活配置:支持语音速度、音调等参数调整
pub use rs_voice_toolkit_tts as tts;
// 重新导出常用的类型和函数
/// 重新导出文件转录函数
///
/// 这是 STT 模块的核心函数,用于转录音频文件。
/// 详见 [`stt::transcribe_file`] 函数文档。
pub use transcribe_file;
/// 重新导出流式转录器
///
/// 用于实时音频流转录的结构体。
/// 详见 [`stt::streaming::StreamingTranscriber`] 结构体文档。
pub use StreamingTranscriber;
// 统一错误处理包装函数
/// 导出统一错误处理函数
///
/// 这是推荐使用的文件转录函数,提供了统一的错误处理接口。
/// 详见 [`transcribe_file_unified`] 函数文档。
pub use transcribe_file_unified;