0.77

load_text.py

@@ -1,56 +1,72 @@
 import os
+from server import PromptServer
+import aiohttp.web as web
 
 class LoadTextFromFolder:
+    default_dir = "Bjornulf/Text"
+    
     @classmethod
     def INPUT_TYPES(cls):
         """Define input parameters for the node"""
-        default_dir = "Bjornulf/Text"
-        available_files = []
-        
-        if os.path.exists(default_dir):
-            available_files = [f for f in os.listdir(default_dir) 
-                             if f.lower().endswith('.txt')]
-            
+        available_files = cls.get_available_files()
         if not available_files:
             available_files = ["no_files_found"]
-        
         return {
             "required": {
                 "text_file": (available_files, {"default": available_files[0]}),
             }
         }
 
+    @classmethod
+    def get_available_files(cls):
+        """Get list of .txt files recursively from the default directory"""
+        available_files = []
+        if os.path.exists(cls.default_dir):
+            for root, dirs, files in os.walk(cls.default_dir):
+                for file in files:
+                    if file.lower().endswith('.txt'):
+                        full_path = os.path.join(root, file)
+                        rel_path = os.path.relpath(full_path, cls.default_dir)
+                        available_files.append(rel_path)
+            available_files.sort()
+        return available_files
+
     RETURN_TYPES = ("STRING", "STRING", "STRING")
     RETURN_NAMES = ("text", "filename", "full_path")
     FUNCTION = "load_text"
     CATEGORY = "Bjornulf"
 
     def load_text(self, text_file):
+        """Load text from the selected file"""
         try:
             if text_file == "no_files_found":
                 raise ValueError("No text files found in Bjornulf/Text folder")
-                
-            filepath = os.path.join("Bjornulf/Text", text_file)
-                
-            # Check if file exists
+            filepath = os.path.join(self.default_dir, text_file)
             if not os.path.exists(filepath):
                 raise ValueError(f"File not found: {filepath}")
-                
-            # Get absolute path
             full_path = os.path.abspath(filepath)
-            
-            # Get just the filename
             filename = os.path.basename(filepath)
-            
-            # Read text from file
             with open(filepath, 'r', encoding='utf-8') as file:
                 text = file.read()
-                
             return (text, filename, full_path)
-            
         except (OSError, IOError) as e:
             raise ValueError(f"Error loading file: {str(e)}")
 
+@PromptServer.instance.routes.post("/get_text_files")
+async def get_text_files(request):
+    try:
+        available_files = LoadTextFromFolder.get_available_files()
+        return web.json_response({
+            "success": True,
+            "files": available_files
+        }, status=200)
+    except Exception as e:
+        error_msg = str(e)
+        return web.json_response({
+            "success": False,
+            "error": error_msg
+        }, status=500)
+
 class LoadTextFromPath:
     @classmethod
     def INPUT_TYPES(cls):