gen

/comfy-gen — Generate an Image

The user wants to generate an image using ComfyUI. Their prompt is provided as the argument to this command.

Instructions

1. Parse the user's prompt. The argument after /comfy-gen is the image description: $ARGUMENTS

   If no argument was provided, ask the user what they'd like to generate.

2. Check available models. Use the list_local_models tool with model_type: "checkpoints" to see what checkpoints are installed.

3. If no checkpoints are installed, acquire one automatically.

   Do NOT ask the user to download a model. Instead, find and download one yourself:

   Option A: CivitAI (preferred if CIVITAI_API_TOKEN is set):

   • Use WebFetch to search CivitAI's REST API: https://civitai.com/api/v1/models?query=SDXL&types=Checkpoint&sort=Most+Downloaded&limit=5
   • Pick the top result that matches the user's needs (SDXL for general, Flux for quality, etc.)
   • Get the download URL from the model version: https://civitai.com/api/download/models/{modelVersionId}?token={CIVITAI_API_TOKEN}
   • Use download_model with target_subfolder: "checkpoints" to download it

   Option B: HuggingFace (default):

   • Use download_model action:"search" to find a suitable checkpoint (e.g., query "SDXL" or "stable diffusion xl")
   • Find the direct .safetensors download URL from the model page (typically https://huggingface.co/{repo}/resolve/main/{filename})
   • Use download_model with target_subfolder: "checkpoints" to download it

   Sensible defaults by category:

   • General purpose: SDXL base (stabilityai/stable-diffusion-xl-base-1.0)
   • Fast generation: SDXL Turbo or Lightning
   • High quality: Flux.1 schnell or dev (if user has enough VRAM)
   • Anime: Anything-XL or similar

   Tell the user what you're downloading and why. Large checkpoints can be 2-7 GB.

4. Create the workflow. Use the create_workflow tool with:

   • template: "txt2img"
   • params: Include positive_prompt from the user's input. Set checkpoint to the model filename. Use sensible defaults (1024x1024 for SDXL, 20 steps, cfg 8).