ComfyUI NAI Prompt Converter

Original source code credit: The prompt_to_stack function is based on the code from: https://github.com/bedovyy/ComfyUI_NAIDGenerator/blob/master/utils.py#L146 Author: bedovyy Modified for Novel AI to ComfyUI conversion purposes.

This code was entirely created by Claude 3.5 Sonnet.

A custom node extension for ComfyUI that enables conversion between NovelAI and ComfyUI prompt formats, along with extraction of NovelAI metadata from PNG images.

Features

NAI Prompt Extractor Node

• Extracts metadata from NovelAI-generated PNG images including:
  • Prompt
  • Negative prompt
  • Seed
  • Steps
  • Sampler
  • CFG scale
  • Raw metadata
  • Image dimensions
• Supports direct image upload through the UI
• Specifically designed for PNG files with NovelAI metadata

Prompt Format Conversion

Provides bidirectional conversion between NovelAI and ComfyUI prompt formats:

Novel AI → ComfyUI

• Converts NovelAI's bracket-based emphasis notation to ComfyUI's weight format
  • {tag} → (tag:1.05)
  • {{tag}} → (tag:1.10)
  • [tag] → (tag:0.95)
  • [[tag]] → (tag:0.90)

ComfyUI → Novel AI

• Converts ComfyUI's weight format to NovelAI's bracket notation
  • (tag:1.05) → {tag}
  • (tag:1.10) → {{tag}}
  • (tag:0.95) → [tag]
  • (tag:0.90) → [[tag]]

Special Tag Handling

• Preserves character tags in various formats:
  • Basic: character (series)
  • With options: character (series) (outfit) (pose)
  • Underscore format: character_(series)
  • Escaped format: character \(series\)
• Maintains correct formatting for artist: tags

Installation

1. Clone this repository into your ComfyUI's custom nodes directory:

cd ComfyUI/custom_nodes
git clone https://github.com/raspie10032/ComfyUI_RS_NAI_Local_Prompt_converter

2. Restart ComfyUI

Example Workflow

The workflow is embedded in the PNG metadata - just drag and drop the image to load it in ComfyUI

Extensions Required for Example Workflow

ComfyUI-Manager

• https://github.com/ltdrdata/ComfyUI-Manager

ComfyUI Impact Pack

• https://github.com/ltdrdata/ComfyUI-Impact-Pack

Efficiency Nodes for ComfyUI Version 2.0+

• https://github.com/jags111/efficiency-nodes-comfyui

pythongosssss/ComfyUI-Custom-Scripts

• https://github.com/pythongosssss/ComfyUI-Custom-Scripts

UltimateSDUpscale

• https://github.com/ssitu/ComfyUI_UltimateSDUpscale

rgthree's ComfyUI Nodes

• https://github.com/rgthree/rgthree-comfy

Use Everywhere (UE Nodes)

• https://github.com/chrisgoringe/cg-use-everywhere

Code References

This project incorporates code and techniques from the following sources:

ComfyUI NAIDGenerator

• Repository: https://github.com/bedovyy/ComfyUI_NAIDGenerator
• Author: bedovyy
• Used for: Prompt conversion functionality and base implementation

NovelAI Image Metadata

• Repository: https://github.com/NovelAI/novelai-image-metadata
• Author: NovelAI
• Used for: LSB-based metadata extraction technique and stealth watermark handling

Special thanks to the authors and contributors of these projects for their valuable work and making their code available to the community.