Merge pull request #3 from pamparamm/master

Fix #2 and add more primitive nodes

Author: theUpsider

README.md

@@ -3,7 +3,7 @@
 This repository contains an extension to [ComfyUI](https://github.com/comfyanonymous/ComfyUI) that introduces logic nodes and conditional rendering capabilities:
 - If
 - Compare
-- Int, String
+- Int, String, Float, Bool
 
 ![image](https://github.com/theUpsider/ComfyUI-Logic/assets/25013640/7807b2a4-989d-4021-9572-1d2d13725304)
 > **_NOTE:_** This extension is still in development and may contain bugs. Please report any issues you encounter. New features are in development!
@@ -15,7 +15,7 @@ This repository contains an extension to [ComfyUI](https://github.com/comfyanony
 ## Features
 
 - **Comparison Nodes**: Compare two values using various comparison operators.
-- **Data Type Nodes**: Convert and handle `Int` and `String` data types.
+- **Data Type Nodes**: Convert and handle `Int`, `String`, `Float` and `Bool` data types.
 - **Conditional Execution**: Execute different nodes based on a boolean condition.
 - **Debugging**: Print any input to the console for debugging purposes.
 
@@ -40,6 +40,14 @@ Accepts an integer value and returns it.
 
 Accepts a string value and returns it.
 
+### Float
+
+Accepts a float value and returns it.
+
+### Bool
+
+Accepts a boolean value and returns it.
+
 ### If
 
 Executes the `IF_TRUE` node if the `ANY` input is `True`, otherwise it executes the `IF_FALSE` node.

nodes.py

@@ -32,7 +32,7 @@ def INPUT_TYPES(s):
     CATEGORY = "Logic"
 
     def execute(self, value):
-        return value
+        return (value,)
 
 
 class Int:
@@ -51,7 +51,45 @@ def INPUT_TYPES(s):
     CATEGORY = "Logic"
 
     def execute(self, value):
-        return value
+        return (value,)
+
+
+class Float:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {"value": ("FLOAT", {"default": 0, "step": 0.01})},
+        }
+
+    RETURN_TYPES = ("FLOAT",)
+
+    RETURN_NAMES = ("FLOAT",)
+
+    FUNCTION = "execute"
+
+    CATEGORY = "Logic"
+
+    def execute(self, value):
+        return (value,)
+
+
+class Bool:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {"value": ("BOOLEAN", {"default": False})},
+        }
+
+    RETURN_TYPES = ("BOOLEAN",)
+
+    RETURN_NAMES = ("BOOLEAN",)
+
+    FUNCTION = "execute"
+
+    CATEGORY = "Logic"
+
+    def execute(self, value):
+        return (value,)
 
 
 class Compare:
@@ -156,6 +194,8 @@ def log_input(self, ANY):
 NODE_CLASS_MAPPINGS = {
     "Compare": Compare,
     "Int": Int,
+    "Float": Float,
+    "Bool": Bool,
     "String": String,
     "If ANY execute A else B": IfExecute,
     "DebugPrint": DebugPrint,
@@ -165,6 +205,8 @@ def log_input(self, ANY):
 NODE_DISPLAY_NAME_MAPPINGS = {
     "Compare": "Compare",
     "Int": "Int",
+    "Float": "Float",
+    "Bool": "Bool",
     "String": "String",
     "If ANY execute A else B": "If",
     "DebugPrint": "DebugPrint",