88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
from .utils import runwareUtils as rwUtils
|
|
|
|
|
|
class imageInferenceInputs:
|
|
"""Image Inference Inputs node for configuring image generation inputs"""
|
|
|
|
MAX_REFERENCE_IMAGES = 14
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
optionalInputs = {
|
|
"image": ("IMAGE", {
|
|
"tooltip": "Specifies an image input for the inference process."
|
|
}),
|
|
}
|
|
|
|
for i in range(1, cls.MAX_REFERENCE_IMAGES + 1):
|
|
ordinal = rwUtils.getOrdinal(i)
|
|
optionalInputs[f"Reference Image {i}"] = ("IMAGE", {
|
|
"tooltip": f"Specifies {ordinal.capitalize()} Reference Image for the inputs. These reference images help guide the image generation process."
|
|
})
|
|
optionalInputs[f"Reference Tag {i}"] = ("STRING", {
|
|
"tooltip": f"Optional tag describing {ordinal.capitalize()} Reference Image. Leave empty to omit.",
|
|
"default": "",
|
|
})
|
|
|
|
return {
|
|
"required": {},
|
|
"optional": optionalInputs
|
|
}
|
|
|
|
DESCRIPTION = "Configure custom inputs for Runware Image Inference, including reference images that can be passed to the inference node."
|
|
FUNCTION = "createInputs"
|
|
RETURN_TYPES = ("RUNWAREIMAGEINFERENCEINPUTS",)
|
|
RETURN_NAMES = ("Inference Inputs",)
|
|
CATEGORY = "Runware"
|
|
|
|
def createInputs(self, **kwargs):
|
|
"""Create image inference inputs from provided parameters"""
|
|
image = kwargs.get("image", None)
|
|
|
|
inputs = {}
|
|
|
|
if image is not None:
|
|
inputs["image"] = rwUtils.convertTensor2IMG(image)
|
|
|
|
references = self._collectReferences(kwargs)
|
|
if len(references) > 0:
|
|
inputs["referenceImages"] = references
|
|
|
|
return (inputs,)
|
|
|
|
def _collectReferences(self, kwargs):
|
|
"""Collect and convert reference images to list"""
|
|
referencePairs = []
|
|
|
|
# Collect all reference image and tag pairs
|
|
for i in range(1, self.MAX_REFERENCE_IMAGES + 1):
|
|
image = kwargs.get(f"Reference Image {i}", None)
|
|
tag = kwargs.get(f"Reference Tag {i}", "")
|
|
if image is not None:
|
|
referencePairs.append((image, tag))
|
|
|
|
if not referencePairs:
|
|
return []
|
|
|
|
# Determine if any tag is provided
|
|
hasTags = any(
|
|
isinstance(tag, str) and tag.strip() != ""
|
|
for _, tag in referencePairs
|
|
)
|
|
|
|
if not hasTags:
|
|
# Use legacy behavior: simple list of images
|
|
return [
|
|
rwUtils.convertTensor2IMG(image)
|
|
for image, _ in referencePairs
|
|
]
|
|
|
|
references = []
|
|
for image, tag in referencePairs:
|
|
entry = {"image": rwUtils.convertTensor2IMG(image)}
|
|
if isinstance(tag, str) and tag.strip() != "":
|
|
entry["tag"] = tag.strip()
|
|
references.append(entry)
|
|
|
|
return references
|