312 lines
9.4 KiB
Python
312 lines
9.4 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
TTS Engine · Edge-TTS 配音引擎
|
|||
|
|
==================================
|
|||
|
|
为视频AI系统提供文本转语音能力，支持多角色音色配置。
|
|||
|
|
|
|||
|
|
依赖：
|
|||
|
|
pip install edge-tts
|
|||
|
|
|
|||
|
|
用法：
|
|||
|
|
# 基本用法（默认中文女声）
|
|||
|
|
python tts-engine.py --text "你好世界" --output output.mp3
|
|||
|
|
|
|||
|
|
# 指定角色（从配置文件读取音色）
|
|||
|
|
python tts-engine.py --text "未来的天下第一宗！" --character "苏白" --output su-bai.mp3
|
|||
|
|
|
|||
|
|
# 指定语音（Edge-TTS 语音名）
|
|||
|
|
python tts-engine.py --text "Hello World" --voice "en-US-JennyNeural" --output hello.mp3
|
|||
|
|
|
|||
|
|
# 调整语速/音调/音量
|
|||
|
|
python tts-engine.py --text "你好" --rate "+20%" --pitch "+5Hz" --volume "+10%" --output output.mp3
|
|||
|
|
|
|||
|
|
# 批量生成（从SRT字幕文件）
|
|||
|
|
python tts-engine.py --srt input.srt --output-dir ./audio/ --character "苏白"
|
|||
|
|
|
|||
|
|
# 作为模块导入
|
|||
|
|
from tts_engine import generate_speech
|
|||
|
|
generate_speech("你好世界", "output.mp3", voice="zh-CN-XiaoxiaoNeural")
|
|||
|
|
|
|||
|
|
角色音色配置：
|
|||
|
|
video-ai-system/config/voices.json
|
|||
|
|
|
|||
|
|
路径：
|
|||
|
|
video-ai-system/engines/tts-engine.py
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import asyncio
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
import edge_tts
|
|||
|
|
except ImportError:
|
|||
|
|
print("[ERROR] 缺少依赖：edge-tts")
|
|||
|
|
print("请先安装：pip install edge-tts")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 默认角色音色配置
|
|||
|
|
DEFAULT_VOICES = {
|
|||
|
|
"苏白": {
|
|||
|
|
"voice": "zh-CN-XiaoxiaoNeural", # 阳光少年音
|
|||
|
|
"rate": "+5%",
|
|||
|
|
"pitch": "+0Hz"
|
|||
|
|
},
|
|||
|
|
"诸葛风": {
|
|||
|
|
"voice": "zh-CN-YunxiNeural", # 沉稳男声
|
|||
|
|
"rate": "+0%",
|
|||
|
|
"pitch": "-5Hz"
|
|||
|
|
},
|
|||
|
|
"萧灵汐": {
|
|||
|
|
"voice": "zh-CN-XiaoyiNeural", # 清冷女声
|
|||
|
|
"rate": "+0%",
|
|||
|
|
"pitch": "+0Hz"
|
|||
|
|
},
|
|||
|
|
"王执事": {
|
|||
|
|
"voice": "zh-CN-YunyangNeural", # 中年男声
|
|||
|
|
"rate": "+0%",
|
|||
|
|
"pitch": "-10Hz"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def load_voice_config(config_path: str = None) -> dict:
|
|||
|
|
"""加载角色音色配置文件"""
|
|||
|
|
if config_path and os.path.isfile(config_path):
|
|||
|
|
try:
|
|||
|
|
with open(config_path, "r", encoding="utf-8") as f:
|
|||
|
|
return json.load(f)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[WARN] 无法读取配置文件 {config_path}：{e}")
|
|||
|
|
print("[INFO] 使用默认音色配置")
|
|||
|
|
return DEFAULT_VOICES
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def generate_speech_async(
|
|||
|
|
text: str,
|
|||
|
|
output_path: str,
|
|||
|
|
voice: str = "zh-CN-XiaoxiaoNeural",
|
|||
|
|
rate: str = "+0%",
|
|||
|
|
pitch: str = "+0Hz",
|
|||
|
|
volume: str = "+0%"
|
|||
|
|
) -> bool:
|
|||
|
|
"""
|
|||
|
|
异步生成语音（Edge-TTS）
|
|||
|
|
|
|||
|
|
:param text: 要合成的文本
|
|||
|
|
:param output_path: 输出音频文件路径
|
|||
|
|
:param voice: Edge-TTS 语音名
|
|||
|
|
:param rate: 语速（如 "+20%"、"-10%"）
|
|||
|
|
:param pitch: 音调（如 "+5Hz"、"-10Hz"）
|
|||
|
|
:param volume: 音量（如 "+10%"、"-5%"）
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
# 确保输出目录存在
|
|||
|
|
os.makedirs(os.path.dirname(os.path.abspath(output_path)), exist_ok=True)
|
|||
|
|
|
|||
|
|
communicate = edge_tts.Communicate(
|
|||
|
|
text, voice,
|
|||
|
|
rate=rate, pitch=pitch, volume=volume
|
|||
|
|
)
|
|||
|
|
await communicate.save(output_path)
|
|||
|
|
|
|||
|
|
# 验证输出文件
|
|||
|
|
if os.path.isfile(output_path):
|
|||
|
|
file_size = os.path.getsize(output_path)
|
|||
|
|
print(f"[OK] 生成语音：{output_path} ({file_size // 1024} KB)")
|
|||
|
|
print(f" 语音：{voice} | 语速：{rate} | 音调：{pitch}")
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
print(f"[ERROR] 输出文件未生成：{output_path}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 生成语音失败：{e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_speech(
|
|||
|
|
text: str,
|
|||
|
|
output_path: str,
|
|||
|
|
voice: str = "zh-CN-XiaoxiaoNeural",
|
|||
|
|
rate: str = "+0%",
|
|||
|
|
pitch: str = "+0Hz",
|
|||
|
|
volume: str = "+0%"
|
|||
|
|
) -> bool:
|
|||
|
|
"""
|
|||
|
|
同步包装器（供外部调用）
|
|||
|
|
|
|||
|
|
:param text: 要合成的文本
|
|||
|
|
:param output_path: 输出音频文件路径
|
|||
|
|
:param voice: Edge-TTS 语音名
|
|||
|
|
:param rate: 语速
|
|||
|
|
:param pitch: 音调
|
|||
|
|
:param volume: 音量
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
return asyncio.run(generate_speech_async(text, output_path, voice, rate, pitch, volume))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_by_character(
|
|||
|
|
text: str,
|
|||
|
|
output_path: str,
|
|||
|
|
character: str,
|
|||
|
|
config: dict = None
|
|||
|
|
) -> bool:
|
|||
|
|
"""
|
|||
|
|
按角色名生成语音（自动读取音色配置）
|
|||
|
|
|
|||
|
|
:param text: 要合成的文本
|
|||
|
|
:param output_path: 输出音频文件路径
|
|||
|
|
:param character: 角色名（如 "苏白"）
|
|||
|
|
:param config: 音色配置字典（可选，默认加载 DEFAULT_VOICES）
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
if config is None:
|
|||
|
|
config = load_voice_config()
|
|||
|
|
|
|||
|
|
if character not in config:
|
|||
|
|
print(f"[WARN] 角色 '{character}' 未配置音色，使用默认音色")
|
|||
|
|
return generate_speech(text, output_path)
|
|||
|
|
|
|||
|
|
voice_config = config[character]
|
|||
|
|
return generate_speech(
|
|||
|
|
text, output_path,
|
|||
|
|
voice=voice_config.get("voice", "zh-CN-XiaoxiaoNeural"),
|
|||
|
|
rate=voice_config.get("rate", "+0%"),
|
|||
|
|
pitch=voice_config.get("pitch", "+0Hz")
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def process_srt(srt_path: str, output_dir: str, character: str = None, config: dict = None):
|
|||
|
|
"""
|
|||
|
|
从SRT字幕文件批量生成语音
|
|||
|
|
|
|||
|
|
:param srt_path: SRT 文件路径
|
|||
|
|
:param output_dir: 输出目录
|
|||
|
|
:param character: 角色名（所有台词用同一音色）
|
|||
|
|
:param config: 音色配置字典
|
|||
|
|
"""
|
|||
|
|
if not os.path.isfile(srt_path):
|
|||
|
|
print(f"[ERROR] SRT 文件不存在：{srt_path}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
os.makedirs(output_dir, exist_ok=True)
|
|||
|
|
|
|||
|
|
# 简单SRT解析（按空行分割）
|
|||
|
|
with open(srt_path, "r", encoding="utf-8") as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
blocks = content.strip().split("\n\n")
|
|||
|
|
print(f"[INFO] 找到 {len(blocks)} 条字幕，开始生成语音...")
|
|||
|
|
|
|||
|
|
success = 0
|
|||
|
|
for block in blocks:
|
|||
|
|
lines = block.strip().split("\n")
|
|||
|
|
if len(lines) < 3:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
idx = lines[0].strip()
|
|||
|
|
# timeine = lines[1].strip() # 暂不使用时序
|
|||
|
|
text = " ".join(lines[2:]).strip()
|
|||
|
|
|
|||
|
|
if not text:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
output_path = os.path.join(output_dir, f"{idx.zfill(4)}.mp3")
|
|||
|
|
|
|||
|
|
if character:
|
|||
|
|
ok = generate_by_character(text, output_path, character, config)
|
|||
|
|
else:
|
|||
|
|
ok = generate_speech(text, output_path)
|
|||
|
|
|
|||
|
|
if ok:
|
|||
|
|
success += 1
|
|||
|
|
|
|||
|
|
print(f"\n[INFO] SRT批量生成完成：{success}/{len(blocks)} 成功")
|
|||
|
|
return success == len(blocks)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def list_voices():
|
|||
|
|
"""列出所有可用的Edge-TTS语音"""
|
|||
|
|
print("[INFO] 正在获取可用语音列表...\n")
|
|||
|
|
asyncio.run(_list_voices_async())
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def _list_voices_async():
|
|||
|
|
"""异步列出语音"""
|
|||
|
|
voices = await edge_tts.list_voices()
|
|||
|
|
print("中文语音：")
|
|||
|
|
for v in voices:
|
|||
|
|
if v["Locale"].startswith("zh-"):
|
|||
|
|
print(f" {v['ShortName']:30s} - {v['FriendlyName']}")
|
|||
|
|
print(f"\n共 {len([v for v in voices if v['Locale'].startswith('zh-')])} 个中文语音")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
description="TTS Engine · Edge-TTS 配音引擎",
|
|||
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|||
|
|
epilog="""
|
|||
|
|
示例：
|
|||
|
|
python tts-engine.py --text "你好世界" --output output.mp3
|
|||
|
|
python tts-engine.py --text "未来的天下第一宗！" --character "苏白" --output su-bai.mp3
|
|||
|
|
python tts-engine.py --text "Hello" --voice "en-US-JennyNeural" --output hello.mp3
|
|||
|
|
python tts-engine.py --srt input.srt --output-dir ./audio/ --character "苏白"
|
|||
|
|
python tts-engine.py --list-voices
|
|||
|
|
"""
|
|||
|
|
)
|
|||
|
|
parser.add_argument("--text", help="要合成的文本")
|
|||
|
|
parser.add_argument("--output", help="输出音频文件路径")
|
|||
|
|
parser.add_argument("--voice", default="zh-CN-XiaoxiaoNeural", help="Edge-TTS 语音名")
|
|||
|
|
parser.add_argument("--character", help="角色名（从配置文件读取音色）")
|
|||
|
|
parser.add_argument("--rate", default="+0%", help="语速（如 +20%%、-10%%）")
|
|||
|
|
parser.add_argument("--pitch", default="+0Hz", help="音调（如 +5Hz、-10Hz）")
|
|||
|
|
parser.add_argument("--volume", default="+0%", help="音量（如 +10%%、-5%%）")
|
|||
|
|
parser.add_argument("--srt", help="从SRT字幕文件批量生成")
|
|||
|
|
parser.add_argument("--output-dir", help="批量生成时的输出目录")
|
|||
|
|
parser.add_argument("--config", help="角色音色配置文件路径")
|
|||
|
|
parser.add_argument("--list-voices", action="store_true", help="列出所有可用语音")
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
if args.list_voices:
|
|||
|
|
list_voices()
|
|||
|
|
sys.exit(0)
|
|||
|
|
|
|||
|
|
if args.srt:
|
|||
|
|
# 批量模式
|
|||
|
|
if not args.output_dir:
|
|||
|
|
print("[ERROR] 批量模式需要指定 --output-dir")
|
|||
|
|
sys.exit(1)
|
|||
|
|
config = load_voice_config(args.config)
|
|||
|
|
ok = process_srt(args.srt, args.output_dir, args.character, config)
|
|||
|
|
sys.exit(0 if ok else 1)
|
|||
|
|
|
|||
|
|
if not args.text or not args.output:
|
|||
|
|
print("[ERROR] 需要指定 --text 和 --output")
|
|||
|
|
parser.print_help()
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 单文件模式
|
|||
|
|
if args.character:
|
|||
|
|
config = load_voice_config(args.config)
|
|||
|
|
ok = generate_by_character(args.text, args.output, args.character, config)
|
|||
|
|
else:
|
|||
|
|
ok = generate_speech(
|
|||
|
|
args.text, args.output,
|
|||
|
|
voice=args.voice, rate=args.rate,
|
|||
|
|
pitch=args.pitch, volume=args.volume
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
sys.exit(0 if ok else 1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|