TravelContentCreator/examples

#!/usr/bin/env python

-- coding: utf-8 --

旅游内容创作工具示例

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

测试工作流程

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
   

单组件测试脚本

以下脚本用于独立测试系统的各个组件，有助于开发者了解每个组件的功能和使用方法：

1. 图像处理测试 - test_image_processing.py

这个脚本专注于测试图像处理功能，无需连接AI服务，可独立测试拼贴图和海报生成功能。

# 基本用法（同时测试拼贴和海报功能）
python examples/test_image_processing.py

# 仅测试拼贴图功能
python examples/test_image_processing.py --collage_only

# 仅测试海报生成功能
python examples/test_image_processing.py --poster_only

# 指定输出目录和配置文件
python examples/test_image_processing.py --output my_test_output --config my_config.json

特点：

• 无需连接AI接口，可在离线环境下测试
• 自动创建测试用的彩色图像作为素材
• 测试多种拼贴风格（网格、非对称、胶片条、重叠、马赛克）
• 使用拼贴图作为海报底图，展示完整的图像处理流程
• 详细的日志输出，便于分析问题

2. 图片拼贴测试 - test_simple_collage.py

这个脚本演示如何使用图片拼贴功能，可以测试不同风格的拼贴效果。

# 基本用法（自动创建测试图片）
python examples/test_simple_collage.py

# 使用自定义图片目录
python examples/test_simple_collage.py --input_dir /path/to/images

# 指定输出目录和生成数量
python examples/test_simple_collage.py --output_dir my_collages --count 5

# 设置自定义尺寸
python examples/test_simple_collage.py --width 1080 --height 1440

特点：

• 支持测试所有拼贴风格（网格、非对称、胶片条、重叠、马赛克）
• 如果未提供输入图片，会自动生成测试用的彩色图片
• 可以指定随机种子，用于获得可重现的结果

3. 海报生成测试 - test_poster_generator.py

这个脚本演示如何使用海报生成功能，可以测试不同文案、配置的海报效果。

# 基本用法（使用随机文本和自动创建底图）
python examples/test_poster_generator.py

# 使用自定义底图和输出目录
python examples/test_poster_generator.py --image /path/to/image.jpg --output_dir my_posters

# 使用自定义素材目录和配置文件
python examples/test_poster_generator.py --base_dir /path/to/assets --config poster_config.json

特点：

• 支持直接使用现有海报配置文件（由ContentGenerator生成的JSON）
• 自动加载系统配置中的素材路径
• 如果未提供底图，会自动创建渐变测试底图
• 自带多种文案示例，可以测试不同文字效果

4. 内容生成测试 - test_content_generator.py

这个脚本演示如何使用内容生成功能，生成海报文本配置数据。

# 基本用法（交互式输入）
python examples/test_content_generator.py

# 按提示进行操作，可以指定描述文件和生成数量

特点：

• 使用交互式命令行提示用户输入参数
• 自动尝试从系统配置中查找描述文件
• 如果未找到描述文件，会创建临时测试文件
• 生成完成后会将配置保存为JSON文件

5. 流式输出测试 - test_stream.py

这个脚本演示如何使用AI_Agent的流式输出功能，逐块接收生成内容。

# 运行流式输出测试
python examples/test_stream.py

特点：

• 展示如何配置和使用AI_Agent的流式输出接口
• 演示实时输出文本块的处理方式
• 说明如何收集完整响应和估计token数

流式处理演示 (test_stream.py)

test_stream.py 是一个演示脚本，展示了AI_Agent类中新增的三种不同流式输出处理方法：

1. 同步流式响应 (generate_text_stream) - 使用流式API但返回完整响应
2. 回调式流式响应 (generate_text_stream_with_callback) - 通过回调函数处理每个文本块
3. 异步流式响应 (async_generate_text_stream) - 使用异步生成器返回文本流

运行演示

cd TravelContentCreator
python examples/test_stream.py

演示脚本会依次执行这三种方法，并展示它们的输出和性能差异。

更多信息

关于流式处理功能的详细文档，请参阅：

docs/streaming.md

其他示例

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

注意事项

• 这些示例依赖于主项目中的配置和资源，确保已正确设置 poster_gen_config.json
• 测试脚本会自动调节某些参数（如生成数量）以加快测试速度
• 实际使用时，您可能需要调整参数以获得更好的效果
• 部分测试脚本需要连接AI模型API，确保您的API配置正确

完整工作流测试脚本

test_workflow.py文件展示了一个完整的内容生成过程，从主题生成到海报创作。

运行完整工作流测试:

python examples/test_workflow.py

该脚本将执行以下步骤:

1. 生成旅游目的地主题
2. 根据主题生成内容
3. 生成配图海报
4. 保存结果

分步测试:

如果您想分阶段测试系统，可以分别执行以下脚本:

第一阶段: 主题生成

python examples/run_step1_topics.py

此命令将生成旅游目的地主题并保存在outputs/topics目录中。

第二阶段: 内容和海报生成

python examples/run_step2_content_posters.py

此命令将读取outputs/topics目录中的主题，生成内容并创建海报，结果保存在outputs/content和outputs/posters目录中。

仅生成海报示例

如果只想测试海报生成功能，可以使用以下示例:

python examples/generate_poster.py

该脚本默认将使用示例内容生成海报并保存在outputs/posters目录。

您也可以指定输入内容和输出路径:

python examples/generate_poster.py --input "您的旅游内容" --output "您的输出路径.jpg"

流式处理示例

基本流式输出测试

python examples/test_stream.py

该脚本展示了如何使用三种不同的流式处理方法:

1. 同步流式输出 (generate_text_stream)
2. 基于回调的流式输出 (generate_text_stream_with_callback)
3. 异步流式输出 (async_generate_text_stream)

超时处理测试

python examples/test_stream_with_timeout_handling.py

该脚本演示了如何处理流式生成中的超时情况，包括全局请求超时和流块超时的配置。

并发流式处理

python examples/concurrent_stream_processing.py

该脚本展示了如何使用异步流式处理同时处理多个不同的AI生成任务，包括:

1. 并发处理多个不同提示的任务
2. 顺序与并发处理的性能对比
3. 任务状态监控和异常处理

该示例特别适用于需要同时生成多个旅游内容片段的场景，如同时生成多个目的地的介绍、美食推荐和旅行建议。

未来示例

我们将继续添加更多示例，展示如何独立使用系统的各个组件。

重要说明

• 这些示例依赖于项目主配置和资源，请确保已正确配置
• 您可能需要调整参数以获得最佳结果
• 如有任何问题，请参阅主文档或提交issue