Spaces:
Running
Running
File size: 903 Bytes
05b9d17 a2455e3 a70c074 63ab978 ebb9b01 63ab978 ebb9b01 63ab978 ebb9b01 63ab978 ebb9b01 a2455e3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
from pytubefix import YouTube
import subprocess
import os
def get_ytdata(link):
return YouTube(link)
def get_ytmetas(link):
yt = YouTube(link)
return yt.thumbnail_url, yt.title, yt.description
def get_ytaudio(ytdata: YouTube):
# 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
|