Image Gen MCP

🚀 圖像生成 MCP 服務器

本項目是一個提供文本轉圖像生成功能的MCP服務器，藉助Stable Diffusion WebUI API（ForgeUI/AUTOMATIC-1111）實現強大的圖像生成能力。

🚀 快速開始

本服務器可幫助你輕鬆實現文本到圖像的轉換以及圖像放大等功能。下面為你詳細介紹如何使用。

📦 安裝指南

先決條件

• Node.js
• 能夠訪問帶有API啟用的Stable Diffusion WebUI實例
• 啟動WebUI時必須使用--api標誌

設置步驟

1. 克隆倉庫

git clone https://github.com/Ichigo3766/image-gen-mcp.git
cd image-gen-mcp

2. 安裝依賴項

npm install

3. 構建服務器

npm run build

4. 添加服務器配置到環境變量

{
  "mcpServers": {
    "image-gen": {
      "command": "node",
      "args": [],
      "env": {
        "SD_WEBUI_URL": "http://localhost:7860",
        "SD_API_KEY": "your_api_key", // 可選，如果WebUI啟用了API密鑰
        "SD_MODEL_NAME": "模型名稱" // 可選，默認為WebUI中的默認模型
      }
    }
  }
}

✨ 主要特性

文本轉圖像生成

使用/api/generate-image端點即可輕鬆生成圖像。請求體包含以下參數：

{
  "prompt": "輸入提示詞",
  "negative_prompt": "排除提示詞", // 可選，默認為空
  "width": 512,
  "height": 512,
  "samples": 1,
  "num_inference_steps": 20,
  "safety_checker": true, // 開關安全檢查，防止生成不適當內容，默認啟用
  "enhance_prompt": true, // 啟用提示詞增強，默認啟用
  "seed": null // 隨機種子，可選，默認為隨機值
}

圖像 upscale

使用/api/upscale-images端點可對圖像進行放大處理。請求體包含以下參數：

{
  "images": ["image1.jpg", "image2.png"], // 要放大的圖像文件路徑數組
  "resize_mode": 0, // 0表示按比例縮放，1表示固定尺寸，默認從環境變量獲取
  "upscaling_resize": 4, // 縮放倍率，當resize_mode=0時使用，默認從環境變量獲取
  "upscaling_resize_w": 1024, // 目標寬度（像素），當resize_mode=1時使用，默認從環境變量獲取
  "upscaling_resize_h": 1024, // 目標高度（像素），當resize_mode=1時使用，默認從環境變量獲取
  "upscaler_1": "RealESRGAN_x4Plus", // 主要放大器模型， 默認從環境變量獲取
  "upscaler_2": null, // 次級放大器模型，默認為null
  "output_path": "upscaled_images" // 放大後圖像的輸出目錄，默認為當前目錄/upscaled-images
}

💻 使用示例

開發模式

若你需要在開發過程中使用自動重建功能，可運行以下命令：

npm run watch

📚 詳細文檔

錯誤處理

在使用過程中，若遇到問題可參考以下常見問題及解決方案：

1. 確保你的Stable Diffusion WebUI正在使用--api標誌運行
2. 檢查WebUI URL是否可以從MCP服務器訪問
3. 如果使用了身份驗證，請確保憑據正確
4. 驗證輸出目錄存在且有寫入權限
5. 當放大圖像時，請確保輸入圖像文件存在且可讀

📄 許可證

此MCP服務器根據MIT License授權。這意味著你可以自由地使用、修改和分發軟件，但需遵守MIT License的條款和條件。有關詳細信息，請參閱項目倉庫中的LICENSE文件。