From 43ce2935c27fb075cecf9beeb72de512018e55bd Mon Sep 17 00:00:00 2001 From: XchanBik Date: Sat, 10 May 2025 17:20:33 +0200 Subject: [PATCH] Update Text.py --- Text.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/Text.py b/Text.py index 897bc3f..2c3dd6a 100644 --- a/Text.py +++ b/Text.py @@ -3,12 +3,19 @@ cat = "XchanBik/Text" import os import folder_paths +def filter_files_by_extension(files: list[str], extension: str) -> list[str]: + # Call the existing function + filename_list, folders_all, timestamp = folder_paths.get_filename_list_("loras") + + # Filter the result by extension outside the base code + return [file for file in filename_list if file.endswith(str)] + class LoraTextLoader: @classmethod def INPUT_TYPES(cls): return { "required": { - "text_file": (['None'] + folder_paths.get_filename_list("loras", extension=".txt")), # Load .txt files + "text_file": (['None'] + filter_files_by_extension("loras", extension=".txt")), # Load .txt files } }