Comfy MCP Server

Comfy MCP Server

FastMCP フレームワークを使用し、リモートの Comfy サーバー経由でプロンプトに基づいて画像を生成するサーバーです。

概要

このスクリプトは、FastMCP フレームワークを使用して、指定したワークフローに基づいてプロンプトから画像を生成するサーバーを構成します。リモートの Comfy サーバーと連携し、プロンプトを送信して生成された画像を取得します。

前提条件

• uv：Python 用のパッケージおよびプロジェクト管理ツールです。
• Comfy UI からエクスポートしたワークフローファイル。このコードにはサンプルの Flux-Dev-ComfyUI-Workflow.json が含まれていますが、これは参照用です。ご自身のワークフローからエクスポートし、それに応じて環境変数を設定する必要があります。

ローカル開発に必要なパッケージは、次のようにインストールできます。

uvx mcp[cli]

設定

次の環境変数を設定します。

• COMFY_URL に Comfy サーバーの URL を指定します。
• COMFY_WORKFLOW_JSON_FILE に Comfy UI ワークフローの API エクスポート 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 # use the correct node id here
export OUTPUT_NODE_ID=9 # use the correct node id here
export OUTPUT_MODE=file

使用方法

Comfy MCP Server は、次のコマンドで起動できます。

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",
      }
    }
  }
}

機能

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

この関数は、指定したプロンプトを使用して画像を生成します。次の手順を実行します。

1. すべての環境変数が設定されていることを確認します。
2. JSON ファイルからプロンプトテンプレートを読み込みます。
3. プロンプトを Comfy サーバーに送信します。
4. プロンプトの処理状況をサーバーに問い合わせます。
5. 生成が完了したら、画像を取得して返します。

generate_prompt(topic: str, ctx: Context) -> str

この関数は、指定したトピックから包括的な画像生成プロンプトを生成します。

依存関係

• mcp：FastMCP サーバーのセットアップに使用します。
• json：JSON データの処理に使用します。
• urllib：HTTP リクエストの送信に使用します。
• time：ポーリング時の待機に使用します。
• os：環境変数へのアクセスに使用します。
• langchain：トピックから画像生成プロンプトを作成するシンプルな LLM プロンプトチェーンの構築に使用します。
• langchain-ollama：LangChain 用の Ollama 固有モジュールです。

ライセンス

このプロジェクトは MIT License のもとでライセンスされています。詳細については、LICENSE ファイルをご覧ください。