ComfyUI Prompt Studio v1.0.0
AI 图像生成工作室 - 智能提示词增强 + ComfyUI 工作流自动化 + 实时 WebSocket 更新
✨ 核心功能
🎨 AI 智能提示词增强
- 支持 Ollama 和 Gemini 双 AI 提供商
- 自动优化和扩展用户输入的提示词
- 提升图像生成质量和创意表达
⚙️ ComfyUI 工作流集成
- 自动工作流选择(普通文生图 / 人脸替换)
- 支持自定义图片尺寸配置
- 随机种子自动化,确保每次生成唯一
🔄 实时进度推送
- WebSocket 实时更新生成进度
- 实时日志流式输出
- 新图片即时展示
📚 历史记录管理
- SQLite 数据库持久化存储
- 提示词和生成图片关联管理
- 一键加载历史记录继续生成
🏗️ 工程化架构
- 模块化代码结构(src/app, src/cli, src/core)
- 前端资源拆分(CSS/JS 独立文件)
- 模板组件化(layouts/partials)
- 单例模式服务层设计
🔒 安全配置
- 环境变量隔离(.env + .env.example)
- 敏感数据自动忽略(.gitignore)
- API 密钥保护
📦 技术栈
- 后端: Flask + Flask-SocketIO
- AI 服务: Ollama / Gemini (OpenAI-compatible)
- 图像生成: ComfyUI API
- 数据存储: SQLite
- 前端: HTML + JavaScript(原生)
🚀 快速开始
# 1. 克隆仓库
git clone https://github.com/yxhpy/comfy-prompt-studio.git
cd comfy-prompt-studio
# 2. 配置环境变量
cp .env.example .env
# 编辑 .env 填入你的 API 密钥
# 3. 安装依赖
pip install -r requirements.txt
# 4. 启动应用
python -m src.cli.run📋 前置要求
- ComfyUI 服务器运行在
127.0.0.1:8188 - Ollama 或 Gemini API 配置
- Python 3.8+
📖 文档
详细文档请查看:
🎯 项目统计
- 53 个文件
- +3,911 行新增代码
- -4,851 行优化重构
- 工程化重构完成度:80%
📸 主要特性截图
- ✅ 智能提示词增强
- ✅ 实时进度展示
- ✅ 历史记录管理
- ✅ 多工作流支持
- ✅ 响应式 Web 界面
🔧 配置说明
环境变量配置
编辑 .env 文件:
# AI Provider
AI_PROVIDER=gemini # 或 ollama
# Gemini Configuration
GEMINI_API_KEY=your-api-key-here
GEMINI_MODEL=your-model-name
GEMINI_BASE_URL=https://your-api-endpoint.com/v1/
# Ollama Configuration
OLLAMA_URL=http://localhost:11434
OLLAMA_MODEL=your-model-nameComfyUI 工作流
- 普通文生图:
config/workflows/flowv_normal.json - 人脸替换:
config/workflows/flow_face.json
🐛 已知问题
- Windows 环境下需要注意文件编码(UTF-8)
- SocketIO 版本需要与 Flask-SocketIO 兼容
- 部分旧代码位置保留以确保向后兼容
📝 更新日志
v1.0.0 (2025-10-27)
工程化重构:
- 🏗️ 代码重组到 src/ 目录结构
- 📁 前端资源拆分为独立文件
- 🧩 模板模块化重构
安全增强:
- 🔒 添加 .env.example 模板
- 🛡️ 修复 .gitignore 配置
- 🗑️ 移除敏感数据文件
文档完善:
- 📚 添加安全配置章节
- 📖 更新配置和部署指南
- 🔗 标注敏感文件
🙏 致谢
感谢以下项目和社区的支持:
📄 许可证
MIT License
完整更新日志: https://github.com/yxhpy/comfy-prompt-studio/commits/v1.0.0