diff --git a/__init__.py b/__init__.py index da996ed..6f3dc85 100644 --- a/__init__.py +++ b/__init__.py @@ -25,7 +25,7 @@ from custom_nodes.EternalKernelLiteGraphNodes.image import Quantize from custom_nodes.EternalKernelLiteGraphNodes.image import SelectFromRGBSimilarity from custom_nodes.EternalKernelLiteGraphNodes.image import LoadImage from custom_nodes.EternalKernelLiteGraphNodes.image import PromptTemplate -from custom_nodes.EternalKernelLiteGraphNodes.image import ExecWidget +from custom_nodes.EternalKernelLiteGraphNodes.image import CodeExecWidget from custom_nodes.EternalKernelLiteGraphNodes.image import ImageDistanceMask from custom_nodes.EternalKernelLiteGraphNodes.image import TextRender @@ -48,7 +48,7 @@ if use_generic_nodes: "Quantize": Quantize, "Select From RGB Similarity": SelectFromRGBSimilarity, "Load Layered Image": LoadImage, - "Eval Widget": ExecWidget, + "Code Execution Widget": CodeExecWidget, "Image Distance Mask": ImageDistanceMask, "Image Text": TextRender, @@ -76,7 +76,7 @@ else: "Select From RGB Similarity": SelectFromRGBSimilarity, "Load Layered Image": LoadImage, "Prompt Template": PromptTemplate, - "Eval Widget": ExecWidget, + "Code Execution Widget": CodeExecWidget, "Tiny Txt 2 Img": TinyTxtToImg, "Image Distance Mask": ImageDistanceMask, "Preview Image Test": PreviewImageTest, diff --git a/image.py b/image.py index ee7bcaa..149d631 100644 --- a/image.py +++ b/image.py @@ -310,7 +310,7 @@ class PreviewImageTest(SaveImage): return my_ret -class ExecWidget: +class CodeExecWidget: """runs eval on the given text""" @classmethod @@ -338,7 +338,7 @@ class ExecWidget: } } - CATEGORY = "text" + CATEGORY = "image" RETURN_TYPES = ("STRING", "IMAGE", "FLOAT", "INT", "TUPLE", "CODE") FUNCTION = "pre_exec_handler" INTERNAL_STATE_DISPLAY_CODE = True @@ -366,7 +366,7 @@ class ExecWidget: def exec_handler(self, text_to_eval, image_obj: torch.Tensor = None, float_obj: float = 0.0, string_obj: str = "", int_obj: int = 0, tuple_obj: tuple = None, name: str = "exec_func"): """ - >>> ExecWidget().exec_handler("2 + 3") + >>> CodeExecWidget().exec_handler("2 + 3") '5' """