LM Studio → ComfyUI генератор изображений

MCP-сервер ComfyUI

MCP-сервер (Model Context Protocol), который предоставляет генерацию изображений и 3D в ComfyUI как единый инструмент.
Он автоматически классифицирует ваш промпт (люди, здания, общее или 3D) и выбирает подходящий воркфлоу автоматически.

Идеально для ИИ-ассистентов – подключите его через MCP, и пусть ваша LLM генерирует ассеты напрямую.

Возможности

• 🧠 Умная классификация промптов – определяет, нужен ли вам человек, здание или 3D-модель.

• 🖼️ 2D-генерация – использует ваши собственные воркфлоу ComfyUI (general, people, buildings).

• 🤖 3D-генерация (Hunyuan3D) – сначала создаёт референсное изображение, затем запускает 3D-воркфлоу.

• 🔌 MCP-инструмент – предоставляет generate_asset_tool(prompt, workflow_type, size) любому MCP-клиенту.

• 📂 Папка воркфлоу – поместите свои .json-воркфлоу в ./workflows/.

Установка

1. Предварительные требования

• Python 3.10+

• ComfyUI, запущенный локально с включённым API (по умолчанию http://127.0.0.1:8188).

• Ваши воркфлоу ComfyUI, экспортированные в файлы .json.

2. Установка зависимостей

pip install -r requirements.txt

3. Разместите свои воркфлоу

Поместите файлы JSON ваших воркфлоу ComfyUI в папку workflows/. Имена файлов должны точно совпадать:

general.json

people.json

buildings.json

3d_simple.json

⚠️ Важно для 3D: ваш 3d_simple.json ОБЯЗАН содержать узел LoadImage с ID "26". Код внедряет загруженное референсное изображение в workflow["26"]["inputs"]["image"].

4. Настройка (необязательно) Создайте файл .env, чтобы переопределить URL ComfyUI:

text COMFY_URL=http://127.0.0.1:8188 Если не задано, по умолчанию используется http://127.0.0.1:8188.

Использование Запустите MCP-сервер bash python mcp_server.py Сервер запустится и зарегистрирует инструмент generate_asset_tool.

Параметры инструмента prompt (str, required): описание того, что нужно сгенерировать.

workflow_type (str, optional): "people", "buildings", "general" или "3d_simple". Если не указано – определяется автоматически.

size (str, optional): "512x512", "512x768" или "768x512". Если не указано – выбирается автоматически.

Пример из MCP-клиента json { "tool": "generate_asset_tool", "arguments": { "prompt": "a futuristic cyberpunk girl with neon hair" } } → Автоматически определяет категорию "people", использует people.json с размером 512x768, возвращает URL изображения.

json { "tool": "generate_asset_tool", "arguments": { "prompt": "turn this into a 3d model: a wooden chair", "workflow_type": "3d_simple" } } → Генерирует референсное изображение (через general), затем запускает 3d_simple.json на нём. Возвращает оба URL.

Как это работает Классификация – сканирует ваш промпт на наличие ключевых слов (girl, building, 3d и т.д.), чтобы выбрать правильный воркфлоу и разрешение.

Промпт-инъекция – находит узлы CLIPTextEncode и заполняет позитивный промпт (пользовательский запрос) и негативный промпт (встроенные пресеты).

Случайный сид – рандомизирует сиды KSampler для уникальных результатов каждый раз.

Очередь и опрос – отправляет воркфлоу в ComfyUI через /prompt и опрашивает /history, пока результат не будет готов.

Вывод – возвращает полный URL для просмотра/скачивания сгенерированного файла.

Обработка ошибок Если ComfyUI не запущен – возвращает ERROR

При сбое – автоматически повторяет попытку до 2 раз.

Подробные логи выводятся в консоль.