图像生成器 MCP 服务器

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

组件

资源

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

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

提示

服务器提供一个提示：

• generate-image：创建使用 Stable Diffusion 生成图像的提示
  • 可选的 "style" 参数来控制图像风格（写实/艺术/抽象）
  • 生成一个带有特定风格指导的提示模板

工具

服务器实现了三个工具：

• generate-image：使用 Replicate 的 Stable Diffusion 模型生成图像
  • 接受 "prompt" 作为必需的字符串参数
  • 可选参数包括 "negative_prompt"、"width"、"height"、"num_inference_steps" 和 "guidance_scale"
  • 返回生成的图像及其 URL
• save-image：将生成的图像保存到本地文件系统
  • 接受 "image_url" 和 "prompt" 作为必需的字符串参数
  • 为图像生成一个唯一的 ID 并将其保存到 "generated_images" 目录
• list-saved-images：列出所有已保存的图像
  • 返回所有已保存图像的列表，其中包含其元数据和缩略图

配置

Replicate API Token

要使用此图像生成器，您需要一个 Replicate API token：

1. 在 Replicate 创建一个帐户
2. 从 https://replicate.com/account 获取您的 API token
3. 基于提供的 .env.example 模板创建一个 .env 文件：

REPLICATE_API_TOKEN=your_replicate_api_token_here

重要提示： .env 文件已通过 .gitignore 从版本控制中排除，以防止意外暴露您的 API token。切勿将敏感信息提交到您的存储库。

环境设置

1. 克隆存储库：

git clone https://github.com/yourusername/image-generator.git
cd image-generator

2. 创建并激活虚拟环境：

# 使用 venv
python -m venv .venv
# 在 Windows 上
.venv\Scripts\activate
# 在 macOS/Linux 上
source .venv/bin/activate

3. 安装依赖项：

pip install -r requirements.txt

4. 按照上述说明设置您的 .env 文件

快速入门

安装

Claude Desktop

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

<details> <summary>开发/未发布的服务器配置</summary>

"mcpServers": {
  "image-generator": {
    "command": "uv",
    "args": [
      "--directory",
      "B:\NEWTEST\image-generator",
      "run",
      "image-generator"
    ]
  }
}

</details>

<details> <summary>已发布的服务器配置</summary>

"mcpServers": {
  "image-generator": {
    "command": "uvx",
    "args": [
      "image-generator"
    ]
  }
}

</details>

用法

服务器运行后，您可以：

1. 使用带有描述性提示的 "generate-image" 工具生成图像
2. 使用带有图像 URL 和提示的 "save-image" 工具保存生成的图像
3. 使用 "list-saved-images" 工具查看所有已保存的图像
4. 通过资源列表访问已保存的图像

开发

构建和发布

要准备用于分发的软件包：

1. 同步依赖项并更新 lockfile：

uv sync

2. 构建软件包分发：

uv build

这将在 dist/ 目录中创建源和 wheel 分发。

3. 发布到 PyPI：

uv publish

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

• Token：--token 或 UV_PUBLISH_TOKEN
• 或用户名/密码：--username/UV_PUBLISH_USERNAME 和 --password/UV_PUBLISH_PASSWORD

调试

由于 MCP 服务器通过 stdio 运行，因此调试可能具有挑战性。 为了获得最佳调试体验，我们强烈建议使用 MCP Inspector。

您可以使用以下命令通过 npm 启动 MCP Inspector：

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

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