From 7ebb7d4caa83266162c754654bd69951c9d8d170 Mon Sep 17 00:00:00 2001
From: Sam Khoze <68170403+SamKhoze@users.noreply.github.com>
Date: Fri, 28 Jun 2024 23:52:47 +0530
Subject: [PATCH] VHS support added

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

diff --git a/audio_playback.py b/audio_playback.py
index bfc689b..9997ca5 100644
--- a/audio_playback.py
+++ b/audio_playback.py
@@ -1,5 +1,5 @@
 import sounddevice
-
+from io import BytesIO
 import os
 import time
 import folder_paths
@@ -38,7 +38,7 @@ class SaveAudio:
 
         #print(f"Temporary folder {frames_output_dir} has been emptied.")
         return {"required": 
-                    {"audio": ("AUDIO", ),
+                    {"audio": ("VHS_AUDIO", ),
                      "METADATA": ("STRING",  {"default": ""}  ), 
                      "start_time": ([str(i) for i in range(10000)],),
                      "end_time": ([str(i) for i in range(10000)],),
@@ -58,7 +58,7 @@ class SaveAudio:
     def save_video(self, audio,METADATA,start_time,end_time,prompt=None, extra_pnginfo=None):
         file_path = os.path.join(audio_path,str(time.time()).replace(".","")+".wav")
         outfile = os.path.join(audio_path,str(time.time()).replace(".","_")+".wav")
-        write(file_path,audio.sample_rate,audio.audio_data)
+        open(file_path,"wb").write(audio())
         Fs, data = wavfile.read(file_path)
         n = data.size
         t = n / Fs
@@ -81,7 +81,7 @@ class PlayBackAudio:
     def INPUT_TYPES(self):
         return {
             "required":{
-                "audio": ("AUDIO",)
+                "audio": ("VHS_AUDIO",)
             }
         }
     OUTPUT_NODE = True
@@ -91,5 +91,8 @@ class PlayBackAudio:
     FUNCTION = "play_audio"
 
     def play_audio(self,audio):
+		file = BytesIO(audio())
+        audio_file = AudioSegment.from_file(file, format="wav")
+        audio = AudioData(audio_file)
         sounddevice.play(audio.audio_data,audio.sample_rate)
         return ()