From 50bcb5e18ce256ce1afa601c0c3d31852064af5e Mon Sep 17 00:00:00 2001
From: Sam Khoze <68170403+SamKhoze@users.noreply.github.com>
Date: Tue, 2 Jul 2024 00:40:04 +0530
Subject: [PATCH] added VHS audio output support

---
 audio_playback.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/audio_playback.py b/audio_playback.py
index 40293ec..151c31b 100644
--- a/audio_playback.py
+++ b/audio_playback.py
@@ -9,7 +9,7 @@ import subprocess
 import sounddevice
 import numpy as np
 from pydub import AudioSegment
-
+from .utils import get_audio
 audio_path = os.path.join(folder_paths.get_input_directory(),"audio")
 
 
@@ -58,15 +58,15 @@ class SaveAudio:
                 }
                 
 
-    RETURN_NAMES = ()
-    RETURN_TYPES = ()
-    FUNCTION = "save_video"
+    RETURN_NAMES = ("AUDIO",)
+    RETURN_TYPES = ("VHS_AUDIO",)
+    FUNCTION = "save_audio"
 
     OUTPUT_NODE = True
 
     CATEGORY = "DeepFuze"
 
-    def save_video(self, audio,start_time,end_time):
+    def save_audio(self, audio,start_time,end_time):
         audio_path = folder_paths.get_input_directory()
         audio_root = os.path.basename(audio_path)
         file_path = os.path.join(audio_path,str(time.time()).replace(".","")+".wav")
@@ -83,8 +83,8 @@ class SaveAudio:
             subprocess.run(['ffmpeg','-i',file_path,'-ss',start_time,'-to',end_time,outfile])
             file_path = outfile
         audio_name = file_path.split("/")[-1]
-        print(audio_name,"---",audio_root)
-        return {"ui": {"audio":[audio_name,audio_root]}}
+        audio = get_audio(file_path)
+        return {"ui": {"audio":[audio_name,audio_root]},"result" : (lambda : audio,)}
 
 
 class PlayBackAudio: