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

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

## 功能特点

- **自动选题生成**：根据提供的景点信息自动生成吸引人的旅游选题
- **内容创作**：基于选题自动生成文字内容，包括标题和正文
- **海报制作**：结合景点图片和生成的文字内容，自动创建精美的宣传海报
- **批量处理**：支持一次性生成多个选题和多个变体内容
- **模块化设计**：各功能模块独立，可单独使用或组合使用

## 安装

### 环境要求
- Python 3.6+
- 依赖库安装：

```bash
# 安装依赖库
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. 复制`example_config.json`为`poster_gen_config.json`并进行配置
4. 运行主程序：

```bash
python main.py
```

## 配置说明

必须配置项：

- `num`: 生成选题数量
- `model`: 使用的大语言模型
- `api_url`: API调用地址
- `api_key`: API密钥
- `resource_dir`: 景点资源信息路径
- `prompts_dir`: 提示词目录
- `output_dir`: 输出结果保存路径
- `variants`: 每个选题生成的变体数量
- `image_base_dir`: 图片资源根目录
- `topic_system_prompt`: 选题生成系统提示词路径
- `topic_user_prompt`: 选题生成用户提示词路径
- `content_system_prompt`: 内容生成系统提示词路径

可选配置项：

- `date`: 日期标记（默认为当前日期）
- `topic_temperature`: 选题生成的创意度参数（默认0.2）
- `content_temperature`: 内容生成的创意度参数（默认0.3）
- `camera_image_subdir`: 相机图片子目录名（默认"相机"）
- `modify_image_subdir`: 修改图片子目录名（默认"modify"）
- `poster_target_size`: 海报尺寸 [宽, 高]（默认[900, 1200]）

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

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

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

配置文件示例：
```json
{
    "date": "5月15日",
    "num": 5,
    "model": "qwen",
    "api_url": "http://your-api-endpoint/v1/completions",
    "api_key": "your-api-key",
    "topic_system_prompt": "./SelectPrompt/systemPrompt.txt",
    "topic_user_prompt": "./SelectPrompt/userPrompt.txt",
    "content_system_prompt": "./genPrompts/systemPrompt.txt",
    "resource_dir": [
        {
            "type": "Object",
            "num": 3,
            "file_path": [
                "./resource/Object/景点信息-泰宁古城.txt",
                "./resource/Object/景点信息-尚书第.txt",
                "./resource/Object/景点信息-明清园.txt"
            ]
        }
    ],
    "prompts_dir": "./genPrompts",
    "output_dir": "./result",
    "image_base_dir": "/path/to/your/image/directory",
    "camera_image_subdir": "相机",
    "modify_image_subdir": "modify",
    "variants": 3,
    "topic_temperature": 0.2,
    "content_temperature": 0.3,
    "poster_target_size": [900, 1200]
}
```

## 项目流程

整个内容生成流程分为以下几个独立步骤：

1. **选题生成**：使用AI根据景点信息生成多个选题方向
2. **内容生成**：根据选题生成具体的旅游推文内容
3. **海报生成**：使用景点图片和生成的文本内容创建海报

每个步骤都有独立的函数处理，也可以作为整体流程一次执行。

## 开始使用

### 1. 准备景点信息文件

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

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

### 2. 准备图片资源

将景点的高质量图片保存在配置的图片目录中，符合以下结构：

```
image_base_dir/
├── 相机/  (camera_image_subdir)
│   ├── 泰宁古城/
│   │   ├── 图片1.jpg
│   │   ├── 图片2.jpg
│   │   └── description.txt  (可选描述文件)
│   └── 其他景点/
└── modify/  (modify_image_subdir)
    ├── 泰宁古城/
    │   ├── 图片1.jpg
    │   ├── 图片2.jpg
    │   └── ...
    └── 其他景点/
```

### 3. 配置文件

复制并修改`example_config.json`为`poster_gen_config.json`，确保:
- 设置正确的API信息
- 配置景点资源文件路径
- 设置图片资源根目录

### 4. 运行生成流程

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

```bash
python main.py
```

生成的结果将保存在配置的`output_dir`目录中。

## 示例

查看`examples`目录中的示例代码，了解如何使用本工具的各个组件。

## 注意事项

- 确保已安装所有依赖库
- 图片目录结构需要按照配置文件中的规范组织
- AI生成内容质量取决于提供的景点信息质量和提示词设计
- 检查您的API密钥和URL配置是否正确