From 9fbe2295371e7637c1125b932be20e3f3b398c58 Mon Sep 17 00:00:00 2001
From: Sam Khoze <68170403+SamKhoze@users.noreply.github.com>
Date: Sun, 30 Jun 2024 14:29:09 +0530
Subject: [PATCH] Update audio_playback.py

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

diff --git a/audio_playback.py b/audio_playback.py
index c618bd9..40293ec 100644
--- a/audio_playback.py
+++ b/audio_playback.py
@@ -44,31 +44,17 @@ class AudioData:
 
 os.makedirs(output_dir,exist_ok=True)
 os.makedirs(os.path.join(output_dir,"videos"),exist_ok=True)
-
 class SaveAudio:
-    def __init__(self):
-        
-        self.type = "output"
-
      
     @classmethod
     def INPUT_TYPES(s):
         
-        try:
-            shutil.rmtree(frames_output_dir)
-            os.mkdir(frames_output_dir)
-        except:
-            pass
-        
-
         #print(f"Temporary folder {frames_output_dir} has been emptied.")
         return {"required": 
                     {"audio": ("VHS_AUDIO", ),
-                     "METADATA": ("STRING",  {"default": ""}  ), 
                      "start_time": ([str(i) for i in range(10000)],),
                      "end_time": ([str(i) for i in range(10000)],),
                      },
-                "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
                 }
                 
 
@@ -80,8 +66,11 @@ class SaveAudio:
 
     CATEGORY = "DeepFuze"
 
-    def save_video(self, audio,METADATA,start_time,end_time,prompt=None, extra_pnginfo=None):
+    def save_video(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")
+        print(audio_path)
         outfile = os.path.join(audio_path,str(time.time()).replace(".","_")+".wav")
         open(file_path,"wb").write(audio())
         Fs, data = wavfile.read(file_path)
@@ -93,11 +82,9 @@ class SaveAudio:
         if int(end_time) > 0:
             subprocess.run(['ffmpeg','-i',file_path,'-ss',start_time,'-to',end_time,outfile])
             file_path = outfile
-        file_path_ = file_path.replace(".wav",".mp4")
-        print(file_path)
-        file_path_ = os.path.join(output_dir,"videos",file_path_.split('/')[-1])
-        os.system(f"ffmpeg -i {file_path} {file_path_}")
-        return {"ui": {"text": [file_path_.split("/")[-1]],}}
+        audio_name = file_path.split("/")[-1]
+        print(audio_name,"---",audio_root)
+        return {"ui": {"audio":[audio_name,audio_root]}}
 
 
 class PlayBackAudio: