Servidor MCP cómodo

insignia de herrería

Un servidor que utiliza el marco FastMCP para generar imágenes basadas en indicaciones a través de un servidor Comfy remoto.

Descripción general

Este script configura un servidor que utiliza el framework FastMCP para generar imágenes basadas en indicaciones mediante un flujo de trabajo específico. Interactúa con un servidor Comfy remoto para enviar indicaciones y recuperar las imágenes generadas.

Related MCP server: Stability AI MCP Server

Prerrequisitos

Paquete uv y administrador de proyectos para Python.
Archivo de flujo de trabajo exportado desde Comfy UI. Este código incluye un ejemplo de Flux-Dev-ComfyUI-Workflow.json que se utiliza aquí solo como referencia. Deberá exportar desde su flujo de trabajo y configurar las variables de entorno según corresponda.

Puede instalar los paquetes necesarios para el desarrollo local:

uvx mcp[cli]

Configuración

Establezca las siguientes variables de entorno:

COMFY_URL para apuntar a la URL de su servidor Comfy.
COMFY_WORKFLOW_JSON_FILE para señalar la ruta absoluta del archivo json de exportación de API para el flujo de trabajo comfyui.
PROMPT_NODE_ID al id del nodo de solicitud de texto.
OUTPUT_NODE_ID al id del nodo de salida con la imagen final.
OUTPUT_MODE a url o file para seleccionar la salida deseada.

Opcionalmente, si tiene un servidor Ollama en ejecución, puede conectarse a él para generar mensajes.

OLLAMA_API_BASE a la URL donde se ejecuta ollama.
PROMPT_LLM al nombre del modelo alojado en ollama para la generación de solicitudes.

Ejemplo:

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

Uso

Se puede iniciar Comfy MCP Server con el siguiente comando:

uvx comfy-mcp-server

Ejemplo de configuración de escritorio de Claude

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

Funcionalidad

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

Esta función genera una imagen mediante una solicitud específica. Sigue estos pasos:

Comprueba si todas las variables de entorno están configuradas.
Carga una plantilla de aviso desde un archivo JSON.
Envía el mensaje al servidor Comfy.
Sondea al servidor para conocer el estado del procesamiento del mensaje.
Recupera y devuelve la imagen generada una vez que está lista.

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

Esta función genera un mensaje de generación de imágenes completo a partir de un tema específico.

Dependencias

mcp : para configurar el servidor FastMCP.
json : para manejar datos JSON.
urllib : Para realizar solicitudes HTTP.
time : para agregar retrasos en el sondeo.
os : Para acceder a las variables de entorno.
langchain : para crear una cadena de indicaciones LLM simple para generar indicaciones de generación de imágenes a partir del tema.
langchain-ollama : para módulos específicos de ollama para LangChain.