39 lines
1.6 KiB
Python
39 lines
1.6 KiB
Python
from . import api, nodes, tile, region, nsfw, translation
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ETN_LoadImageBase64": nodes.LoadImageBase64,
|
|
"ETN_LoadMaskBase64": nodes.LoadMaskBase64,
|
|
"ETN_SendImageWebSocket": nodes.SendImageWebSocket,
|
|
"ETN_CropImage": nodes.CropImage,
|
|
"ETN_ApplyMaskToImage": nodes.ApplyMaskToImage,
|
|
"ETN_TileLayout": tile.TileLayout,
|
|
"ETN_ExtractImageTile": tile.ExtractImageTile,
|
|
"ETN_ExtractMaskTile": tile.ExtractMaskTile,
|
|
"ETN_GenerateTileMask": tile.GenerateTileMask,
|
|
"ETN_MergeImageTile": tile.MergeImageTile,
|
|
"ETN_BackgroundRegion": region.BackgroundRegion,
|
|
"ETN_DefineRegion": region.DefineRegion,
|
|
"ETN_ListRegionMasks": region.ListRegionMasks,
|
|
"ETN_AttentionMask": region.AttentionMask,
|
|
"ETN_NSFWFilter": nsfw.NSFWFilter,
|
|
"ETN_Translate": translation.Translate,
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ETN_LoadImageBase64": "Load Image (Base64)",
|
|
"ETN_LoadMaskBase64": "Load Mask (Base64)",
|
|
"ETN_SendImageWebSocket": "Send Image (WebSocket)",
|
|
"ETN_CropImage": "Crop Image",
|
|
"ETN_ApplyMaskToImage": "Apply Mask to Image",
|
|
"ETN_TileLayout": "Create Tile Layout",
|
|
"ETN_ExtractImageTile": "Extract Image Tile",
|
|
"ETN_ExtractMaskTile": "Extract Mask Tile",
|
|
"ETN_MergeImageTile": "Merge Image Tile",
|
|
"ETN_GenerateTileMask": "Generate Tile Mask",
|
|
"ETN_BackgroundRegion": "Background Region",
|
|
"ETN_DefineRegion": "Define Region",
|
|
"ETN_ListRegionMasks": "List Region Masks",
|
|
"ETN_AttentionMask": "Regions Attention Mask",
|
|
"ETN_NSFWFilter": "NSFW Filter",
|
|
"ETN_Translate": "Translate Text",
|
|
}
|