@webgal-tools/voice
Version:
WebGAL GPT-SoVITS语音合成应用
328 lines • 13.7 kB
JavaScript
import fs from 'fs';
import path from 'path';
import { getVoiceConfig } from '@webgal-tools/config';
import { LANGUAGE_OPTIONS } from './request.js';
import { logger } from '@webgal-tools/logger';
export class VoiceConfigManager {
config = null;
configPath;
constructor(workDirectory) {
const workDir = workDirectory;
this.configPath = path.join(workDir, 'voice.config.json');
}
/**
* 加载语音配置文件
*/
loadConfig() {
// 清除实例缓存，强制重新加载
const timestamp = new Date().toISOString();
logger.info(`[${timestamp}] VoiceConfigManager: 清除实例缓存，从全局配置系统重新加载`);
this.config = null;
// 尝试从新的配置系统加载
const voiceConfig = getVoiceConfig();
if (voiceConfig) {
// 转换配置格式以兼容本地类型
const convertedConfig = {
volume: voiceConfig.volume,
gpt_sovits_url: voiceConfig.gpt_sovits_url,
gpt_sovits_path: voiceConfig.gpt_sovits_path,
model_version: voiceConfig.model_version,
translate: voiceConfig.translate,
characters: voiceConfig.characters
};
this.config = convertedConfig;
const configHash = JSON.stringify(voiceConfig).slice(0, 50) + '...';
logger.info(`[${timestamp}] ✅ 从配置系统加载语音配置`);
logger.info(`[${timestamp}] 配置了 ${voiceConfig.characters.length} 个角色`);
logger.info(`[${timestamp}] 配置摘要: ${configHash}`);
return convertedConfig;
}
// 回退到文件系统加载
if (!fs.existsSync(path.resolve(this.configPath))) {
throw new Error(`语音配置文件不存在: ${this.configPath}\n请创建 voice.config.json 文件或参考示例配置`);
}
try {
const configContent = fs.readFileSync(this.configPath, 'utf-8');
const config = JSON.parse(configContent);
// 验证配置文件格式
this.validateConfig(config);
this.config = config;
console.error(`✅ 成功加载语音配置: ${this.configPath}`);
console.error(`配置了 ${config.characters.length} 个角色`);
return config;
}
catch (error) {
throw new Error(`解析语音配置文件失败: ${error instanceof Error ? error.message : error}`);
}
}
/**
* 验证配置文件格式
*/
validateConfig(config) {
if (!config.volume || typeof config.volume !== 'number') {
throw new Error('配置文件缺少有效的 volume 设置');
}
if (!config.gpt_sovits_url || typeof config.gpt_sovits_url !== 'string') {
throw new Error('配置文件缺少有效的 gpt_sovits_url 设置');
}
if (!config.gpt_sovits_path || typeof config.gpt_sovits_path !== 'string') {
throw new Error('配置文件缺少有效的 gpt_sovits_path 设置');
}
if (!fs.existsSync(config.gpt_sovits_path)) {
throw new Error(`GPT-SoVITS项目路径不存在: ${config.gpt_sovits_path}`);
}
if (!config.model_version || typeof config.model_version !== 'string') {
throw new Error('配置文件缺少有效的 model_version 设置');
}
if (!config.translate) {
throw new Error('配置文件缺少 translate 配置');
}
this.validateTranslateConfig(config.translate);
if (!config.characters || !Array.isArray(config.characters)) {
throw new Error('配置文件缺少 characters 数组');
}
for (const [index, character] of config.characters.entries()) {
const prefix = `角色 ${character.character_name} 配置 [${index}]`;
if (!character.character_name) {
throw new Error(`${prefix} 缺少 character_name`);
}
// 检查是否需要情绪识别
const needsEmotionRecognition = character.auto === true;
if (!character.gpt) {
throw new Error(`${prefix} 缺少 gpt 模型路径`);
}
if (needsEmotionRecognition) {
// 情绪识别模式：验证扫描目录存在
const gptDir = path.resolve(config.gpt_sovits_path, character.gpt);
if (!fs.existsSync(gptDir) || !fs.statSync(gptDir).isDirectory()) {
throw new Error(`${prefix} GPT模型扫描目录不存在或不是目录: ${gptDir}`);
}
}
else {
// 非情绪识别模式：验证具体文件存在
const gptPath = path.resolve(config.gpt_sovits_path, character.gpt);
if (!fs.existsSync(gptPath)) {
throw new Error(`${prefix} GPT模型文件不存在: ${gptPath}`);
}
// 检查文件扩展名
if (!gptPath.endsWith('.ckpt')) {
throw new Error(`${prefix} GPT模型文件必须以 .ckpt 结尾: ${gptPath}`);
}
}
if (!character.sovits) {
throw new Error(`${prefix} 缺少 sovits 模型路径`);
}
if (needsEmotionRecognition) {
// 情绪识别模式：验证扫描目录存在
const sovitsDir = path.resolve(config.gpt_sovits_path, character.sovits);
if (!fs.existsSync(sovitsDir) || !fs.statSync(sovitsDir).isDirectory()) {
throw new Error(`${prefix} SoVITS模型扫描目录不存在或不是目录: ${sovitsDir}`);
}
}
else {
// 非情绪识别模式：验证具体文件存在
const sovitsPath = path.resolve(config.gpt_sovits_path, character.sovits);
if (!fs.existsSync(sovitsPath)) {
throw new Error(`${prefix} SoVITS模型文件不存在: ${sovitsPath}`);
}
// 检查文件扩展名
if (!sovitsPath.endsWith('.pth')) {
throw new Error(`${prefix} SoVITS模型文件必须以 .pth 结尾: ${sovitsPath}`);
}
}
if (!character.ref_audio) {
throw new Error(`${prefix} 缺少参考音频路径`);
}
if (needsEmotionRecognition) {
// 情绪识别模式：验证扫描目录存在
if (!fs.existsSync(character.ref_audio) || !fs.statSync(character.ref_audio).isDirectory()) {
throw new Error(`${prefix} 参考音频扫描目录不存在或不是目录: ${character.ref_audio}`);
}
}
else {
// 非情绪识别模式：验证具体文件存在
if (!fs.existsSync(character.ref_audio)) {
throw new Error(`${prefix} 参考音频文件不存在: ${character.ref_audio}`);
}
// 检查文件扩展名
const audioExt = path.extname(character.ref_audio).toLowerCase();
const supportedAudioFormats = ['.wav', '.mp3', '.flac', '.m4a'];
if (!supportedAudioFormats.includes(audioExt)) {
throw new Error(`${prefix} 参考音频文件格式不支持，支持格式: ${supportedAudioFormats.join(', ')}`);
}
}
// 非情绪识别模式需要参考文本
if (!needsEmotionRecognition && !character.ref_text) {
throw new Error(`${prefix} 非情绪识别模式需要提供参考文本 ref_text`);
}
// 验证推理配置
if (character.inferrence_config) {
this.validateInferenceConfig(character.inferrence_config, prefix);
}
}
}
/**
* 验证翻译配置
*/
validateTranslateConfig(config) {
if (typeof config.check !== 'boolean') {
throw new Error('配置文件的 translate.check 必须是布尔值');
}
// 如果禁用翻译，跳过其他验证
if (!config.check) {
return;
}
if (!config.model_type) {
throw new Error('配置文件缺少 translate.model_type 设置');
}
const validModelTypes = ['ollama', 'openai', 'anthropic', 'google', 'mistral', 'cohere', 'custom'];
if (!validModelTypes.includes(config.model_type)) {
throw new Error(`无效的 model_type: ${config.model_type}。支持的类型: ${validModelTypes.join(', ')}`);
}
if (!config.base_url || typeof config.base_url !== 'string') {
throw new Error('配置文件缺少有效的 translate.base_url 设置');
}
if (!config.model_name || typeof config.model_name !== 'string') {
throw new Error('配置文件缺少有效的 translate.model_name 设置');
}
// 验证API密钥
if (this.requiresApiKey(config)) {
if (!config.api_key || typeof config.api_key !== 'string') {
throw new Error(`${config.model_type} 模型供应商需要提供 api_key`);
}
}
// 验证温度参数
if (config.temperature !== undefined) {
if (typeof config.temperature !== 'number' || config.temperature < 0 || config.temperature > 2) {
throw new Error('translate.temperature 必须在 0-2 之间');
}
}
// 验证最大token数
if (config.max_tokens !== undefined) {
if (typeof config.max_tokens !== 'number' || config.max_tokens < 1 || config.max_tokens > 4000) {
throw new Error('translate.max_tokens 必须在 1-4000 之间');
}
}
}
/**
* 检查是否需要API密钥
*/
requiresApiKey(config) {
// Ollama 本地服务不需要API密钥
if (config.model_type === 'ollama') {
const url = new URL(config.base_url);
return !(url.hostname === 'localhost' || url.hostname === '127.0.0.1' || url.hostname === '::1');
}
// 其他供应商都需要API密钥
return config.model_type !== 'custom';
}
/**
* 验证推理配置
*/
validateInferenceConfig(config, prefix) {
if (config.prompt_language && !(config.prompt_language in LANGUAGE_OPTIONS)) {
throw new Error(`${prefix} 无效的 prompt_language: ${config.prompt_language}`);
}
if (config.text_language && !(config.text_language in LANGUAGE_OPTIONS)) {
throw new Error(`${prefix} 无效的 text_language: ${config.text_language}`);
}
if (config.top_k !== undefined && (config.top_k < 1 || config.top_k > 100)) {
throw new Error(`${prefix} top_k 必须在 1-100 之间`);
}
if (config.top_p !== undefined && (config.top_p < 0 || config.top_p > 1)) {
throw new Error(`${prefix} top_p 必须在 0-1 之间`);
}
if (config.temperature !== undefined && (config.temperature < 0 || config.temperature > 1)) {
throw new Error(`${prefix} temperature 必须在 0-1 之间`);
}
if (config.speed !== undefined && (config.speed < 0.6 || config.speed > 1.65)) {
throw new Error(`${prefix} speed 必须在 0.6-1.65 之间`);
}
}
/**
* 获取角色配置
*/
getCharacterConfig(characterName) {
const config = this.loadConfig();
return config.characters.find(char => char.character_name === characterName) || null;
}
/**
* 获取所有角色名称
*/
getAllCharacterNames() {
const config = this.loadConfig();
return config.characters.map(char => char.character_name);
}
/**
* 获取默认音量设置
*/
getDefaultVolume() {
const config = this.loadConfig();
return config.volume;
}
/**
* 检查角色是否已配置
*/
hasCharacterConfig(characterName) {
return this.getCharacterConfig(characterName) !== null;
}
/**
* 获取配置文件路径
*/
getConfigPath() {
return this.configPath;
}
/**
* 获取GPT-SoVITS服务URL
*/
getGptSovitsUrl() {
const config = this.loadConfig();
return config.gpt_sovits_url;
}
/**
* 获取GPT-SoVITS项目路径
*/
getGptSovitsPath() {
const config = this.loadConfig();
return config.gpt_sovits_path;
}
/**
* 获取模型版本
*/
getModelVersion() {
const config = this.loadConfig();
return config.model_version;
}
/**
* 获取翻译配置
*/
getTranslateConfig() {
const config = this.loadConfig();
if (!config.translate) {
throw new Error('翻译配置未设置');
}
return config.translate;
}
/**
* 检查是否启用翻译
*/
isTranslateEnabled() {
const config = this.loadConfig();
return config.translate?.check ?? false;
}
/**
* 获取角色的翻译目标语言
*/
getCharacterTranslateTarget(characterName) {
const characterConfig = this.getCharacterConfig(characterName);
return characterConfig?.translate_to || null;
}
/**
* 重新加载配置
*/
reloadConfig() {
this.config = null;
return this.loadConfig();
}
}
//# sourceMappingURL=config.js.map