2.6 KiB

旅游内容创作工具示例

本目录包含使用旅游内容创作工具各个组件的示例代码。通过这些示例,您可以了解如何调用系统的各个模块,或测试整个工作流程。

测试工作流程

test_workflow.py 是一个完整的测试脚本,它展示了如何测试整个内容生成流程,包括:

  1. 完整流程测试:执行从选题生成到海报制作的全过程
  2. 分步骤测试:将流程拆分为独立的步骤,模拟在不同时间或机器上执行的场景

使用方法

确保已经正确配置了项目根目录的 poster_gen_config.json 文件。

# 测试完整工作流程(从选题生成到海报制作的全过程)
python examples/test_workflow.py --mode full

# 测试分步骤工作流程(模拟分阶段执行的场景)
python examples/test_workflow.py --mode steps

# 两种测试模式都执行(默认)
python examples/test_workflow.py

分阶段执行示例

这两个脚本演示了如何将工作流程拆分为两个独立的阶段:

  1. run_step1_topics.py: 仅执行选题生成。
  2. run_step2_content_posters.py: 加载第一阶段生成的选题结果,然后执行内容和海报生成。

使用方法

  1. 执行阶段 1 (生成选题):

    python examples/run_step1_topics.py
    

    运行结束后,脚本会打印出生成的 Run ID 和选题文件路径。请记下这个 Run ID

  2. 执行阶段 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
  • 测试脚本会自动调节某些参数(如生成数量)以加快测试速度
  • 实际使用时,您可能需要调整参数以获得更好的效果