abo/news_agent/README.md

115 lines
2.6 KiB
Markdown
Raw Normal View History

2025-07-16 15:34:54 +08:00
# 新闻稿生成器
一个支持多种大语言模型的新闻稿生成工具,可以根据新闻事件描述自动生成不同风格的新闻稿。
## 支持的大模型
- **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 # 前端模板
```