Initial working version

2025-04-01 23:29:05 +01:00
parent 6982e4c598
commit bc0a12c50f
5 changed files with 250 additions and 0 deletions
@@ -0,0 +1,54 @@
+import os
+import random
+from typing import List
+from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+class TT_TextFileSelectorNode:
+    @classmethod
+    def INPUT_TYPES(cls):
+        """
+        Defines the input types for the node. The user only provides the 'file_path' and 'seed'.
+        No 'selection' input is provided.
+        """
+        file_path = os.path.join(os.path.dirname(__file__), "DefaultList.txt")  # Relative path to the .txt file
+        return {
+            "required": {
+                "seed": ("INT", {"default": 42}),
+                "file_path": ("STRING", {"default": file_path}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)  # The output will be a string
+    FUNCTION = "output_selection"  # Function that will process the data
+    CATEGORY = "TT_TextTools"  # Category for organization in the UI
+
+    @staticmethod
+    def load_options(file_path: str = "DefaultList.txt") -> List[str]:
+        """
+        Loads options from a specified .txt file and forces reload on every call.
+        """
+        # Check if the file exists and is not empty
+        if not os.path.exists(file_path) or os.path.getsize(file_path) == 0:
+            return ["No options available"]  # Default message if file is empty or missing
+        
+        # Open and read the file fresh every time
+        with open(file_path, "r", encoding="utf-8") as f:
+            return [line.strip() for line in f.readlines() if line.strip()]
+
+    def output_selection(self, seed: int, file_path: str):
+        """
+        Processes the selection and returns either a random choice from the file or a specific choice.
+        """
+        options = self.load_options(file_path)  # Force reload of the options from the file
+        
+        if options == ["No options available"]:
+            return ("No options available",)
+        
+        random.seed(seed)  # Set the random seed for reproducibility
+        
+        # Return a random choice from the options
+        return (random.choice(options),)
+
+# Register the new node class
+NODE_CLASS_MAPPINGS["TT_TextFileSelectorNode"] = TT_TextFileSelectorNode
+NODE_DISPLAY_NAME_MAPPINGS["TT_TextFileSelectorNode"] = "Text File Selector Node"  # Friendly name for UI