update workflow

CROP_OBJECT/crop.py

@@ -37,8 +37,6 @@ class ObjectCrop:
 
         return (cropped, )
 
-# A dictionary that contains all nodes you want to export with their names
-# NOTE: names should be globally unique
 NODE_CLASS_MAPPINGS = {
     "ObjectCrop": ObjectCrop
 }

Custom_comfyui_llm_assistant/AssistantNodes.py

@@ -64,12 +64,12 @@ class GenerateStableDiffsutionPromptLLM:
                 "stop": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "response_pattern": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "temperature": ("FLOAT", {
                     "default": 1.0,
@@ -98,7 +98,7 @@ class GenerateStableDiffsutionPromptLLM:
 
     FUNCTION = "generateStableDiffsutionPrompt"
 
-    CATEGORY = "cropcropcropcrop"
+    CATEGORY = "LLM"
 
     def generateStableDiffsutionPrompt(self, object1, desc_obj1, object2, desc_obj2 ,template_system, template_user, stop, response_pattern,  temperature, max_tokens, model_name):
         config = load_config()
@@ -170,12 +170,12 @@ class TranslateTextLLM:
                 "stop": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "response_pattern": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "temperature": ("FLOAT", {
                     "default": 1.0,
@@ -274,12 +274,12 @@ class ChatWithLLM:
                 "stop": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "response_pattern": ("STRING", {
                     "dynamicPrompts": False,
                     "multiline": False,
-                    "default": None
+                    "default": ""
                 }),
                 "temperature": ("FLOAT", {
                     "default": 1.0,
@@ -329,12 +329,12 @@ class ChatWithLLM:
 
 NODE_CLASS_MAPPINGS = {
     "Generate Stable Diffsution Prompt With LLM duci": GenerateStableDiffsutionPromptLLM,
-    # "Translate Text With LLM": TranslateTextLLM,
-    # "Chat With LLM": ChatWithLLM,
+    "Translate Text With LLM": TranslateTextLLM,
+    "Chat With LLM": ChatWithLLM,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "Generate Stable Diffsution Prompt With LLM duci": "HUHU",
-    # "Translate Text With LLM": "Translate Text With LLM",
-    # "Chat With LLM": "Chat With LLM",
+    "Generate Stable Diffsution Prompt With LLM duci": "Custom Generate Stable Diffsution Prompt With LLM",
+    "Translate Text With LLM": "Translate Text With LLM",
+    "Chat With LLM": "Chat With LLM",
 }

README.md

@@ -51,7 +51,6 @@ _All the folders, except `CROP_OBJECT`, are from other repositories. However, I
 - **[comfyui-llm-assistant](https://github.com/longgui0318/comfyui-llm-assistant)**
   - Removed input field: `prompt`.
   - Added 4 input fields: `object1`, `desc_obj1`, `object2`, `desc_obj2`.
-  - Change all `default` value to `""` because the newest frontend of ComfyUI do not accept `None`
 
 ## 🤝 Contributing
 Contributions are welcome! Please open an issue or submit a pull request.

__init__.py

@@ -1,14 +1,7 @@
-# from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_EfficientSAM
-# from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_SEGS
+from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_EfficientSAM
+from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_SEGS
 from .Custom_comfyui_llm_assistant import AssistantNodes
 from .CROP_OBJECT import crop
 
-# NODE_CLASS_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_CLASS_MAPPINGS, **YOLO_WORLD_SEGS.NODE_CLASS_MAPPINGS, **AssistantNodes.NODE_CLASS_MAPPINGS, **crop.NODE_CLASS_MAPPINGS}
-# NODE_DISPLAY_NAME_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_DISPLAY_NAME_MAPPINGS, **YOLO_WORLD_SEGS.NODE_DISPLAY_NAME_MAPPINGS, **AssistantNodes.NODE_DISPLAY_NAME_MAPPINGS, **crop.NODE_DISPLAY_NAME_MAPPINGS}
-
-
-NODE_CLASS_MAPPINGS = {**AssistantNodes.NODE_CLASS_MAPPINGS, **crop.NODE_CLASS_MAPPINGS}
-NODE_DISPLAY_NAME_MAPPINGS = {**AssistantNodes.NODE_DISPLAY_NAME_MAPPINGS, **crop.NODE_DISPLAY_NAME_MAPPINGS}
-
-print(NODE_CLASS_MAPPINGS)
-print(NODE_DISPLAY_NAME_MAPPINGS)
+NODE_CLASS_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_CLASS_MAPPINGS, **YOLO_WORLD_SEGS.NODE_CLASS_MAPPINGS, **AssistantNodes.NODE_CLASS_MAPPINGS, **crop.NODE_CLASS_MAPPINGS}
+NODE_DISPLAY_NAME_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_DISPLAY_NAME_MAPPINGS, **YOLO_WORLD_SEGS.NODE_DISPLAY_NAME_MAPPINGS, **AssistantNodes.NODE_DISPLAY_NAME_MAPPINGS, **crop.NODE_DISPLAY_NAME_MAPPINGS}