image-gen MCP Server

基本能力

产品定位

image-gen-mcp 是一个基于 Stable Diffusion WebUI API 的 MCP 服务器，主要用于文本到图像的生成和图像增强。

核心功能

1. 文本到图像生成：根据文本提示生成图像。
2. 模型管理：获取和设置可用的 Stable Diffusion 模型。
3. 图像放大：使用不同的放大模型对图像进行放大处理。
4. 参数定制：支持多种参数定制，如采样步骤、图像尺寸、CFG 比例等。

适用场景

1. 内容创作：快速生成基于文本描述的图像。
2. 图像增强：对现有图像进行放大和优化。
3. 模型测试：测试不同的 Stable Diffusion 模型效果。

工具列表

1. generate_image：生成图像。
2. 参数：prompt（必需）、negative_prompt、steps、width、height、cfg_scale、sampler_name、scheduler_name、seed、batch_size、restore_faces、tiling、output_path。
3. get_sd_models：获取可用的 Stable Diffusion 模型列表。
4. set_sd_model：设置当前使用的 Stable Diffusion 模型。
5. get_sd_upscalers：获取可用的放大模型列表。
6. upscale_images：放大图像。
7. 参数：images（必需）、resize_mode、upscaling_resize、upscaling_resize_w、upscaling_resize_h、upscaler_1、upscaler_2、output_path。

常见问题解答

1. Stable Diffusion WebUI 未启用 API：确保启动 WebUI 时添加 --api 标志。
2. WebUI URL 不可访问：检查 URL 是否正确且可从 MCP 服务器访问。
3. 认证失败：确保提供的用户名和密码正确。
4. 输出目录权限问题：确保输出目录存在且有写入权限。
5. 图像放大失败：确保输入图像文件存在且可读。

使用教程

使用依赖

1. Node.js：确保已安装 Node.js。
2. Stable Diffusion WebUI：确保已安装并启用 API（启动时添加 --api 标志）。

安装教程

1. 克隆仓库： bash git clone https://github.com/Ichigo3766/image-gen-mcp.git cd image-gen-mcp
2. 安装依赖： bash npm install
3. 构建服务器： bash npm run build
4. 配置环境变量： json { "mcpServers": { "image-gen": { "command": "node", "args": [ "/path/to/image-gen-mcp/build/index.js" ], "env": { "SD_WEBUI_URL": "http://your-sd-webui-url:7860", "SD_AUTH_USER": "your-username", "SD_AUTH_PASS": "your-password", "SD_OUTPUT_DIR": "/path/to/output/directory", "SD_RESIZE_MODE": "0", "SD_UPSCALE_MULTIPLIER": "4", "SD_UPSCALE_WIDTH": "512", "SD_UPSCALE_HEIGHT": "512", "SD_UPSCALER_1": "R-ESRGAN 4x+", "SD_UPSCALER_2": "None" } } } }

调试方式

1. 开发模式（自动重建）： bash npm run watch
2. 检查日志：确保所有环境变量正确且服务正常运行。

许可证

该项目遵循 MIT 开源许可条款，请参阅 MIT 了解完整条款。