279 lines
8.8 KiB
Python
279 lines
8.8 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
Script to SRT with Timing · 剧本到 SRT 时间轴生成器
|
|||
|
|
===============================================
|
|||
|
|
用剧本台词和配音时长生成正式 SRT，别手写测试时间轴。
|
|||
|
|
|
|||
|
|
功能：
|
|||
|
|
1. 读取剧本台词（JSON/MD/TXT）
|
|||
|
|
2. 读取配音文件（获取每条台词的实际时长）
|
|||
|
|
3. 自动计算时间轴（考虑角色对话间隔）
|
|||
|
|
4. 生成标准 SRT 字幕文件
|
|||
|
|
5. 支持多角色、旁白、音效标注
|
|||
|
|
|
|||
|
|
依赖：
|
|||
|
|
pip install pysrt
|
|||
|
|
|
|||
|
|
输入格式：
|
|||
|
|
# 剧本 JSON 格式（推荐）
|
|||
|
|
[
|
|||
|
|
{"id": 1, "character": "苏白", "text": "付费才能修仙？", "audio_file": "./audio/subai_001.wav"},
|
|||
|
|
{"id": 2, "character": "旁白", "text": "天道宗门口", "audio_file": "./audio/narrator_001.wav"}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# 剧本 MD 格式（兼容）
|
|||
|
|
## E1-SHOT01
|
|||
|
|
**苏白**：付费才能修仙？（audio: subai_001.wav）
|
|||
|
|
**旁白**：天道宗门口（audio: narrator_001.wav）
|
|||
|
|
|
|||
|
|
输出 SRT 格式：
|
|||
|
|
1
|
|||
|
|
00:00:01,000 --> 00:00:03,500
|
|||
|
|
付费才能修仙？
|
|||
|
|
|
|||
|
|
2
|
|||
|
|
00:00:03,800 --> 00:00:05,200
|
|||
|
|
天道宗门口
|
|||
|
|
|
|||
|
|
用法：
|
|||
|
|
# 从剧本 JSON + 配音文件生成 SRT
|
|||
|
|
python script-to-srt-with-timing.py --script script.json --output subtitles.srt
|
|||
|
|
|
|||
|
|
# 指定配音目录（自动匹配 audio_file）
|
|||
|
|
python script-to-srt-with-timing.py --script script.json --audio-dir ./audio/ --output subtitles.srt
|
|||
|
|
|
|||
|
|
# 从 MD 剧本生成
|
|||
|
|
python script-to-srt-with-timing.py --script script.md --output subtitles.srt
|
|||
|
|
|
|||
|
|
# 自定义对话间隔（秒）
|
|||
|
|
python script-to-srt-with-timing.py --script script.json --gap 0.3 --output subtitles.srt
|
|||
|
|
|
|||
|
|
路径：
|
|||
|
|
video-ai-system/engines/subtitle-pipeline/srt-tools/script-to-srt-with-timing.py
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import re
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
import pysrt
|
|||
|
|
except ImportError:
|
|||
|
|
print("[ERROR] 缺少依赖：pysrt")
|
|||
|
|
print("请先安装：pip install pysrt")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def parse_script_json(script_path: str) -> list:
|
|||
|
|
"""
|
|||
|
|
解析剧本 JSON 文件
|
|||
|
|
|
|||
|
|
:param script_path: 剧本 JSON 文件路径
|
|||
|
|
:return: 台词列表 [{"id", "character", "text", "audio_file", "start", "end"}]
|
|||
|
|
"""
|
|||
|
|
with open(script_path, "r", encoding="utf-8") as f:
|
|||
|
|
script = json.load(f)
|
|||
|
|
|
|||
|
|
print(f"[INFO] 读取剧本 JSON：{len(script)} 条台词")
|
|||
|
|
return script
|
|||
|
|
|
|||
|
|
|
|||
|
|
def parse_script_md(script_path: str) -> list:
|
|||
|
|
"""
|
|||
|
|
解析剧本 MD 文件（兼容格式）
|
|||
|
|
|
|||
|
|
:param script_path: 剧本 MD 文件路径
|
|||
|
|
:return: 台词列表
|
|||
|
|
"""
|
|||
|
|
with open(script_path, "r", encoding="utf-8") as f:
|
|||
|
|
lines = f.readlines()
|
|||
|
|
|
|||
|
|
script = []
|
|||
|
|
current_id = 0
|
|||
|
|
|
|||
|
|
for line in lines:
|
|||
|
|
# 匹配：**角色**：台词（audio: file.wav）
|
|||
|
|
match = re.match(r"\*\*(.+?)\*\*[：:]\s*(.+?)\s*\(audio:\s*(.+?)\)", line.strip())
|
|||
|
|
if match:
|
|||
|
|
character = match.group(1).strip()
|
|||
|
|
text = match.group(2).strip()
|
|||
|
|
audio_file = match.group(3).strip()
|
|||
|
|
|
|||
|
|
script.append({
|
|||
|
|
"id": current_id + 1,
|
|||
|
|
"character": character,
|
|||
|
|
"text": text,
|
|||
|
|
"audio_file": audio_file
|
|||
|
|
})
|
|||
|
|
current_id += 1
|
|||
|
|
|
|||
|
|
print(f"[INFO] 读取剧本 MD：{len(script)} 条台词")
|
|||
|
|
return script
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_audio_duration(audio_path: str) -> float:
|
|||
|
|
"""
|
|||
|
|
获取音频文件时长（秒）
|
|||
|
|
|
|||
|
|
:param audio_path: 音频文件路径
|
|||
|
|
:return: 时长（秒）
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
import ffmpeg
|
|||
|
|
probe = ffmpeg.probe(audio_path)
|
|||
|
|
duration = float(probe['format']['duration'])
|
|||
|
|
return duration
|
|||
|
|
except ImportError:
|
|||
|
|
# 如果没有 ffmpeg-python，使用 ffprobe 命令行
|
|||
|
|
try:
|
|||
|
|
import subprocess
|
|||
|
|
cmd = [
|
|||
|
|
"ffprobe", "-v", "quiet", "-print_format", "json",
|
|||
|
|
"-show_format", "-show_streams", audio_path
|
|||
|
|
]
|
|||
|
|
result = subprocess.run(cmd, capture_output=True, text=True, check=True)
|
|||
|
|
probe = json.loads(result.stdout)
|
|||
|
|
duration = float(probe['format']['duration'])
|
|||
|
|
return duration
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[WARN] 无法获取音频时长：{audio_path} ({e})")
|
|||
|
|
# 估算：中文普通话约 4-6 字/秒
|
|||
|
|
return 2.0 # 默认 2 秒
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[WARN] 无法获取音频时长：{audio_path} ({e})")
|
|||
|
|
return 2.0
|
|||
|
|
|
|||
|
|
|
|||
|
|
def calculate_timings(script: list, audio_dir: str = None, gap: float = 0.2) -> list:
|
|||
|
|
"""
|
|||
|
|
计算时间轴
|
|||
|
|
|
|||
|
|
:param script: 台词列表
|
|||
|
|
:param audio_dir: 配音文件目录（如果 audio_file 是相对路径）
|
|||
|
|
:param gap: 对话间隔（秒）
|
|||
|
|
:return: 带时间轴的台词列表
|
|||
|
|
"""
|
|||
|
|
current_time = 0.0
|
|||
|
|
|
|||
|
|
for item in script:
|
|||
|
|
# 获取配音文件时长
|
|||
|
|
audio_file = item.get("audio_file")
|
|||
|
|
|
|||
|
|
if audio_file and os.path.isfile(audio_file):
|
|||
|
|
duration = get_audio_duration(audio_file)
|
|||
|
|
elif audio_file and audio_dir:
|
|||
|
|
audio_path = os.path.join(audio_dir, audio_file)
|
|||
|
|
if os.path.isfile(audio_path):
|
|||
|
|
duration = get_audio_duration(audio_path)
|
|||
|
|
else:
|
|||
|
|
# 估算时长
|
|||
|
|
text = item.get("text", "")
|
|||
|
|
duration = len(text) / 5.0 # 假设 5 字/秒
|
|||
|
|
else:
|
|||
|
|
# 没有配音文件，估算时长
|
|||
|
|
text = item.get("text", "")
|
|||
|
|
duration = len(text) / 5.0 # 假设 5 字/秒
|
|||
|
|
|
|||
|
|
# 设置开始和结束时间
|
|||
|
|
item["start"] = current_time
|
|||
|
|
item["end"] = current_time + duration
|
|||
|
|
|
|||
|
|
# 更新当前时间（加上间隔）
|
|||
|
|
current_time = item["end"] + gap
|
|||
|
|
|
|||
|
|
print(f"[INFO] 时间轴计算完成：总时长 {current_time:.2f} 秒")
|
|||
|
|
return script
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_srt(script: list, output_path: str):
|
|||
|
|
"""
|
|||
|
|
生成 SRT 字幕文件
|
|||
|
|
|
|||
|
|
:param script: 带时间轴的台词列表
|
|||
|
|
:param output_path: 输出 SRT 文件路径
|
|||
|
|
"""
|
|||
|
|
# 创建 pysrt SubRipFile
|
|||
|
|
subs = pysrt.SubRipFile()
|
|||
|
|
|
|||
|
|
for item in script:
|
|||
|
|
# 创建字幕项
|
|||
|
|
sub = pysrt.SubRipItem()
|
|||
|
|
sub.index = item["id"]
|
|||
|
|
sub.text = item["text"]
|
|||
|
|
|
|||
|
|
# 设置时间
|
|||
|
|
start_seconds = item["start"]
|
|||
|
|
end_seconds = item["end"]
|
|||
|
|
|
|||
|
|
sub.start.hours = int(start_seconds // 3600)
|
|||
|
|
sub.start.minutes = int((start_seconds % 3600) // 60)
|
|||
|
|
sub.start.seconds = int(start_seconds % 60)
|
|||
|
|
sub.start.milliseconds = int((start_seconds % 1) * 1000)
|
|||
|
|
|
|||
|
|
sub.end.hours = int(end_seconds // 3600)
|
|||
|
|
sub.end.minutes = int((end_seconds % 3600) // 60)
|
|||
|
|
sub.end.seconds = int(end_seconds % 60)
|
|||
|
|
sub.end.milliseconds = int((end_seconds % 1) * 1000)
|
|||
|
|
|
|||
|
|
subs.append(sub)
|
|||
|
|
|
|||
|
|
# 保存 SRT 文件
|
|||
|
|
os.makedirs(os.path.dirname(os.path.abspath(output_path)), exist_ok=True)
|
|||
|
|
subs.save(output_path, encoding="utf-8")
|
|||
|
|
|
|||
|
|
print(f"[OK] SRT 字幕文件已生成：{output_path}")
|
|||
|
|
print(f"[INFO] 共 {len(script)} 条字幕")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
description="Script to SRT with Timing · 剧本到 SRT 时间轴生成器",
|
|||
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|||
|
|
epilog="""
|
|||
|
|
示例：
|
|||
|
|
python script-to-srt-with-timing.py --script script.json --output subtitles.srt
|
|||
|
|
python script-to-srt-with-timing.py --script script.md --audio-dir ./audio/ --output subtitles.srt
|
|||
|
|
python script-to-srt-with-timing.py --script script.json --gap 0.3 --output subtitles.srt
|
|||
|
|
"""
|
|||
|
|
)
|
|||
|
|
parser.add_argument("--script", required=True, help="剧本文件路径（JSON 或 MD 格式）")
|
|||
|
|
parser.add_argument("--audio-dir", help="配音文件目录（如果 audio_file 是相对路径）")
|
|||
|
|
parser.add_argument("--output", required=True, help="输出 SRT 文件路径")
|
|||
|
|
parser.add_argument("--gap", type=float, default=0.2, help="对话间隔（秒，默认 0.2）")
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
# 检查剧本文件是否存在
|
|||
|
|
if not os.path.isfile(args.script):
|
|||
|
|
print(f"[ERROR] 剧本文件不存在：{args.script}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 解析剧本
|
|||
|
|
if args.script.lower().endswith(".json"):
|
|||
|
|
script = parse_script_json(args.script)
|
|||
|
|
elif args.script.lower().endswith(".md"):
|
|||
|
|
script = parse_script_md(args.script)
|
|||
|
|
else:
|
|||
|
|
print(f"[ERROR] 不支持的剧本格式：{args.script}")
|
|||
|
|
print("支持格式：.json, .md")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
if not script:
|
|||
|
|
print("[ERROR] 剧本为空，请检查格式")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 计算时间轴
|
|||
|
|
script = calculate_timings(script, args.audio_dir, args.gap)
|
|||
|
|
|
|||
|
|
# 生成 SRT 文件
|
|||
|
|
generate_srt(script, args.output)
|
|||
|
|
|
|||
|
|
print("\n[OK] 处理完成")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|