Download twitch audio

translatube.py

@@ -2,6 +2,8 @@ import gradio as gr
 import pyperclip
 import urllib.parse as urlparse
 from pytube import YouTube
+import re
+import subprocess
 
 YOUTUBE = "youtube"
 TWITCH = "twitch"
@@ -67,7 +69,13 @@ def get_audio_from_video(url, stream_page):
 
         return gr.Audio("audio.mp3", label="Original audio", elem_id="original_audio", visible=True)
     elif stream_page == TWITCH:
-        pass
+        # Get the video id
+        video_id = re.search("\d{10}", url).group(0)
+
+        # Download the video
+        subprocess.run(["twitch-dl", "download", "--overwrite", "-q", "audio_only", "--output", "audio.mkv", video_id])
+
+        return gr.Audio("audio.mkv", label="Original audio", elem_id="original_audio", visible=True)
 
 
 with gr.Blocks() as demo: