Update app.py

app.py

@@ -1,7 +1,7 @@
-import gradio as gr, glob, os, auditok, zipfile, wave, pytube.exceptions, urllib.error, requests, json
+import gradio as gr, glob, os, auditok, zipfile, wave, pytube.exceptions, urllib.error, requests, json, traceback, yt_dlp, tempfile
 from pytube import YouTube
 from moviepy.editor import *
-import traceback, yt_dlp
+from pydub import AudioSegment
 
 
 def download_video(url, download_as, use_ytdlp):
@@ -107,9 +107,9 @@ def split_wav_or_mp3_file(audiofileuploader, mindur2, maxdur2, name_for_split_fi
     if audiofileuploader == None:
         raise gr.Error("Audio file cannot be empty!")
     if mindur2 == maxdur2:
-        raise gr.Error(f"Cannot split mindur={mindur} and maxdur={maxdur}, min and max are the same number.")
+        raise gr.Error(f"Cannot split mindur={mindur2} and maxdur={maxdur2}, min and max are the same number.")
     elif mindur2 > maxdur2:
-        raise gr.Error(f"Cannot split mindur={mindur} and maxdur={maxdur}, mindur is higher than maxdur.")
+        raise gr.Error(f"Cannot split mindur={mindur2} and maxdur={maxdur2}, mindur is higher than maxdur.")
     elif name_for_split_files2 == None:
         raise gr.Error("Split files name cannot be empty!")
     else:
@@ -303,6 +303,19 @@ def mvsep_yt_link_request(mvsep_key2, sep_dropdown2, yt_link):
         json_format = r.json()
         hash_val = json_format['data']['hash']
         return f"Request sent successfully! Your hash is: {hash_val}\n\nUse the next tab to check the status of your request."
+
+def split_by_dur(file_upload, dur_to_cut):
+    if file_upload == None:
+        raise gr.Error("Audio file cannot be empty!")
+    try:
+        audio = AudioSegment.from_file(file_upload)
+        trimmed_audio = audio[:dur_to_cut]
+        ms_to_min = round((dur_to_cut/(1000*60))%60, 1)
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as temp_file:
+            trimmed_audio.export(temp_file.name, format="mp3")
+            return f"Trimmed to {ms_to_min} minutes successfully.", temp_file.name
+    except Exception as e:
+        return gr.Error(traceback.format_exc())
         
 
 with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") as app:
@@ -369,6 +382,20 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             [gr.Text(label="Output"), gr.File(label="Zipped files")]
                         )
 
+            with gr.Tab("Split audio file by duration"):
+                gr.Markdown("If you have an audio file thats too long for MVSEP (or any other use cases), use this section to split the audio by duration.")
+                with gr.Row():
+                    with gr.Column():
+                        with gr.Row():
+                            file_upload = gr.File(file_count='single', file_types=[".wav", ".mp3"], label="WAV or MP3 file")
+                            dur_to_cut = gr.Number(label="Duration to split audio (in ms). Cannot be larger than the duration of the audio file.", precision=0)
+                            begin = gr.Button("Split at given duration", variant='primary')
+                        begin.click(
+                            split_by_dur,
+                            [file_upload, dur_to_cut],
+                            [gr.Text(label="Output"), gr.File(label="Trimmed audio")]
+                        )
+
             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.")
                 gr.Markdown("# Currently broken. Use Stacher in the meantime.")
@@ -556,6 +583,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             )
 
         with gr.TabItem("Changelog"):
+            gr.Markdown("v0.99.9 - Added new tool: Split audio file by duration.")
             gr.Markdown("v0.99.8 - Added new MVSep models.")
             gr.Markdown("v0.99.7 - Temporarily fixed an issue causing a 400 error to happen whenever a youtube url was placed.")
             gr.Markdown("v0.99.6 - Added a yt link request method for MVSEP.")