abo/news_agent/README.md

新闻稿生成器

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

支持的大模型

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

功能特性

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

安装依赖

pip install -r requirements.txt

配置管理

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

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

python setup_config.py

该脚本提供以下功能：

• 交互式设置各模型的API密钥
• 显示当前配置状态
• 导出环境变量文件

方法二：环境变量配置

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

# 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密钥：

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

运行应用

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     # 前端模板