旅游内容创作工具 (Travel Content Creator)

这是一个基于AI的旅游内容自动生成工具可以根据景点信息自动生成高质量的旅游推文和宣传海报。

功能特点

  • 自动选题生成:根据提供的景点信息自动生成吸引人的旅游选题
  • 内容创作:基于选题自动生成文字内容,包括标题和正文
  • 海报制作:结合景点图片和生成的文字内容,自动创建精美的宣传海报
  • 批量处理:支持一次性生成多个选题和多个变体内容

安装

环境要求

  • Python 3.6+
  • 依赖库安装:
# 安装依赖库
pip install numpy pandas opencv-python pillow

项目依赖项

  • OpenCV (cv2):图像处理
  • NumPy数据操作
  • Pandas数据处理
  • PIL (Pillow):图像处理和绘制

目录结构

  • core/: 核心功能模块
    • ai_agent.py: AI代理接口
    • topic_parser.py: 选题解析器
    • contentGen.py: 内容生成器
    • posterGen.py: 海报生成器
    • simple_collage.py: 图片拼贴工具
  • utils/: 工具函数
    • resource_loader.py: 资源加载器
    • tweet_generator.py: 推文生成工具
  • genPrompts/: 提示词模板
    • systemPrompt.txt: 系统提示词
    • Style/: 风格提示词
    • Refer/: 参考提示词
    • Demand/: 需求提示词
  • SelectPrompt/: 选题提示词
  • resource/: 景点资源信息
  • examples/: 使用示例
  • result/: 输出结果保存目录

使用方法

  1. 准备景点资源信息,放入resource/Object/目录中
  2. 准备景点图片资源
  3. 配置poster_gen_config.json文件(或直接修改main.py中的配置)
  4. 运行主程序:
python main.py

配置说明

主要配置项说明:

  • date: 日期标记
  • num: 生成选题数量
  • model: 使用的大语言模型
  • api_url: API调用地址
  • api_key: API密钥
  • resource_dir: 景点资源信息路径
  • prompts_dir: 提示词目录
  • output_dir: 输出结果保存路径
  • variants: 每个选题生成的变体数量
  • topic_temperature: 选题生成的创意度参数
  • content_temperature: 内容生成的创意度参数

项目提供了一个示例配置文件 example_config.json,可以复制并根据需要修改:

# 复制示例配置
cp example_config.json poster_gen_config.json

# 根据自己的需求修改配置
nano poster_gen_config.json

开始使用

1. 准备景点信息文件

resource/Object/目录中创建景点信息文件,例如景点信息-泰宁古城.txt

景点名称:泰宁古城
位置:福建省三明市泰宁县
简介:泰宁古城始建于宋代,是福建省保存较为完好的古代县城之一,具有悠久的历史和丰富的文化遗产。
特色:古城墙、古街巷、古民居、古牌坊等历史建筑保存完好。

2. 准备图片资源

将景点的高质量图片保存在指定图片目录中。

3. 运行生成流程

执行主程序,系统将自动生成选题、内容和海报:

python main.py

生成的结果将保存在result/目录中。

示例

可以查看examples目录中的示例代码,了解如何使用本工具生成海报。

注意事项

  • 确保已安装所有依赖库
  • 图片目录结构需要按照配置文件中的规范组织
  • AI生成内容质量取决于提供的景点信息质量和提示词设计
Description
No description provided
Readme 145 MiB
Languages
Python 98%
Shell 2%