Comfy MCP Server

🚀 舒適的MCP服務器

這是一個使用FastMCP框架的服務器，可通過遠程Comfy服務器依據提示生成圖像，為圖像生成工作提供了便捷的解決方案。

🚀 快速開始

此腳本藉助FastMCP框架搭建服務器，該服務器能根據提示，通過指定工作流程與遠程Comfy服務器交互來生成圖像。

✨ 主要特性

• 圖像生成：通過generate_image函數，依據指定提示生成圖像，涵蓋檢查環境變量、加載提示模板、提交提示、監控狀態以及檢索圖像等步驟。
• 提示生成：generate_prompt函數可根據指定主題生成全面的圖像生成提示。

📦 安裝指南

先決條件

• uv 包和Python項目管理器。
• 從Comfy UI導出的工作流文件。此代碼包含示例Flux - Dev - ComfyUI - Workflow.json，僅作參考，您需導出自己的工作流程並設置相應環境變量。

您可使用以下命令安裝本地開發所需的包：

uvx mcp[cli]

配置

設置以下環境變量：

• COMFY_URL：指向您的Comfy服務器URL。
• COMFY_WORKFLOW_JSON_FILE：Comfy UI工作流導出的JSON文件的絕對路徑。
• PROMPT_NODE_ID：文本提示節點的ID。
• OUTPUT_NODE_ID：具有最終圖像的輸出節點的ID。
• OUTPUT_MODE：設置為url或file以選擇所需輸出。

可選：若運行Ollama服務器，可連接以生成提示。

• OLLAMA_API_BASE：Ollama運行的URL。
• PROMPT_LLM：Ollama上託管的用於提示生成的模型名稱。

示例：

export COMFY_URL=http://your-comfy-server-url:port
export COMFY_WORKFLOW_JSON_FILE=/path/to/the/comfyui_workflow_export.json
export PROMPT_NODE_ID=6 # 使用正確的節點ID
export OUTPUT_NODE_ID=9 # 使用正確的節點ID
export OUTPUT_MODE=file

💻 使用示例

基礎用法

通過以下命令啟動Comfy MCP服務器：

uvx comfy-mcp-server

高級用法

Claude Desktop配置示例

{
  "mcpServers": {
    "Comfy MCP Server": {
      "command": "/path/to/uvx",
      "args": [
        "comfy-mcp-server"
      ],
      "env": {
        "COMFY_URL": "http://your-comfy-server-url:port",
        "COMFY_WORKFLOW_JSON_FILE": "/path/to/the/comfyui_workflow_export.json",
        "PROMPT_NODE_ID": "6",
        "OUTPUT_NODE_ID": "9",
        "OUTPUT_MODE": "file"
      }
    }
  }
}

🔧 技術細節

依賴項

• mcp：用於設置FastMCP服務器。
• json：用於處理JSON數據。
• urllib：用於進行HTTP請求。
• time：用於在輪詢時添加延遲。
• os：用於訪問環境變量。
• langchain：用於創建簡單的LLM提示鏈，從主題生成圖像生成提示。
• langchain - ollama：用於與Ollama集成。

📄 許可證

此項目受MIT許可證的保護。