add mp3 support for yt vid

also fixed bug with wav files remaining when show_amount_of_files_and_file_dur was unchecked.

app.py

@@ -2,7 +2,9 @@ import gradio as gr, glob, os, auditok, zipfile, wave, pytube.exceptions, libros
 from pytube import YouTube
 from moviepy.editor import VideoFileClip
 
-def download_video(url):
+def download_video(url, download_as):
+    if download_as == "mp3":
+        gr.Warning("MP3 support is experimental at the moment, but is in current testing.")
     try:
         yt = YouTube(url)
     except pytube.exceptions.RegexMatchError:
@@ -12,18 +14,24 @@ def download_video(url):
     video_path = f"{video.default_filename}"
     video_clip = VideoFileClip(video_path)
     audio_clip = video_clip.audio
-    audio_clip.write_audiofile("output.wav")
+    if download_as == "wav":
+        audio_clip.write_audiofile("output.wav")
+    elif download_as == "mp3":
+        audio_clip.write_audiofile("output.mp3")
     audio_clip.close()
     video_clip.close()
     for removalmp4 in glob.glob("*.mp4"):
         os.remove(removalmp4)
-    return "Finished downloading! Please proceed to next tab."
+    
+    #return "Finished downloading! Please proceed to next tab."
 
 def split_audio_from_yt_video(mindur, maxdur, name_for_split_files, show_amount_of_files_and_file_dur):
     if show_amount_of_files_and_file_dur == True:
         gr.Warning(f"show_amount_of_files_and_file_dur set to True. This will take longer if your audio file is long.")
-    if not os.path.exists("output.wav"):
-        raise gr.Error("Output.wav does not exist! Did you do the first tab correctly or at all?")
+    if not os.path.exists("output.wav") and os.path.exists("output.mp3"):
+        gr.Warning("Output.wav does not exist, but output.mp3 does. Splitting MP3 file instead...")
+    if not os.path.exists("output.mp3") and os.path.exists("output.wav"):
+        gr.Warning("Output.mp3 does not exist, but output.wav does. Splitting WAV file instead...")
     if mindur == maxdur:
         raise gr.Error(f"Cannot split mindur={mindur} and maxdur={maxdur}, min and max are the same number.")
     elif mindur > maxdur:
@@ -31,7 +39,7 @@ def split_audio_from_yt_video(mindur, maxdur, name_for_split_files, show_amount_
     elif name_for_split_files == None:
         raise gr.Error("Split files name cannot be empty! This will be replaced with an alternative naming style in the future.")
     else:
-        audio_path = "output.wav"
+        audio_path = "output.wav" if not os.path.exists("output.mp3") else "output.mp3"
         audio_regions = auditok.split(
             audio_path,
             min_dur=mindur,
@@ -49,6 +57,8 @@ def split_audio_from_yt_video(mindur, maxdur, name_for_split_files, show_amount_
                 for audio_file in audio_files:
                     zip_file.write(audio_file, os.path.basename(audio_file))
         if show_amount_of_files_and_file_dur == False:
+            for file2 in glob.glob("*.wav"):
+                os.remove(file2)
             return "Files split successfully!\nCheck below for zipped files.", zip_file_name
         elif show_amount_of_files_and_file_dur == True:
             largest_file = ("", 0)
@@ -142,16 +152,18 @@ with gr.Blocks(theme='NoCrypt/miku', title="Global Dataset Maker") as app:
     gr.Markdown(
         "This Space will create a dataset for you, all automatically. **Please be warned that due to not having a GPU on this Space, some steps might take longer to complete.**"
     )
+    gr.HTML("<h1> MP3 file support does work for downloading a video only (for now).")
     with gr.Tabs():
         with gr.TabItem("Download Video"):
             with gr.Row():
                 with gr.Column():
                     with gr.Row():
                         url = gr.Textbox(label="URL")
+                        download_as = gr.Radio(["wav", "mp3"], label="Audio format output", value="wav", info="What should the audio format be output as?")
                         convertion = gr.Button("Download", variant='primary')
             convertion.click(
                 fn=download_video,
-                inputs=[url],
+                inputs=[url, download_as],
                 outputs=gr.Text(label="Output")
             )
         with gr.TabItem("Split audio files"):
@@ -212,5 +224,4 @@ with gr.Blocks(theme='NoCrypt/miku', title="Global Dataset Maker") as app:
                         [upload],
                         [gr.Text(label="Result")]
                     )
-
 app.launch()