From 6092ab77939e1dbc863a572a773bcfa139f41203 Mon Sep 17 00:00:00 2001
From: cd0304 <yuanfanglin@joyutech.com>
Date: Wed, 17 Jul 2024 13:17:40 +0800
Subject: [PATCH 1/2] Update ImageNode.py

---
 nodes/ImageNode.py | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/nodes/ImageNode.py b/nodes/ImageNode.py
index da862dc0..f02e6316 100644
--- a/nodes/ImageNode.py
+++ b/nodes/ImageNode.py
@@ -30,17 +30,21 @@ def opencv_to_pil(image):
     return pil_image
 
 # 列出目录下面的所有文件
-def get_files_with_extension(directory, extension):
+def get_files_with_extensions(directory, extensions):
     file_list = []
+    # 确保extensions参数是一个list，即使只有一个元素
+    if not isinstance(extensions, (tuple, list)):
+        extensions = [extensions]
     for root, dirs, files in os.walk(directory):
         for file in files:
-            if file.endswith(extension):
-                file = os.path.splitext(file)[0]
-                file_path = os.path.join(root, file)
+            # 检查文件是否以任何一个提供的扩展名结尾
+            if any(file.endswith(ext) for ext in extensions):
+                file_without_ext = os.path.splitext(file)[0]
+                file_path = os.path.join(root, file_without_ext)
                 file_name = os.path.relpath(file_path, directory)
                 file_list.append(file_name)
     return file_list
-
+	
 def composite_images(foreground, background, mask, is_multiply_blend=False, position="overall", scale=0.25):
     width, height = foreground.size
     bg_image = background
@@ -1498,7 +1502,7 @@ def INPUT_TYPES(s):
         return {"required": { 
             
                     "text": ("STRING",{"multiline": True,"default": "龍馬精神迎新歲","dynamicPrompts": False}),
-                    "font": (get_files_with_extension(FONT_PATH,'.ttf'),),#后缀为 ttf
+                    "font": (get_files_with_extension(FONT_PATH,['.ttf','.otf']),),#后缀为 ttf
                     "font_size": ("INT",{
                                 "default":100, 
                                 "min": 100, #Minimum value
@@ -3201,4 +3205,4 @@ def run(self, images):
 
         out = torch.cat(out, dim=0)
 
-        return (out,)
\ No newline at end of file
+        return (out,)

From 0efa4c38c038ac1e330ab1a8b9f0fe2e79ff7ef8 Mon Sep 17 00:00:00 2001
From: cd0304 <yuanfanglin@joyutech.com>
Date: Wed, 17 Jul 2024 13:59:22 +0800
Subject: [PATCH 2/2] Update ImageNode.py

---
 nodes/ImageNode.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/nodes/ImageNode.py b/nodes/ImageNode.py
index f02e6316..98cd0530 100644
--- a/nodes/ImageNode.py
+++ b/nodes/ImageNode.py
@@ -30,19 +30,18 @@ def opencv_to_pil(image):
     return pil_image
 
 # 列出目录下面的所有文件
-def get_files_with_extensions(directory, extensions):
+def get_files_with_extension(directory, extensions):
     file_list = []
     # 确保extensions参数是一个list，即使只有一个元素
     if not isinstance(extensions, (tuple, list)):
         extensions = [extensions]
     for root, dirs, files in os.walk(directory):
+        # print(f"Files at {root}: {files}")  # 确认files是一个字符串列表
         for file in files:
             # 检查文件是否以任何一个提供的扩展名结尾
             if any(file.endswith(ext) for ext in extensions):
-                file_without_ext = os.path.splitext(file)[0]
-                file_path = os.path.join(root, file_without_ext)
-                file_name = os.path.relpath(file_path, directory)
-                file_list.append(file_name)
+                # 直接将文件名添加到列表中
+                file_list.append(file)
     return file_list
 	
 def composite_images(foreground, background, mask, is_multiply_blend=False, position="overall", scale=0.25):
@@ -1542,7 +1541,7 @@ def INPUT_TYPES(s):
 
     def run(self,text,font,font_size,spacing,padding,text_color,vertical,stroke):
         
-        font_path=os.path.join(FONT_PATH,font+'.ttf')
+        font_path=os.path.join(FONT_PATH,font)
 
         if text=="":
             text=" "