Download original youtube audio

create conda environment.md

@@ -17,7 +17,7 @@ pip install huggingface-hub
 pip install torch
 pip install torchaudio
 <!-- pip install pysndfile==1.0.0 -->
-conda install -c conda-forge libsndfile==1.0.31
+mamba install -y -c conda-forge libsndfile==1.0.31 pyperclip
 
 # Download videos
 pip install twitch-dl

requirements.txt

@@ -4,6 +4,8 @@ gradio
 huggingface_hub
 torch
 torchaudio
+libsndfile==1.0.31
 # pysndfile
 twitch-dl
 pytube
+pyperclip

translatube.py

@@ -1,8 +1,92 @@
 import gradio as gr
+import pyperclip
+import urllib.parse as urlparse
+from pytube import YouTube
 
-def greet(name):
-    return "Hello " + name + "!"
+YOUTUBE = "youtube"
+TWITCH = "twitch"
+
+def copy_url_from_clipboard():
+    return pyperclip.paste()
+
+def clear_video_url():
+    return ""
+
+def get_youtube_thumbnail(video_id):
+    thumbnail_url = f"https://img.youtube.com/vi/{video_id}/0.jpg"
+    return thumbnail_url
+
+def get_youtube_video_id(url):
+    parsed_url = urlparse.urlparse(url)
+    video_id = urlparse.parse_qs(parsed_url.query).get('v')
+    if video_id:
+        thumbnail_url = get_youtube_thumbnail(video_id[0])
+        return thumbnail_url
+    else:
+        return None
+
+def is_valid_url(url):
+    button = gr.Button(size="sm", value="translate", min_width="10px", scale=0, visible=True)
+    original_audio = gr.Audio(label="Original audio", elem_id="original_audio", visible=True)
+    translated_audio = gr.Audio(label="Translated audio", elem_id="translated_audio", visible=True)
+    if "youtube" in url.lower() or "youtu.be" in url.lower():
+        thumbnail = get_youtube_video_id(url)
+        if thumbnail:
+            return (
+                gr.Image(value=thumbnail, visible=True, show_download_button=False, container=False), 
+                button, 
+                gr.Textbox(value=YOUTUBE, label="Stream page", elem_id="stream_page", visible=False),
+                original_audio,
+                translated_audio,
+                )
+    elif "twitch" in url.lower() or "twitch.tv" in url.lower():
+        return (
+            gr.Image(value="assets/twitch.webp", visible=True), 
+            button, 
+            gr.Textbox(value=TWITCH, label="Stream page", elem_id="stream_page", visible=False),
+            original_audio,
+            translated_audio,
+            )
+
+def get_audio_from_video(url, stream_page):
+    if stream_page == YOUTUBE:
+        yt = YouTube(url)
+        audio_streams = yt.streams.filter(mime_type="audio/mp4")
+
+        # Get all available audio bitrates
+        abr_list = []
+        for stream in audio_streams:
+            abr_list.append(stream.abr)
+        abr_list = sorted(set(abr_list))
+
+        # Get the highest audio bitrate
+        audio_stream = audio_streams.filter(abr=abr_list[0]).first()
+
+        # Download the audio
+        audio_stream.download(filename="audio.mp3")
+
+        return gr.Audio("audio.mp3", label="Original audio", elem_id="original_audio", visible=True)
+    elif stream_page == TWITCH:
+        pass
+
+
+with gr.Blocks() as demo:
+    with gr.Row(variant="panel"):
+        url_textbox = gr.Textbox(placeholder="Add video URL here", label="Video URL", elem_id="video_url", scale=1, interactive=True)
+        copy_button   = gr.Button(size="sm", icon="icons/copy.svg",   value="", min_width="10px", scale=0)
+        delete_button = gr.Button(size="sm", icon="icons/delete.svg", value="", min_width="10px", scale=0)
+    copy_button.click(fn=copy_url_from_clipboard, outputs=url_textbox)
+    delete_button.click(fn=clear_video_url, outputs=url_textbox)
+
+    stream_page = gr.Textbox(label="Stream page", elem_id="stream_page", visible=False)
+    visible = False
+    with gr.Row(equal_height=False):
+        image = gr.Image(visible=visible, scale=1)
+        translate_button = gr.Button(size="sm", value="translate", min_width="10px", scale=0, visible=visible)
+
+    original_audio = gr.Audio(label="Original audio", elem_id="original_audio", visible=visible)
+    translated_audio = gr.Audio(label="Translated audio", elem_id="translated_audio", visible=visible)
+    url_textbox.change(fn=is_valid_url, inputs=url_textbox, outputs=[image, translate_button, stream_page, original_audio, translated_audio])
+    translate_button.click(fn=get_audio_from_video, inputs=[url_textbox, stream_page], outputs=original_audio)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-    
 demo.launch()