AI艺术生成服务器

一个增强的模型上下文协议(MCP)服务器,为Stable Diffusion提供全面的AI艺术生成能力,通过简单直观的界面实现文本到图像、图像到图像的转换以及模型管理等功能。

  • by hkhkkh
  • 本地部署
  • 更新于
在 GitHub 查看 hkhkkh/stable-diffusion-mcp-server

使用说明

🎨 稳定的扩散MCP服务器

用于稳定扩散的增强型模型上下文协议(MCP)服务器,通过简单直观的界面提供全面的AI艺术生成功能。

✨ 特性

🎯 核心AI艺术一代

  • 文本到图像(txt2img):根据文本描述生成图像
  • 图像到图像(img2img):使用AI转换现有图像
  • 即时分析:图像中的逆向工程提示(询问)

🔧 高性能

  • 模型管理:动态列出、切换和管理SD模型
  • LoRA支持:应用LoRA模型以增强艺术效果
  • 智能推荐:人工智能驱动的模型和参数建议
  • 自动启动:SD WebUI的自动检测和启动
  • 多语言支持:适用于中英文提示

🛠️ MCP工具可用

工具名称 描述 关键参数
generate_image 文本到图像生成 提示、型号、lora、尺寸
img2img 图像到图像的转换 图像路径、提示、降噪_强度
interrogate_image 从图像中提取提示 image_path,model
list_models 列出可用的SD型号 -
list_loras 列出可用的LoRA型号 -
switch_model 更改当前SD型号 model_name
check_sd_status 检查SD WebUI状态 -
start_sd_webui 启动SD WebUI -

🚀 快速开始

先决条件

  • Python 3.8+
  • 稳定扩散WebUI(AUTOMATIC1111或兼容)
  • SD WebUI正在运行 http://127.0.0.1:7860 (默认)

安装

  1. 克隆存储库

    git clone https://github.com/hkhkkh/stable-diffusion-mcp-server.git
    cd stable-diffusion-mcp-server
    
  2. 安装依赖项

    pip install -r requirements.txt
    
  3. 配置SD WebUI路径 (可选) 编辑 sd_mcp_server.py 并更新 SD_LAUNCHER_PATH 变量指向您的SD WebUI可执行文件。

  4. 启动MCP服务器

    python sd_mcp_server.py
    

使用示例

基本图像生成

Generate a cute cat sitting in a garden
Create a cyberpunk cityscape at night

采用Model/LoRA的先进一代

Using realistic model, draw a portrait with lora:detailed_face:0.8
Generate anime style artwork using anything model

图像到图像

Transform this image: C:\path\to\image.jpg into anime style
Based on this photo, create a fantasy version

模型管理

List all available SD models
Switch to majicmix_v7 model
Show current model information

🎨 支持的型号

服务器包括基于内容类型的智能模型推荐:

  • 真实模型:majicmix(v2,v7),chilloutix,写实系列
  • 动漫模型:任何东西,动漫,卡通风格
  • 专业型号:建筑、风景、肖像
  • LoRA支持:面部细节、风格提升、特定角色

📁 项目结构

stable-diffusion-mcp-server/
├── sd_mcp_server.py          # Main MCP server
├── requirements.txt          # Python dependencies
├── generated_images/         # Output directory
├── demo_ai_prompt_writing.py # AI prompt enhancement demo
├── test_*.py                # Test scripts
└── docs/                    # Additional documentation

🔧 配置

SD WebUI设置

  • 默认API URL: http://127.0.0.1:7860
  • 输出目录: generated_images/
  • Windows SD WebUI的自动启动支持

模型数据库

该服务器包括一个用于智能推荐的综合模型知识库:

  • 模型的优缺点
  • 最佳用例
  • 标签关联
  • 质量评级

🧪 测试

运行附带的测试脚本以验证功能:

python test_smart_generate.py    # Test smart generation
python test_current_model.py     # Test model info
python test_recommendations.py   # Test model recommendations

📖 文档

🤝 贡献

欢迎投稿!请阅读 许可证.md 许可条款。

非商业用途

  • 分叉存储库
  • 创建要素分支
  • 提交拉取请求

商业用途

  • 联系作者获取商业许可
  • 需要收入分成协议

📄 许可证

此项目使用 双重许可模式:

  • 非商业性:CC BY-NC-SA 4.0(个人/教育用途免费)
  • 商业的:具有收入分成要求的自定义许可证

看 许可证.md 了解全部细节。

📞 联系

  • 作者饰:Yang Haokang
  • GitHub: @香港
  • 电子邮件: 0221123105@mail.imu.edu.cn

对于商业许可咨询,请打开问题或直接联系。

🌟 致谢

  • 基于模型上下文协议(MCP)构建
  • 兼容AUTOMATIC1111稳定扩散WebUI
  • 受开源AI艺术社区的启发

由以下材料制成❤️ AI艺术社区