FlowState-Creator-Nodes/FlowState_Chef_Ingredients.py at main · flowstatelabs/FlowState-Creator-Nodes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
# Project: FlowState Chef Ingredients
# Description: Input your ingredients for the FlowState Chef node.
# Version: 0.0.1
# Author: Johnathan Chivington
# Contact: flowstateeng@gmail.com | youtube.com/@flowstateeng


##
# SYSTEM STATUS
##
print(f'\t - 🟢 🥗 Loaded FlowState Chef Ingredients.')


##
# FS IMPORTS
##
from .FS_Types import *
from .FlowState_Node import FlowState_Node


##
# OUTSIDE IMPORTS
##
from comfy_extras.nodes_images import ImageStitch


##
# NODES
##
class FlowState_Chef_Ingredients(FlowState_Node):
    CATEGORY = 'FlowState Creator Suite/Alteration'
    DESCRIPTION = '👨‍🍳 Input your ingredients for the FlowState Chef node.'
    FUNCTION = 'execute'
    RETURN_TYPES = (TYPE_ANY,)
    RETURN_NAMES = ('chef_ingredients',)
    OUTPUT_TOOLTIPS = ('Ingredients to feed into the FlowState Chef node.',)

    def __init__(self):
        super().__init__('🌊🥗 FlowState Chef Ingredients')

    @classmethod
    def INPUT_TYPES(s):
        return {
            'required': {
                # PROMPT SETTINGS
                'qwen_prompt_label': TYPE_FLOWSTATE_LABEL_PROMPT_CHEF_QWEN,
                'qwen_edit_prompt': TYPE_PROMPT_CHEF_QWEN,
                'flux_prompt_label': TYPE_FLOWSTATE_LABEL_PROMPT_CHEF_FLUX,
                'flux_refinement_prompt': TYPE_PROMPT_CHEF_FLUX,
                'image': TYPE_CHEF_IMAGE_IN
            },
            'optional': {
                'image_2': TYPE_CHEF_IMAGE_IN,
                'image_3': TYPE_CHEF_IMAGE_IN,
                'image_4': TYPE_CHEF_IMAGE_IN,
            }
        }

    def stitch_images(self):
        images = [
            self.params['image'],
            self.params['image_2'],
            self.params['image_3'],
            self.params['image_4']
        ]

        filtered = [image for image in images if image is not None]

        if not filtered:
            raise ValueError(self.format_value_error('ingredients_img'))

        self.stitched = filtered[0]

        for image in filtered[1:]:
            self.stitched = ImageStitch().stitch(
                image1=self.stitched,
                direction='right',
                match_image_size='true',
                spacing_width=0,
                spacing_color='white',
                image2=image
            )[0]

    def execute(self,
        qwen_prompt_label, qwen_edit_prompt, flux_prompt_label, flux_refinement_prompt,
        image, image_2=None, image_3=None, image_4=None
    ):

        self.print_status([('Preparing Chef Ingredients...',)], init=True)

        self.params = locals()
        self.stitch_images()

        self.print_status([('Ingredients Prepared.',)], end=True)

        return ({
            'qwen': qwen_edit_prompt,
            'flux': flux_refinement_prompt,
            'img': self.stitched
        }, )