Reformat audio after downloading

modules/utils/youtube_manager.py

@@ -1,4 +1,5 @@
 from pytubefix import YouTube
+import subprocess
 import os
 
 
@@ -12,4 +13,21 @@ def get_ytmetas(link):
 
 
 def get_ytaudio(ytdata: YouTube):
-    return ytdata.streams.get_audio_only().download(filename=os.path.join("modules", "yt_tmp.wav"))
+    # Somehow the audio is corrupted so need to convert to valid audio file.
+    # Fix for : https://github.com/jhj0517/Whisper-WebUI/issues/304
+
+    audio_path = ytdata.streams.get_audio_only().download(filename=os.path.join("modules", "yt_tmp.wav"))
+    temp_audio_path = os.path.join("modules", "yt_tmp_fixed.wav")
+
+    try:
+        subprocess.run([
+            'ffmpeg', '-y',
+            '-i', audio_path,
+            temp_audio_path
+        ], check=True)
+
+        os.replace(temp_audio_path, audio_path)
+        return audio_path
+    except subprocess.CalledProcessError as e:
+        print(f"Error during ffmpeg conversion: {e}")
+        return None