check changelog tab for new update

app.py

@@ -194,6 +194,31 @@ def all_in_one_inator(ytvideo, download_yt_video_as, min_duration, max_duration,
         time.sleep(2)
         return "Process done successfully! Check below for zipped files!", zip_file_name
 
+def download_video_as_audio_only(yt_video, audio_output_format):
+    try:
+        yt = YouTube(yt_video)
+    except pytube.exceptions.RegexMatchError:
+        raise gr.Error("URL not valid or is empty! Please fix the link or enter one!")
+    video = yt.streams.get_highest_resolution()
+    video.download()
+    video_path = f"{video.default_filename}"
+    video_clip = VideoFileClip(video_path)
+    audio_clip = video_clip.audio
+    if audio_output_format == "wav":
+        audio_clip.write_audiofile("output.wav")
+    elif audio_output_format == "mp3":
+        audio_clip.write_audiofile("output.mp3")
+    audio_clip.close()
+    video_clip.close()
+    for mp4remove in glob.glob("*.mp4"):
+        os.remove(mp4remove)
+    single_zip_name = "only_audio.zip"
+    audio_files = glob.glob("*.wav")
+    with zipfile.ZipFile(single_zip_name, 'w') as zip_file:
+        for audio_file in audio_files:
+            zip_file.write(audio_file, os.path.basename(audio_file))
+    return "Done! Download the zip file below! This only contains the audio file.", single_zip_name
+
 with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") as app:
     gr.HTML(
         "<h1> Welcome to the GDMGS! (GlobalDatasetMaker Gradio Space) </h1>"
@@ -286,7 +311,21 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             [ytvideo, download_yt_video_as, min_duration, max_duration, name_for_outputted_split_files],
                             [gr.Text(label="Result"), gr.File(label="Zipped files")]
                         )
+            with gr.Tab("Audio only download"):
+                gr.Markdown("If you want to download only the audio (to isolate bgm using UVR, etc), use this method, which will only extract audio and not split the audio.")
+                with gr.Row():
+                    with gr.Column():
+                        with gr.Row():
+                            yt_video = gr.Textbox(label="URL")
+                            audio_output_format = gr.Radio(["wav", "mp3"], value="wav", label="Download audio as:")
+                            commence_download = gr.Button("Download", variant='primary')
+                        commence_download.click(
+                            download_video_as_audio_only,
+                            [yt_video, audio_output_format],
+                            [gr.Text(label="Output"), gr.File(label="Zipped audio file")]
+                        )
         with gr.TabItem("Changelog"):
+            gr.Markdown("v0.94 - Added new tool: YouTube-to-audio.")
             gr.Markdown("v0.93 - Removed obsolete warnings and fixed issue with all-in-one if output.mp3 or output.wav doesnt exist.")
             gr.Markdown("v0.92 - Added all-in-one tab under Misc Tools.")
             gr.Markdown("v0.91 - Added mp3 file support for single file splitting, and also fixed bug if neither output.wav or output.mp3 exists.")