Merge pull request #305 from jhj0517/fix/youtube-pipeline

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

modules/whisper/whisper_base.py

@@ -356,6 +356,9 @@ class WhisperBase(ABC):
             )
             result_str = f"Done in {self.format_time(time_for_task)}! Subtitle file is in the outputs folder.\n\n{subtitle}"
 
+            if os.path.exists(audio):
+                os.remove(audio)
+
             return [result_str, result_file_path]
 
         except Exception as e: