# 新闻稿生成器

一个支持多种大语言模型的新闻稿生成工具，可以根据新闻事件描述自动生成不同风格的新闻稿。

## 支持的大模型

- **OpenAI GPT-4**: 使用OpenAI的GPT-4模型
- **火山引擎豆包**: 使用火山引擎的豆包大模型
- **阿里通义千问**: 使用阿里云的通义千问模型

## 功能特性

- 支持多种文案风格：正式、简洁、感性、科技
- 支持上传新闻照片
- 自动生成Word文档格式的新闻稿
- 现代化的Web界面

## 安装依赖

```bash
pip install -r requirements.txt
```

## 配置管理

### 方法一：使用配置管理脚本（推荐）

运行配置管理脚本来交互式设置API密钥：

```bash
python setup_config.py
```

该脚本提供以下功能：
- 交互式设置各模型的API密钥
- 显示当前配置状态
- 导出环境变量文件

### 方法二：环境变量配置

在使用前，请设置相应的API密钥环境变量：

```bash
# OpenAI API密钥
export OPENAI_API_KEY="your-openai-api-key"

# 火山引擎API密钥
export VOLCENGINE_API_KEY="your-volcengine-api-key"

# 阿里云DashScope API密钥
export DASHSCOPE_API_KEY="your-dashscope-api-key"

# Flask密钥（可选）
export FLASK_SECRET_KEY="your-secret-key"
```

### 方法三：直接修改配置文件

编辑 `config.py` 文件中的 `MODELS` 配置，直接填入API密钥：

```python
MODELS = {
    'openai': {
        'name': 'OpenAI GPT-4',
        'api_key': 'your-openai-api-key-here',
        'model': 'gpt-4',
        # ... 其他配置
    },
    # ... 其他模型
}
```

## 运行应用

```bash
python app.py
```

然后在浏览器中访问 `http://localhost:5000`

## 使用说明

1. 选择要使用的大模型
2. 输入新闻事件的简要描述
3. 选择文案风格
4. 可选择上传新闻照片
5. 点击"生成新闻稿"按钮
6. 系统会自动下载生成的Word文档

## 注意事项

- 确保已正确设置相应模型的API密钥
- 不同模型的响应速度和效果可能有所不同
- 建议在使用前测试API密钥的有效性

## 技术栈

- **后端**: Flask
- **大模型**: OpenAI GPT-4, 火山引擎豆包, 阿里通义千问
- **文档处理**: python-docx
- **图像处理**: Pillow
- **配置管理**: 自定义配置类

## 项目结构

```
news_agent/
├── app.py              # 主应用文件
├── config.py           # 配置文件
├── setup_config.py     # 配置管理脚本
├── requirements.txt    # 依赖包列表
├── README.md          # 项目说明
└── templates/
    └── index.html     # 前端模板
```