Bjornulf_custom_nodes/write_image_allinone.py

class WriteImageAllInOne:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "art_style": (["none", "drawing", "digital art", "photography"],),
                "location": ("STRING", {"forceInput": "True"}),
                "lighting": ("STRING", {"multiline": True}),
                "camera_angle": ("STRING", {"multiline": True}),
            },
            "optional": {
                "other": ("STRING", {"multiline": True},),
            }
        }

    RETURN_TYPES = ("STRING",)
    RETURN_NAMES = ("text",)
    FUNCTION = "write_image_allinone"
    OUTPUT_NODE = True
    CATEGORY = "Bjornulf"

    def write_image_allinone(self, art_style, location, lighting, camera_angle, other=""):
        text = f"Art Style: {art_style}\n\n"
        text += f"Location:\n{location}\n\n"
        text += f"Lighting:\n{lighting}\n\n"
        text += f"Camera Angle:\n{camera_angle}\n\n"
        if other:
            text += f"Other:\n{other}\n\n"
        return (text,)

    @classmethod
    def CREATE_CONNECTED_NODES(cls):
        return [
            {
                "class_type": "PrimitiveNode",
                "inputs": {
                    "value": "drawing",
                    "label": "Art Style",
                    "type": "combo",
                    "options": ["drawing", "digital art", "photography"]
                },
                "output": ["art_style"]
            },
            {
                "class_type": "PrimitiveNode",
                "inputs": {
                    "value": "",
                    "label": "Location",
                    "type": "text",
                    "multiline": True
                },
                "output": ["location"]
            },
            {
                "class_type": "PrimitiveNode",
                "inputs": {
                    "value": "",
                    "label": "Lighting",
                    "type": "text",
                    "multiline": True
                },
                "output": ["lighting"]
                },
            {
                "class_type": "PrimitiveNode",
                "inputs": {
                    "value": "",
                    "label": "Camera Angle",
                    "type": "text",
                    "multiline": True
                },
                "output": ["camera_angle"]
            },
            {
                "class_type": "PrimitiveNode",
                "inputs": {
                    "value": "",
                    "label": "Other",
                    "type": "text",
                    "multiline": True
                },
                "output": ["other"]
            }
        ]