added preview widgets

custom_nodes/output.py

@@ -1,10 +1,13 @@
 import os
+import shutil
 
 from ..lib.audio import SUPPORTED_AUDIO, audio_to_bytes, bytes_to_audio, load_input_audio, save_input_audio
 from ..lib.utils import get_hash
 from .utils import increment_filename_no_overwrite
 import folder_paths
   
+temp_path = folder_paths.get_temp_directory()
+
 class PreviewAudio:
     def __init__(self):
         pass
@@ -18,6 +21,7 @@ class PreviewAudio:
                 "save_format": (SUPPORTED_AUDIO,{"default": "flac"},),
                 "save_channels": ([1,2],{"default": 1}),
                 "overwrite_existing": ("BOOLEAN", {"default": True}),
+                "autoplay": ("BOOLEAN", {"default": True}),
             },
         }
 
@@ -30,7 +34,7 @@ class PreviewAudio:
 
     FUNCTION = "save_audio"
 
-    def save_audio(self, audio, filename, save_format, save_channels, overwrite_existing):
+    def save_audio(self, audio, filename, save_format, save_channels, overwrite_existing, autoplay):
 
         filename = filename.strip()
         assert filename, "Filename cannot be empty"
@@ -44,10 +48,19 @@ class PreviewAudio:
         if os.path.isfile(output_path) and not overwrite_existing:
             output_path = increment_filename_no_overwrite(output_path)
         
+        
         input_audio = bytes_to_audio(audio())
         print(save_input_audio(output_path,input_audio,to_int16=True,to_stereo=save_channels==2))
 
-        return output_path, lambda:audio_to_bytes(*input_audio)
+        tempdir = os.path.join(temp_path,"preview")
+        os.makedirs(tempdir, exist_ok=True)
+        widgetId = get_hash(output_path,save_channels)
+        audio_name = f"{widgetId}.{save_format}"
+        preview_file = os.path.join(tempdir,audio_name)
+        if not os.path.isfile(preview_file): shutil.copyfile(output_path,preview_file)
+        return {"ui": {"preview": [{
+            "filename": audio_name, "type": "temp", "subfolder": "preview", "widgetId": widgetId, "autoplay": autoplay
+            }]}, "result": (output_path, lambda:audio_to_bytes(*input_audio))}
     
     @classmethod
     def IS_CHANGED(cls, audio):