Lalanikarim Comfy MCP Server

🚀 舒適的MCP服務器

使用FastMCP框架，藉助遠程Comfy服務器依據提示生成圖像的服務器。

🚀 快速開始

此腳本藉助FastMCP框架搭建一個服務器，可依據提示語句，通過指定的工作流程生成圖像。它會與遠程Comfy服務器進行交互，提交提示信息並獲取生成的圖像。

✨ 主要特性

• 利用FastMCP框架搭建服務器，實現圖像生成功能。
• 與遠程Comfy服務器交互，提交提示並檢索生成的圖像。

📦 安裝指南

先決條件

• 已安裝Python 3.x。
• 必需的軟件包：mcp、json、urllib、time、os。
• 需從Comfy UI導出工作流程文件。此代碼包含示例Flux-Dev-ComfyUI-Workflow.json，僅作參考。你需要從自己的工作流程中導出並相應設置環境變量。

你可以使用pip安裝所需的軟件包：

pip install "mcp[cli]"

配置

設置以下環境變量：

• COMFY_URL：指向你的Comfy服務器URL。
• COMFY_WORKFLOW_JSON_FILE：指向你工作流程的API導出JSON文件的絕對路徑。
• PROMPT_NODE_ID：設置為你使用的文本提示節點ID。
• OUTPUT_NODE_ID：設置為包含最終圖像的輸出節點的ID。

示例：

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

💻 使用示例

基礎用法

直接運行腳本：

python comfy-mcp-server.py

服務器將啟動並監聽基於提供的提示生成圖像的請求。

📚 詳細文檔

功能性

generate_image(prompt: str, ctx: Context) -> Image | str

此函數根據指定的提示生成圖像，具體步驟如下：

1. 檢查所有環境變量是否已設置。
2. 從JSON文件加載提示模板。
3. 將提示提交到Comfy服務器。
4. 輪詢服務器以獲取提示處理的狀態。
5. 提示完成後檢索並返回生成的圖像。

依賴項

• mcp：用於設置FastMCP服務器。
• json：用於處理JSON數據。
• urllib：用於進行HTTP請求。
• time：用於在輪詢時添加延遲。
• os：用於訪問環境變量。

📄 許可證

此項目根據MIT許可證發佈，具體請參見LICENSE文件。