图像生成服务器

一个基于Replicate和Stable Diffusion的图像生成服务器,支持图像生成、保存和管理功能。

  • by rmcendarfer2017
  • Python
  • 本地部署
  • 更新于
在 GitHub 查看 rmcendarfer2017/mcp-image-gen

使用说明

图像生成器MCP服务器

使用Replicate生成图像并允许用户保存图像的MCP服务器。

组件

资源

服务器实现了一个图像存储系统,该系统具有:

  • 自定义图像://用于访问单个生成图像的URI方案
  • 每个图像资源都有一个基于其提示、包含创建日期的描述和image/png-mimety的名称

鼓励

服务器提供一个提示:

  • 生成图像:使用稳定扩散创建生成图像的提示
    • 可选的“风格”参数来控制图像风格(现实/艺术/抽象)
    • 生成具有特定样式指导的提示模板

工具

服务器实现了三个工具:

  • 生成图像:使用Replicate的稳定扩散模型生成图像
    • 将“prompt”作为必需的字符串参数
    • 可选参数包括“negative_prompt”、“width”、“height”、“num_inference_steps”和“guidance_scale”
    • 返回生成的图像及其URL
  • 保存映像:将生成的映像保存到本地文件系统
    • 将“image_url”和“prompt”作为必需的字符串参数
    • 为图像生成一个唯一的ID并将其保存到“generated_images”目录
  • 列出保存的图像:列出所有保存的图像
    • 返回所有已保存图像及其元数据和缩略图的列表

配置

复制API令牌

若要使用此图像生成器,您需要Replicate API令牌:

  1. 在以下网址创建帐户 复制
  2. 从获取您的API令牌 https://replicate.com/account
  3. 创建一个 .env 基于提供的文件 .env.example 模板:
REPLICATE_API_TOKEN=your_replicate_api_token_here

重要提示: 这 .env 通过以下方式将文件排除在版本控制之外 .gitignore 以防止意外暴露API令牌。切勿将敏感信息提交到存储库。

环境设置

  1. 克隆存储库:
git clone https://github.com/yourusername/image-generator.git
cd image-generator
  1. 创建并激活虚拟环境:
# Using venv
python -m venv .venv
# On Windows
.venv\Scripts\activate
# On macOS/Linux
source .venv/bin/activate
  1. 安装依赖项:
pip install -r requirements.txt
  1. 设置您的 .env 如上所述的文件

快速启动

安装

克劳德桌面

在MacOS上: ~/Library/Application\ Support/Claude/claude_desktop_config.json 在Windows上: %APPDATA%/Claude/claude_desktop_config.json

Development/Unpublished Servers Configuration ``` "mcpServers": { "image-generator": { "command": "uv", "args": [ "--directory", "B:\NEWTEST\image-generator", "run", "image-generator" ] } } ```
Published Servers Configuration ``` "mcpServers": { "image-generator": { "command": "uvx", "args": [ "image-generator" ] } } ```

用法

服务器运行后,您可以:

  1. 使用带有描述性提示的“生成图像”工具生成图像
  2. 使用带有图像URL和提示的“保存图像”工具保存生成的图像
  3. 使用“列出保存的图像”工具查看所有保存的图像
  4. 通过资源列表访问保存的图像

发展

建筑与出版

准备分发包裹:

  1. 同步依赖关系并更新锁文件:
uv sync
  1. 构建包分发:
uv build

这将在 dist/ 目录。

  1. 发布到PyPI:
uv publish

注意:您需要通过环境变量或命令标志设置PyPI凭据:

  • 令牌: --token 或 UV_PUBLISH_TOKEN
  • 或用户名/密码: --username/UV_PUBLISH_USERNAME 和 --password/UV_PUBLISH_PASSWORD

调试

由于MCP服务器在stdio上运行,调试可能具有挑战性。为了达到最佳调试效果 经验,我们强烈建议使用 MCP检查员.

您可以通过以下方式启动MCP检查器 npm 使用此命令:

npx @modelcontextprotocol/inspector uv --directory B:\NEWTEST\image-generator run image-generator

启动后,检查器将显示一个URL,您可以在浏览器中访问该URL以开始调试。