旅游内容创作工具示例
本目录包含使用旅游内容创作工具各个组件的示例代码。通过这些示例,您可以了解如何调用系统的各个模块,或测试整个工作流程。
测试工作流程
test_workflow.py 是一个完整的测试脚本,它展示了如何测试整个内容生成流程,包括:
- 完整流程测试:执行从选题生成到海报制作的全过程
- 分步骤测试:将流程拆分为独立的步骤,模拟在不同时间或机器上执行的场景
使用方法
确保已经正确配置了项目根目录的 poster_gen_config.json 文件。
# 测试完整工作流程(从选题生成到海报制作的全过程)
python examples/test_workflow.py --mode full
# 测试分步骤工作流程(模拟分阶段执行的场景)
python examples/test_workflow.py --mode steps
# 两种测试模式都执行(默认)
python examples/test_workflow.py
分阶段执行示例
这两个脚本演示了如何将工作流程拆分为两个独立的阶段:
run_step1_topics.py: 仅执行选题生成。run_step2_content_posters.py: 加载第一阶段生成的选题结果,然后执行内容和海报生成。
使用方法
-
执行阶段 1 (生成选题):
python examples/run_step1_topics.py运行结束后,脚本会打印出生成的
Run ID和选题文件路径。请记下这个Run ID。 -
执行阶段 2 (生成内容和海报): 将
<your_run_id>替换为你在阶段 1 记下的实际 Run ID。python examples/run_step2_content_posters.py <your_run_id>例如:
python examples/run_step2_content_posters.py 20240515_110000
海报生成示例
generate_poster.py 是一个单独的海报生成示例,它展示了如何使用系统的海报生成模块(不依赖于选题和内容生成)。
使用方法
# 使用默认参数生成海报
python examples/generate_poster.py
# 指定输入图片和输出路径
python examples/generate_poster.py --input_image /path/to/image.jpg --output_path /path/to/output.jpg
其他示例
未来将添加更多示例,演示如何独立使用系统的各个组件,例如:
- 仅使用选题生成功能
- 仅使用内容生成功能
- 自定义AI模型和提示词
注意事项
- 这些示例依赖于主项目中的配置和资源,确保已正确设置
poster_gen_config.json - 测试脚本会自动调节某些参数(如生成数量)以加快测试速度
- 实际使用时,您可能需要调整参数以获得更好的效果