Image Gen With MCP

🚀 基於MCP接口的AI圖像生成服務器

本項目構建了一個集成穩定擴散（Stable Diffusion）圖像生成功能的模型上下文協議（MCP）服務器。藉助該服務器，AI代理能夠便捷地請求並接收生成的圖像。項目參考了Block's Goose自定義擴展教程中的示例。

🚀 快速開始

安裝配置

1. 創建虛擬環境，強制使用.venv：

virtualenv .venv

2. 激活虛擬環境：

source .venv/bin/activate

3. 安裝MCP包（用於Goose集成）：

pip install -e .

啟動服務

MCP服務器集成了圖像生成服務，可通過一條命令啟動兩者。

標準模式：

source .venv/bin/activate  # 激活你的虛擬環境
export IMAGE_GEN_DIR=/絕對路徑/to/folder # 設置生成圖像的目標文件夾
image-gen-mcp

開發模式（帶FastMCP Inspector）： 需打開兩個終端。

終端1：

source .venv/bin/activate  # 激活你的虛擬環境
export IMAGE_GEN_DIR=/絕對路徑/to/folder # 設置生成圖像的目標文件夾
image-gen-mcp # 啟動圖像生成服務（和一個我們不會使用的MCP服務器）

終端2：

source .venv/bin/activate  # 激活你的虛擬環境
export IMAGE_GEN_DIR=/絕對路徑/to/folder # 設置生成圖像的目標文件夾
mcp dev src/image_gen_mcp/server.py # 使用Inspector啟動MCP服務器

⚠️ 重要提示 僅在使用開發模式時，圖像生成服務必須單獨啟動。

啟動帶有FastMCP Inspector的MCP服務器後，可提供以下功能：

1. 一個Web界面 http://127.0.0.1:8000 進行調試
2. 檢查和調整圖像生成設置

直接API訪問

發送請求

使用curl命令或任何HTTP客戶端向服務發送請求，例如：

curl -X POST "http://localhost:5000/api/generate" \
     --data-raw '{"prompt": "一隻飛翔的貓頭鷹", "negative_prompt": ""}'

接收響應

{
  "status": "success",
  "message": "生成成功",
  "data": {
    "url": "http://localhost:5000/images/... png",
    "dimensions": "1024x1024"
  }
}

📦 安裝指南

詳細安裝步驟見上文“安裝配置”部分。

💻 使用示例

基礎用法

通過curl命令向服務發送請求生成圖像：

curl -X POST "http://localhost:5000/api/generate" \
     --data-raw '{"prompt": "一隻飛翔的貓頭鷹", "negative_prompt": ""}'

高級用法

在開發模式下，使用FastMCP Inspector啟動MCP服務器，可進行調試和調整圖像生成設置： 終端1：

source .venv/bin/activate  # 激活你的虛擬環境
export IMAGE_GEN_DIR=/絕對路徑/to/folder # 設置生成圖像的目標文件夾
image-gen-mcp # 啟動圖像生成服務（和一個我們不會使用的MCP服務器）

終端2：

source .venv/bin/activate  # 激活你的虛擬環境
export IMAGE_GEN_DIR=/絕對路徑/to/folder # 設置生成圖像的目標文件夾
mcp dev src/image_gen_mcp/server.py # 使用Inspector啟動MCP服務器

📚 詳細文檔

文件組織

項目文件結構如下：

.
├── src/
│   ├── image_gen_mcp/
│   │   ├── generator.py       # 處理圖像生成的主邏輯
│   │   └── server.py          # 提供MCP接口的服務器
│   └── requirements.txt      # 項目依賴項
└── README.md                  # 項目文檔

與Goose集成

環境變量配置

在Goose中添加一個環境變量IMAGE_GEN_DIR，指定生成圖像存儲的位置。

使用方法

在Goose中，你可以通過請求生成圖像並接收包含URL和元數據的MCP對象。該服務支持本地運行，但建議使用OpenAI GPT - 4o以獲得更快的響應速度。

服務架構

1. 圖像生成服務（src/image_gen_mcp/generator.py）

   • 使用穩定擴散模型進行實際圖像生成
   • 提供簡單的HTTP API接口
   • 返回圖像URL、維度和元數據
   • 默認運行在5000端口（可自定義）
   • 作為獨立線程運行，與MCP服務器共存

2. MCP服務器（src/image_gen_mcp/server.py）

   • 提供標準化的MCP接口供AI代理使用
   • 將請求轉發到集成的圖像生成服務
   • 返回包含URL和元數據的MCP對象

停止服務

按下Ctrl + C即可停止兩者服務，因為它們現在運行在同一進程中。