115 lines
2.6 KiB
Markdown
115 lines
2.6 KiB
Markdown
# 新闻稿生成器
|
||
|
||
一个支持多种大语言模型的新闻稿生成工具，可以根据新闻事件描述自动生成不同风格的新闻稿。
|
||
|
||
## 支持的大模型
|
||
|
||
- **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 # 前端模板
|
||
``` |