48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
class MultiPromptNode:
|
|
"""Simple shim that exposes four multiline prompt sockets."""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"subject_mask": (
|
|
"STRING",
|
|
{"default": "", "multiline": True, "placeholder": "subject mask"},
|
|
),
|
|
"inpaint_prompt": (
|
|
"STRING",
|
|
{"default": "", "multiline": True, "placeholder": "inpaint prompt"},
|
|
),
|
|
"outpaint_prompt": (
|
|
"STRING",
|
|
{"default": "", "multiline": True, "placeholder": "outpaint prompt"},
|
|
),
|
|
"stitch_mask": (
|
|
"STRING",
|
|
{"default": "", "multiline": True, "placeholder": "stitch mask"},
|
|
),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING", "STRING", "STRING", "STRING")
|
|
RETURN_NAMES = ("subject_mask", "inpaint_prompt", "outpaint_prompt", "stitch_mask")
|
|
FUNCTION = "apply"
|
|
CATEGORY = "config"
|
|
|
|
def apply(self, subject_mask, inpaint_prompt, outpaint_prompt, stitch_mask):
|
|
return (
|
|
str(subject_mask),
|
|
str(inpaint_prompt),
|
|
str(outpaint_prompt),
|
|
str(stitch_mask),
|
|
)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"MultiPromptNode": MultiPromptNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"MultiPromptNode": "Multi-Prompt",
|
|
}
|