From 9c25f65bca3e71e442d1b797de5984dd7e2182c3 Mon Sep 17 00:00:00 2001
From: aven <dengybouo@gmail.com>
Date: Fri, 13 Jun 2025 13:47:19 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BF=9D=E5=AD=98=E7=BB=93?=
 =?UTF-8?q?=E7=82=B9=E4=BD=BF=E5=85=B6=E4=B8=8Ecomfyui=E5=AE=98=E6=96=B9?=
 =?UTF-8?q?=E4=BF=9D=E5=AD=98=E6=96=B9=E5=BC=8F=E4=B8=80=E8=87=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes.py | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/nodes.py b/nodes.py
index 895fb19..2ef5f63 100644
--- a/nodes.py
+++ b/nodes.py
@@ -7,6 +7,7 @@ import torchvision.utils as vutils
 import torch
 import json
 import uuid
+from comfy.comfy_types import FileLocator
 
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
@@ -215,12 +216,21 @@ class AVSaveSubtitles:
     DESCRIPTION = "Saves the subtitles to a file."
 
     def save_subtitles(self, subtitles, filename_prefix="subtitles"):
-        filename = f"{filename_prefix}-{time.time()}.srt"
-        save_path = os.path.join(self.output_dir, filename)
-        with open(save_path, "w", encoding='utf-8') as f:
-            f.write(subtitles)
+        filename_prefix += self.prefix_append
+        full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir)
+        results: list[FileLocator] = []
 
-        return { "ui": { "text": save_path } }
+        file = f"{filename}_{counter:05}_.srt"
+        with open(os.path.join(full_output_folder, file), 'w', encoding='utf-8') as f:
+            f.write(subtitles)
+        results.append({
+            "filename": file,
+            "subfolder": subfolder,
+            "type": self.type
+        })
+        counter += 1
+
+        return { "ui": { "subtitles": results } }