🚀 舒适的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许可证的保护。