Alex
7130dcb2df
New node ETN_KritaStyleAndPrompt that works like KritaStyle but: - Prompts and style sync between Generate/Live/Animation/Graph workspaces - Outputs fully prepared prompts (wildcards evaluated, style merged) - Model output includes extracted LoRAs from prompts
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from comfy_api.latest import ComfyExtension, io
|
|
from . import api as api, nodes, tile, region, nsfw, translation, krita
|
|
|
|
|
|
class ExternalToolingNodes(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
nodes.LoadImageCache,
|
|
nodes.SaveImageCache,
|
|
nodes.LoadImageBase64,
|
|
nodes.LoadMaskBase64,
|
|
nodes.SendImageWebSocket,
|
|
nodes.ApplyMaskToImage,
|
|
nodes.ReferenceImage,
|
|
nodes.ApplyReferenceImages,
|
|
tile.CreateTileLayout,
|
|
tile.ExtractImageTile,
|
|
tile.ExtractMaskTile,
|
|
tile.GenerateTileMask,
|
|
tile.MergeImageTile,
|
|
region.BackgroundRegion,
|
|
region.DefineRegion,
|
|
region.ListRegionMasks,
|
|
region.AttentionMask,
|
|
nsfw.NSFWFilter,
|
|
translation.Translate,
|
|
krita.KritaOutput,
|
|
krita.KritaSendText,
|
|
krita.KritaCanvas,
|
|
krita.KritaSelection,
|
|
krita.KritaImageLayer,
|
|
krita.KritaMaskLayer,
|
|
krita.Parameter,
|
|
krita.KritaStyle,
|
|
krita.KritaStyleAndPrompt,
|
|
]
|
|
|
|
|
|
async def comfy_entrypoint():
|
|
return ExternalToolingNodes()
|
|
|
|
|
|
WEB_DIRECTORY = "./js"
|