Files
ai-short-video-generator-20…/core/script_gen.py
T

43 lines
1.3 KiB
Python

# -*- coding: utf-8 -*-
"""脚本生成模块
默认使用模板方式生成脚本。
如需接入大模型(OpenAI、通义千问等),只需替换 generate_script 的实现。
"""
def generate_script(topic: str, scenes: int = 5) -> list:
"""根据主题生成分场景脚本。
Args:
topic: 视频主题
scenes: 场景数量
Returns:
list[dict]: 每个元素形如 {"index": 1, "text": "..."}
"""
templates = [
"大家好,今天我们来聊一聊{topic}。",
"首先,{topic}正在改变我们的生活方式。",
"其次,{topic}背后蕴含着巨大的机会与挑战。",
"再次,掌握{topic}的关键在于持续学习与实践。",
"最后,让我们一起拥抱{topic}带来的未来。",
"关于{topic},还有很多值得探索的方向。",
"相信不久的将来,{topic}会走进千家万户。",
"感谢观看,我们下期再见!",
]
result = []
for i in range(scenes):
tpl = templates[i % len(templates)]
result.append({
"index": i + 1,
"text": tpl.format(topic=topic),
})
return result
if __name__ == "__main__":
for s in generate_script("人工智能", 3):
print(s)