mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-06 18:07:03 +08:00
532 lines
17 KiB
Python
532 lines
17 KiB
Python
from inspect import cleandoc
|
|
from comfy.comfy_types.node_typing import IO
|
|
from comfy_api_nodes.apis.recraft_api import (
|
|
RecraftImageGenerationRequest,
|
|
RecraftImageGenerationResponse,
|
|
RecraftImageSize,
|
|
RecraftModel,
|
|
RecraftStyle,
|
|
RecraftStyleV3,
|
|
RecraftColor,
|
|
RecraftColorChain,
|
|
RecraftControls,
|
|
RecraftIO,
|
|
get_v3_substyles,
|
|
)
|
|
from comfy_api_nodes.apis.client import (
|
|
ApiEndpoint,
|
|
HttpMethod,
|
|
SynchronousOperation,
|
|
)
|
|
from comfy_api_nodes.apinode_utils import (
|
|
bytesio_to_image_tensor,
|
|
download_url_to_bytesio,
|
|
)
|
|
import folder_paths
|
|
|
|
import os
|
|
import torch
|
|
from io import BytesIO
|
|
|
|
|
|
class SVG:
|
|
"""
|
|
Stores SVG representations via a list of BytesIO objects.
|
|
"""
|
|
def __init__(self, data: list[BytesIO]):
|
|
self.data = data
|
|
|
|
|
|
class SaveSVGNode:
|
|
"""
|
|
Save SVG files on disk.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.output_dir = folder_paths.get_output_directory()
|
|
self.type = "output"
|
|
self.prefix_append = ""
|
|
|
|
RETURN_TYPES = ()
|
|
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
|
|
FUNCTION = "save_svg"
|
|
CATEGORY = "api node/image/Recraft"
|
|
OUTPUT_NODE = True
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"svg": (RecraftIO.SVG,),
|
|
"filename_prefix": ("STRING", {"default": "svg/ComfyUI", "tooltip": "The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."})
|
|
},
|
|
"hidden": {
|
|
"prompt": "PROMPT",
|
|
"extra_pnginfo": "EXTRA_PNGINFO"
|
|
}
|
|
}
|
|
|
|
def save_svg(self, svg: SVG, filename_prefix="svg/ComfyUI", prompt=None, extra_pnginfo=None):
|
|
filename_prefix += self.prefix_append
|
|
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir)
|
|
results = list()
|
|
for batch_number, svg_bytes in enumerate(svg.data):
|
|
# NOTE: no way to do metadata for SVG right now, maybe figure this out later
|
|
# metadata = None
|
|
# if not args.disable_metadata:
|
|
# metadata = PngInfo()
|
|
# if prompt is not None:
|
|
# metadata.add_text("prompt", json.dumps(prompt))
|
|
# if extra_pnginfo is not None:
|
|
# for x in extra_pnginfo:
|
|
# metadata.add_text(x, json.dumps(extra_pnginfo[x]))
|
|
|
|
filename_with_batch_num = filename.replace("%batch_num%", str(batch_number))
|
|
file = f"{filename_with_batch_num}_{counter:05}_.svg"
|
|
with open(os.path.join(full_output_folder, file), 'wb') as svg_file:
|
|
svg_bytes.seek(0)
|
|
svg_file.write(svg_bytes.read())
|
|
results.append({
|
|
"filename": file,
|
|
"subfolder": subfolder,
|
|
"type": self.type
|
|
})
|
|
counter += 1
|
|
return { "ui": { "images": results } }
|
|
|
|
|
|
class RecraftColorRGBNode:
|
|
"""
|
|
Create Recraft Color by choosing specific RGB values.
|
|
"""
|
|
|
|
RETURN_TYPES = (RecraftIO.COLOR,)
|
|
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
|
|
RETURN_NAMES = ("recraft_color",)
|
|
FUNCTION = "create_color"
|
|
CATEGORY = "api node/image/Recraft"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"r": (IO.INT, {
|
|
"default": 0,
|
|
"min": 0,
|
|
"max": 255,
|
|
"tooltip": "Red value of color."
|
|
}),
|
|
"g": (IO.INT, {
|
|
"default": 0,
|
|
"min": 0,
|
|
"max": 255,
|
|
"tooltip": "Green value of color."
|
|
}),
|
|
"b": (IO.INT, {
|
|
"default": 0,
|
|
"min": 0,
|
|
"max": 255,
|
|
"tooltip": "Blue value of color."
|
|
}),
|
|
},
|
|
"optional": {
|
|
"recraft_color": (RecraftIO.COLOR,),
|
|
}
|
|
}
|
|
|
|
def create_color(self, r: int, g: int, b: int, recraft_color: RecraftColorChain=None):
|
|
recraft_color = recraft_color.clone() if recraft_color else RecraftColorChain()
|
|
recraft_color.add(RecraftColor(r, g, b))
|
|
return (recraft_color, )
|
|
|
|
|
|
class RecraftControlsNode:
|
|
"""
|
|
Create Recraft Controls for customizing Recraft generation.
|
|
"""
|
|
|
|
RETURN_TYPES = (RecraftIO.CONTROLS,)
|
|
RETURN_NAMES = ("recraft_controls",)
|
|
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
|
|
FUNCTION = "create_controls"
|
|
CATEGORY = "api node/image/Recraft"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
},
|
|
"optional": {
|
|
"colors": (RecraftIO.COLOR,),
|
|
"background_color": (RecraftIO.COLOR,),
|
|
}
|
|
}
|
|
|
|
def create_controls(self, colors: RecraftColorChain=None, background_color: RecraftColorChain=None):
|
|
return (RecraftControls(colors=colors, background_color=background_color), )
|
|
|
|
|
|
class RecraftStyleV3RealisticImageNode:
|
|
"""
|
|
Select realistic_image style and optional substyle.
|
|
"""
|
|
|
|
RETURN_TYPES = (RecraftIO.STYLEV3,)
|
|
RETURN_NAMES = ("recraft_style",)
|
|
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
|
|
FUNCTION = "create_style"
|
|
CATEGORY = "api node/image/Recraft"
|
|
|
|
RECRAFT_STYLE = RecraftStyleV3.realistic_image
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"substyle": (get_v3_substyles(s.RECRAFT_STYLE),),
|
|
}
|
|
}
|
|
|
|
def create_style(self, substyle: str):
|
|
if substyle == "None":
|
|
substyle = None
|
|
return (RecraftStyle(self.RECRAFT_STYLE, substyle),)
|
|
|
|
|
|
class RecraftStyleV3DigitalIllustrationNode(RecraftStyleV3RealisticImageNode):
|
|
"""
|
|
Select digital_illustration style and optional substyle.
|
|
"""
|
|
|
|
RECRAFT_STYLE = RecraftStyleV3.digital_illustration
|
|
|
|
|
|
class RecraftStyleV3VectorIllustrationNode(RecraftStyleV3RealisticImageNode):
|
|
"""
|
|
Select vector_illustration style and optional substyle.
|
|
"""
|
|
|
|
RECRAFT_STYLE = RecraftStyleV3.vector_illustration
|
|
|
|
|
|
class RecraftStyleV3LogoRasterNode(RecraftStyleV3RealisticImageNode):
|
|
"""
|
|
Select vector_illustration style and optional substyle.
|
|
"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"substyle": (get_v3_substyles(s.RECRAFT_STYLE, include_none=False),),
|
|
}
|
|
}
|
|
|
|
RECRAFT_STYLE = RecraftStyleV3.logo_raster
|
|
|
|
|
|
class RecraftStyleInfiniteStyleLibrary:
|
|
"""
|
|
Select style based on preexisting UUID from the Infinite Style Library.
|
|
"""
|
|
|
|
RETURN_TYPES = (RecraftIO.STYLEV3,)
|
|
RETURN_NAMES = ("recraft_style",)
|
|
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
|
|
FUNCTION = "create_style"
|
|
CATEGORY = "api node/image/Recraft"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"style_id": (IO.STRING, {
|
|
"default": "",
|
|
"tooltip": "UUID of style from Infinite Style Library.",
|
|
})
|
|
}
|
|
}
|
|
|
|
def create_style(self, style_id: str):
|
|
if not style_id:
|
|
raise Exception("The style_id input cannot be empty.")
|
|
return (RecraftStyle(style_id=style_id),)
|
|
|
|
|
|
class RecraftTextToImageNode:
|
|
"""
|
|
Generates images synchronously based on prompt and resolution.
|
|
"""
|
|
|
|
RETURN_TYPES = (IO.IMAGE,)
|
|
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
|
|
FUNCTION = "api_call"
|
|
API_NODE = True
|
|
CATEGORY = "api node/image/Recraft"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"prompt": (
|
|
IO.STRING,
|
|
{
|
|
"multiline": True,
|
|
"default": "",
|
|
"tooltip": "Prompt for the image generation.",
|
|
},
|
|
),
|
|
"size": (
|
|
[res.value for res in RecraftImageSize],
|
|
{
|
|
"default": RecraftImageSize.res_1024x1024,
|
|
"tooltip": "The size of the generated image.",
|
|
},
|
|
),
|
|
"n": (
|
|
IO.INT,
|
|
{
|
|
"default": 1,
|
|
"min": 1,
|
|
"max": 6,
|
|
"tooltip": "The number of images to generate.",
|
|
},
|
|
),
|
|
"seed": (
|
|
IO.INT,
|
|
{
|
|
"default": 0,
|
|
"min": 0,
|
|
"max": 0xFFFFFFFFFFFFFFFF,
|
|
"control_after_generate": True,
|
|
"tooltip": "Seed to determine if node should re-run; actual results are nondeterministic regardless of seed.",
|
|
},
|
|
),
|
|
},
|
|
"optional": {
|
|
"recraft_style": (RecraftIO.STYLEV3,),
|
|
"negative_prompt": (
|
|
IO.STRING,
|
|
{
|
|
"default": "",
|
|
"forceInput": True,
|
|
"tooltip": "An optional text description of undesired elements on an image.",
|
|
},
|
|
),
|
|
"recraft_controls": (
|
|
RecraftIO.CONTROLS,
|
|
{
|
|
"tooltip": "Optional additional controls over the generation via the Recraft Controls node."
|
|
},
|
|
),
|
|
},
|
|
"hidden": {
|
|
"auth_token": "AUTH_TOKEN_COMFY_ORG",
|
|
},
|
|
}
|
|
|
|
def api_call(
|
|
self,
|
|
prompt: str,
|
|
size: str,
|
|
n: int,
|
|
seed,
|
|
recraft_style: RecraftStyle = None,
|
|
negative_prompt: str = None,
|
|
recraft_controls: RecraftControls = None,
|
|
auth_token=None,
|
|
**kwargs,
|
|
):
|
|
default_style = RecraftStyle(RecraftStyleV3.realistic_image)
|
|
if recraft_style is None:
|
|
recraft_style = default_style
|
|
|
|
controls_api = None
|
|
if recraft_controls:
|
|
controls_api = recraft_controls.create_api_model()
|
|
|
|
if not negative_prompt:
|
|
negative_prompt = None
|
|
|
|
operation = SynchronousOperation(
|
|
endpoint=ApiEndpoint(
|
|
path="/proxy/recraft/image_generation",
|
|
method=HttpMethod.POST,
|
|
request_model=RecraftImageGenerationRequest,
|
|
response_model=RecraftImageGenerationResponse,
|
|
),
|
|
request=RecraftImageGenerationRequest(
|
|
prompt=prompt,
|
|
negative_prompts=negative_prompt,
|
|
model=RecraftModel.recraftv3,
|
|
size=size,
|
|
n=n,
|
|
style=recraft_style.style,
|
|
substyle=recraft_style.substyle,
|
|
style_id=recraft_style.style_id,
|
|
controls=controls_api,
|
|
),
|
|
auth_token=auth_token,
|
|
)
|
|
response: RecraftImageGenerationResponse = operation.execute()
|
|
images = []
|
|
for data in response.data:
|
|
image = bytesio_to_image_tensor(
|
|
download_url_to_bytesio(data.url, timeout=1024)
|
|
)
|
|
if len(image.shape) < 4:
|
|
image = image.unsqueeze(0)
|
|
images.append(image)
|
|
output_image = torch.cat(images, dim=0)
|
|
|
|
return (output_image,)
|
|
|
|
|
|
class RecraftTextToVectorNode:
|
|
"""
|
|
Generates SVG synchronously based on prompt and resolution.
|
|
"""
|
|
|
|
RETURN_TYPES = (RecraftIO.SVG,)
|
|
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
|
|
FUNCTION = "api_call"
|
|
API_NODE = True
|
|
CATEGORY = "api node/image/Recraft"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"prompt": (
|
|
IO.STRING,
|
|
{
|
|
"multiline": True,
|
|
"default": "",
|
|
"tooltip": "Prompt for the image generation.",
|
|
},
|
|
),
|
|
"substyle": (get_v3_substyles(RecraftStyleV3.vector_illustration),),
|
|
"size": (
|
|
[res.value for res in RecraftImageSize],
|
|
{
|
|
"default": RecraftImageSize.res_1024x1024,
|
|
"tooltip": "The size of the generated image.",
|
|
},
|
|
),
|
|
"n": (
|
|
IO.INT,
|
|
{
|
|
"default": 1,
|
|
"min": 1,
|
|
"max": 6,
|
|
"tooltip": "The number of images to generate.",
|
|
},
|
|
),
|
|
"seed": (
|
|
IO.INT,
|
|
{
|
|
"default": 0,
|
|
"min": 0,
|
|
"max": 0xFFFFFFFFFFFFFFFF,
|
|
"control_after_generate": True,
|
|
"tooltip": "Seed to determine if node should re-run; actual results are nondeterministic regardless of seed.",
|
|
},
|
|
),
|
|
},
|
|
"optional": {
|
|
"negative_prompt": (
|
|
IO.STRING,
|
|
{
|
|
"default": "",
|
|
"forceInput": True,
|
|
"tooltip": "An optional text description of undesired elements on an image.",
|
|
},
|
|
),
|
|
"recraft_controls": (
|
|
RecraftIO.CONTROLS,
|
|
{
|
|
"tooltip": "Optional additional controls over the generation via the Recraft Controls node."
|
|
},
|
|
),
|
|
},
|
|
"hidden": {
|
|
"auth_token": "AUTH_TOKEN_COMFY_ORG",
|
|
},
|
|
}
|
|
|
|
def api_call(
|
|
self,
|
|
prompt: str,
|
|
substyle: str,
|
|
size: str,
|
|
n: int,
|
|
seed,
|
|
negative_prompt: str = None,
|
|
recraft_controls: RecraftControls = None,
|
|
auth_token=None,
|
|
**kwargs,
|
|
):
|
|
# create RecraftStyle so strings will be formatted properly (i.e. "None" will become None)
|
|
recraft_style = RecraftStyle(RecraftStyleV3.vector_illustration, substyle=substyle)
|
|
|
|
controls_api = None
|
|
if recraft_controls:
|
|
controls_api = recraft_controls.create_api_model()
|
|
|
|
if not negative_prompt:
|
|
negative_prompt = None
|
|
|
|
operation = SynchronousOperation(
|
|
endpoint=ApiEndpoint(
|
|
path="/proxy/recraft/image_generation",
|
|
method=HttpMethod.POST,
|
|
request_model=RecraftImageGenerationRequest,
|
|
response_model=RecraftImageGenerationResponse,
|
|
),
|
|
request=RecraftImageGenerationRequest(
|
|
prompt=prompt,
|
|
negative_prompts=negative_prompt,
|
|
model=RecraftModel.recraftv3,
|
|
size=size,
|
|
n=n,
|
|
style=recraft_style.style,
|
|
substyle=recraft_style.substyle,
|
|
controls=controls_api,
|
|
),
|
|
auth_token=auth_token,
|
|
)
|
|
response: RecraftImageGenerationResponse = operation.execute()
|
|
svg_data = []
|
|
for data in response.data:
|
|
svg_data.append(download_url_to_bytesio(data.url, timeout=1024))
|
|
|
|
return (SVG(svg_data),)
|
|
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
# NOTE: names should be globally unique
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RecraftTextToImageNode": RecraftTextToImageNode,
|
|
"RecraftTextToVectorNode": RecraftTextToVectorNode,
|
|
"RecraftStyleV3RealisticImage": RecraftStyleV3RealisticImageNode,
|
|
"RecraftStyleV3DigitalIllustration": RecraftStyleV3DigitalIllustrationNode,
|
|
"RecraftStyleV3LogoRaster": RecraftStyleV3LogoRasterNode,
|
|
"RecraftStyleV3InfiniteStyleLibrary": RecraftStyleInfiniteStyleLibrary,
|
|
"RecraftColorRGB": RecraftColorRGBNode,
|
|
"RecraftControls": RecraftControlsNode,
|
|
"SaveSVG": SaveSVGNode,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RecraftTextToImageNode": "Recraft Text to Image",
|
|
"RecraftTextToVectorNode": "Recraft Text to Vector",
|
|
"RecraftStyleV3RealisticImage": "Recraft Style - Realistic Image",
|
|
"RecraftStyleV3DigitalIllustration": "Recraft Style - Digital Illustration",
|
|
"RecraftStyleV3LogoRaster": "Recraft Style - Logo Raster",
|
|
"RecraftStyleV3InfiniteStyleLibrary": "Recraft Style - Infinite Style Library",
|
|
"RecraftColorRGB": "Recraft Color RGB",
|
|
"RecraftControls": "Recraft Controls",
|
|
"SaveSVG": "Save SVG",
|
|
}
|