mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-12 11:22:51 +00:00
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
"""
|
|
合并生成最终脚本
|
|
"""
|
|
import os
|
|
import json
|
|
from typing import List, Dict, Tuple
|
|
|
|
|
|
def merge_script(
|
|
plot_points: List[Dict],
|
|
output_path: str
|
|
):
|
|
"""合并生成最终脚本
|
|
|
|
Args:
|
|
plot_points: 校对后的剧情点
|
|
output_path: 输出文件路径，如果提供则保存到文件
|
|
|
|
Returns:
|
|
str: 最终合并的脚本
|
|
"""
|
|
def parse_timestamp(ts: str) -> Tuple[float, float]:
|
|
"""解析时间戳，返回开始和结束时间（秒）"""
|
|
start, end = ts.split('-')
|
|
|
|
def parse_time(time_str: str) -> float:
|
|
time_str = time_str.strip()
|
|
if ',' in time_str:
|
|
time_parts, ms_parts = time_str.split(',')
|
|
ms = float(ms_parts) / 1000
|
|
else:
|
|
time_parts = time_str
|
|
ms = 0
|
|
|
|
hours, minutes, seconds = map(int, time_parts.split(':'))
|
|
return hours * 3600 + minutes * 60 + seconds + ms
|
|
|
|
return parse_time(start), parse_time(end)
|
|
|
|
def format_timestamp(seconds: float) -> str:
|
|
"""将秒数转换为时间戳格式 HH:MM:SS"""
|
|
hours = int(seconds // 3600)
|
|
minutes = int((seconds % 3600) // 60)
|
|
secs = int(seconds % 60)
|
|
return f"{hours:02d}:{minutes:02d}:{secs:02d}"
|
|
|
|
# 创建包含所有信息的临时列表
|
|
final_script = []
|
|
|
|
# 处理原生画面条目
|
|
number = 1
|
|
for plot_point in plot_points:
|
|
start, end = parse_timestamp(plot_point["timestamp"])
|
|
script_item = {
|
|
"_id": number,
|
|
"timestamp": plot_point["timestamp"],
|
|
"picture": plot_point["picture"],
|
|
"narration": f"播放原生_{os.urandom(4).hex()}",
|
|
"OST": 1, # OST=0 仅保留解说 OST=2 保留解说和原声
|
|
}
|
|
final_script.append(script_item)
|
|
number += 1
|
|
|
|
# 保存结果
|
|
with open(output_path, 'w', encoding='utf-8') as f:
|
|
json.dump(final_script, f, ensure_ascii=False, indent=4)
|
|
|
|
print(f"脚本生成完成：{output_path}")
|
|
return final_script
|