fixed wav split bug

app.py

@@ -17,7 +17,7 @@ def download_video(url):
     video_clip.close()
     for removalmp4 in glob.glob("*.mp4"):
         os.remove(removalmp4)
-    return "Finished downloading! Please proceed to final 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:
@@ -122,6 +122,8 @@ def split_wav_file(audiofileuploader, mindur2, maxdur2, name_for_split_files2):
             with zipfile.ZipFile(zip_file_name2, "w") as zip_file:
                 for audio_file in audio_files:
                     zip_file.write(audio_file, os.path.basename(audio_file))
+    for file2 in glob.glob("*.wav"):
+        os.remove(file2)
     return f"File split successfully!\nCheck below for zipped files.\nAmount created: {len(audio_files)}", zip_file_name2
 
 with gr.Blocks(theme=gr.themes.Monochrome(), title="Global Dataset Maker") as app:
@@ -151,6 +153,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), title="Global Dataset Maker") as ap
                         maxdur = gr.Number(label="Max duration", minimum=1, maximum=10, value=5)
                         name_for_split_files = gr.Textbox(label="Name for split files")
                         show_amount_of_files_and_file_dur = gr.Checkbox(False, label="Show total amount of files and duration?")
+                        keep_original_wav_file = gr.Checkbox(False, label="Keep wav file?")
                         splitbtn = gr.Button("Split", variant='primary')
             splitbtn.click(
                 split_audio_from_yt_video,
@@ -170,10 +173,10 @@ with gr.Blocks(theme=gr.themes.Monochrome(), title="Global Dataset Maker") as ap
                         [zipuploader],
                         [gr.Text(label="Result")]
                     )
-            with gr.Tab("File splitter (IN PROGRESS)"):
+            with gr.Tab("File splitter"):
                 gr.Markdown("If you would rather split a single WAV (mp3 support soon) audio file, use this method instead.")
                 gr.HTML(
-                    "<h1> This is a work in progress, there are still some bugs to be fixed. </h1>"
+                    "<h1> Most bugs should be fixed now, if not, let me know. </h1>"
                 )
                 with gr.Row():
                     with gr.Column():