SamuraiBuddha/mcp-comfyui

# Clone the repository
git clone https://github.com/SamuraiBuddha/mcp-comfyui.git
cd mcp-comfyui

# Install dependencies
pip install -e .

# If installed locally
cd /path/to/ComfyUI
python main.py --listen

# Or use Docker
docker-compose up -d

# Copy example config
cp .env.example .env

# Edit .env with your settings
# COMFYUI_HOST=localhost
# COMFYUI_PORT=8188

{
  "mcpServers": {
    "comfyui": {
      "command": "python",
      "args": ["-m", "mcp_comfyui"],
      "cwd": "/path/to/mcp-comfyui",
      "env": {
        "COMFYUI_HOST": "localhost",
        "COMFYUI_PORT": "8188"
      }
    }
  }
}

generate_image(
    prompt="A futuristic robot logo for Crisis Corps",
    negative_prompt="blurry, low quality",
    width=512,
    height=512,
    steps=20,
    cfg_scale=7.0,
    seed=-1,  # Random seed
    model="sd_xl_base_1.0.safetensors",
    sampler="euler",
    scheduler="normal"
)

build_workflow(
    nodes=[
        {
            "id": "1",
            "type": "CheckpointLoaderSimple",
            "inputs": {"ckpt_name": "sd_xl_base_1.0.safetensors"}
        },
        {
            "id": "2",
            "type": "CLIPTextEncode",
            "inputs": {"text": "robot logo", "clip": ["1", 1]}
        },
        {
            "id": "3",
            "type": "KSampler",
            "inputs": {
                "model": ["1", 0],
                "positive": ["2", 0],
                "seed": 42,
                "steps": 20
            }
        }
    ]
)

save_workflow(
    name="my_logo_workflow",
    workflow=built_workflow,
    description="Custom workflow for Crisis Corps logos"
)

execute_workflow(
    workflow_name="logo_generator",
    inputs={
        "prompt": "Crisis Corps emblem",
        "style": "military insignia"
    }
)

list_models()
# Returns: ["sd_xl_base_1.0.safetensors", "dreamshaper_8.safetensors", ...]

list_samplers()
# Returns: ["euler", "euler_ancestral", "dpm_2", "dpm_2_ancestral", ...]

list_schedulers()
# Returns: ["normal", "karras", "exponential", "sgm_uniform", ...]

get_node_types()
# Returns complete node definitions with inputs/outputs

upload_image(
    image_path="/path/to/image.png",
    name="reference_image"
)

list_workflows()
# Returns: ["logo_generator.json", "crisis_corps_logo.json", ...]

get_queue_status()
# Returns: {"queue_remaining": 2, "currently_processing": "prompt_123"}

get_history(limit=10)
# Returns list of recent generations with IDs and parameters

get_image(prompt_id="abc123")
# Returns: base64 encoded image or filepath

interrupt_generation()

clear_queue()

create_workflow_advanced(
    template="txt2img_basic",  # or "txt2img_with_lora", "upscale_workflow", "custom"
    settings={
        "prompt": "your prompt",
        "model": "model_name.safetensors",
        # ... other settings
    }
)

add_workflow_node(
    node_type="KSampler",
    inputs={
        "seed": 42,
        "steps": 20,
        "cfg": 7.0
    },
    node_id="sampler_1"  # optional
)

connect_nodes(
    from_node="1",
    from_output=0,
    to_node="2",
    to_input="model"
)

apply_quality_preset(
    quality="balanced"  # "draft", "fast", "balanced", "quality", "extreme"
)

apply_style_preset(
    style="photorealistic"  # "artistic", "anime", "digital_art", "oil_painting"
)

batch_generate(
    base_prompt="tech company logo",
    variations=[
        {"prompt_suffix": "blue theme", "seed": 100},
        {"prompt_suffix": "minimalist", "seed": 200, "style": "digital_art"},
        {"prompt_suffix": "3D render", "seed": 300, "cfg_scale": 9.0}
    ],
    base_settings={"model": "sd_xl_base_1.0.safetensors"}
)

# SDXL for high quality
result = await generate_image(
    prompt="Crisis Corps logo, heroic robot emblem, orange and blue",
    model="sd_xl_base_1.0.safetensors",
    width=1024,
    height=1024,
    steps=35
)

# DreamShaper for stylized look
result = await generate_image(
    prompt="Crisis Corps logo, heroic robot emblem, orange and blue",
    model="dreamshaper_8.safetensors",
    sampler="dpm_2_ancestral",
    scheduler="karras"
)

# Create a workflow with LoRA for consistent style
workflow = await build_workflow(
    nodes=[
        {"id": "1", "type": "CheckpointLoaderSimple", 
         "inputs": {"ckpt_name": "sd_xl_base_1.0.safetensors"}},
        {"id": "2", "type": "LoraLoader",
         "inputs": {"model": ["1", 0], "clip": ["1", 1], 
                   "lora_name": "logo_style.safetensors",
                   "strength_model": 0.8, "strength_clip": 0.8}},
        {"id": "3", "type": "CLIPTextEncode",
         "inputs": {"text": "Crisis Corps emblem", "clip": ["2", 1]}},
        # ... rest of workflow
    ]
)

# Save for reuse
await save_workflow(
    name="crisis_corps_lora_workflow",
    workflow=workflow,
    description="Logo generation with consistent style LoRA"
)

# Create a complex workflow with multiple models and processing stages
await create_workflow_advanced(template="custom")

# Stage 1: Base generation
await add_workflow_node("CheckpointLoaderSimple", {"ckpt_name": "sd_xl_base_1.0.safetensors"}, "1")
await add_workflow_node("CLIPTextEncode", {"text": "detailed portrait"}, "2")
await add_workflow_node("KSampler", {"steps": 30, "cfg": 7.5}, "3")

# Stage 2: Enhancement
await add_workflow_node("FreeU", {"b1": 1.1, "b2": 1.2}, "4")
await add_workflow_node("UpscaleModelLoader", {"model_name": "4x-UltraSharp.pth"}, "5")

# Connect everything
await connect_nodes("1", 0, "4", "model")  # checkpoint -> FreeU
await connect_nodes("4", 0, "3", "model")  # FreeU -> sampler
# ... more connections

workflow = await get_workflow_json()

# Get optimal settings for any model
optimal = await get_optimal_settings("sd3.5_large.safetensors")
# Returns: {"sampling": {"steps": 28, "cfg_scale": 4.5, "scheduler": "simple"}}

# Apply quality presets
await apply_quality_preset("extreme")  # 50 steps, high quality
await apply_style_preset("photorealistic")  # Optimized for realism

# Save your configuration
await save_settings_preset("my_portrait_settings")