Update app.py

app.py

@@ -1,108 +1,7 @@
-import gradio as gr, glob, os, auditok, zipfile, wave, pytube.exceptions, urllib.error, requests, json, traceback, yt_dlp, tempfile
-from pytube import YouTube
+import gradio as gr, glob, os, auditok, zipfile, requests, json, traceback, tempfile
 from moviepy.editor import *
 from pydub import AudioSegment
 
-
-def download_video(url, download_as, use_ytdlp):
-    if use_ytdlp == True:
-        try:
-            ydl_opts = {
-                'format': f"{download_as}/bestaudio/best",
-                'postprocessors': [{
-                    'key': 'FFmpegExtractAudio',
-                    'preferredcodec': download_as,
-                }]
-            }
-            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                ydl.download(url)
-            for i in glob.glob(f"*.{download_as}"):
-                if os.path.exists(i):
-                    os.rename(i, f"output.{download_as}")
-            return "Finished downloading! Please proceed to next tab."
-        except:
-            raise gr.Error(traceback.format_exc())
-    else:
-        try:
-            yt = YouTube(url)
-        except pytube.exceptions.RegexMatchError:
-            raise gr.Error("URL not valid or is empty! Please fix the link or enter one!")
-        except urllib.error.HTTPError as not_ok:
-            raise gr.Error(f"Recieved {not_ok}")
-        except pytube.exceptions.AgeRestrictedError:
-            raise gr.Error("The video you inputted is age-restricted! Please try another link.")
-        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 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."
-
-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.mp3") and not os.path.exists("output.wav"):
-        raise gr.Error("Neither output.mp3 or output.wav exist! Did the video download correctly?")
-    if mindur == maxdur:
-        raise gr.Error(f"Cannot split mindur={mindur} and maxdur={maxdur}, min and max are the same number.")
-    elif mindur > maxdur:
-        raise gr.Error(f"Cannot split mindur={mindur} and maxdur={maxdur}, mindur is higher than maxdur.")
-    elif name_for_split_files == None:
-        raise gr.Error("Split files name cannot be empty!")
-    else:
-        audio_path = "output.wav" if not os.path.exists("output.mp3") else "output.mp3"
-        audio_regions = auditok.split(
-            audio_path,
-            min_dur=mindur,
-            max_dur=maxdur,
-            max_silence=0.3,
-            energy_threshold=45
-        )
-        os.remove(audio_path)
-        for i, r in enumerate(audio_regions):
-            filename = r.save(f"{name_for_split_files}-{i+1}.wav")
-        for f in sorted(glob.glob("*.wav")):
-            audio_files = glob.glob("*.wav")
-            zip_file_name = "audio_files.zip"
-            with zipfile.ZipFile(zip_file_name, "w") as zip_file:
-                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)
-            total_files = 0
-            total_length = 0.0
-            for file_name in glob.glob("*.wav"):
-                file_path = os.path.join(os.getcwd(), file_name)
-                if file_path.lower().endswith(".wav"):
-                    try:
-                        with wave.open(file_path, 'r') as audio_file:
-                            frames = audio_file.getnframes()
-                            rate = audio_file.getframerate()
-                            duration = frames / float(rate)
-                            file_size = os.path.getsize(file_path)
-                            if file_size > largest_file[1]:
-                                largest_file = (file_name, file_size)
-                            total_length += duration
-                            total_files += 1
-                    except wave.Error as e:
-                        raise gr.Error(f"Error reading file: {e}")
-            length_mins = total_length / 60
-            for file2 in glob.glob("*.wav"):
-                os.remove(file2)
-            return f"Files split successfully!\nCheck below for zipped files.\n\n{total_files} files created, {length_mins:.2f} minutes total.", zip_file_name
-
 def split_wav_or_mp3_file(audiofileuploader, mindur2, maxdur2, name_for_split_files2, strict):
     if audiofileuploader == None:
         raise gr.Error("Audio file cannot be empty!")
@@ -135,50 +34,6 @@ def split_wav_or_mp3_file(audiofileuploader, mindur2, maxdur2, name_for_split_fi
         os.remove(file2)
     return f"File split successfully!\nCheck below for zipped files.\nAmount created: {len(audio_files)}", zip_file_name2
 
-def download_video_as_audio_only(yt_video, audio_output_format):      
-    try:
-        ydl_opts = {
-            'format': f"{audio_output_format}/bestaudio/best",
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': "mp3"
-            }]
-        }
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.download(yt_video)
-    except pytube.exceptions.RegexMatchError:
-        raise gr.Error("URL not valid or is empty! Please fix the link or enter one!")
-    except urllib.error.HTTPError as not_ok:
-        raise gr.Error(f"Recieved {not_ok}")
-    except pytube.exceptions.AgeRestrictedError:
-        raise gr.Error("The video you inputted is age-restricted! Please try another link.")
-    #video = yt.streams.get_audio_only()
-    #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") if audio_output_format == "wav" else glob.glob("*.mp3")
-    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))
-    for outputwavremoval in glob.glob("*.wav"):
-        if os.path.exists(outputwavremoval):
-            os.remove(outputwavremoval)
-    for outputmp3removal in glob.glob("*.mp3"):
-        if os.path.exists(outputmp3removal):
-            os.remove(outputmp3removal)
-    #return f"Done! Download the zip file below! This only contains the audio file.\n\nYou have downloaded a video.", single_zip_name
-    return "Done! Download the zip file below! This only contains the audio file.", single_zip_name
-
 def mp4_to_wav_or_mp3(mp4fileuploader, file_format):
     if mp4fileuploader == None:
         raise gr.Error("Input cannot be empty!")
@@ -259,50 +114,6 @@ def mvsep_download_separated_audio(hash_textbox):
         return json.dumps(urls, indent=4)
     except requests.exceptions.JSONDecodeError:
         return gr.Info("Nothing to download yet. Check back later.")
-    
-def mvsep_yt_link_request(mvsep_key2, sep_dropdown2, yt_link):
-        try:
-            ydl_opts = {
-            'format': "mp3/bestaudio/best",
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': "mp3"
-            }]
-            }
-            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                ydl.download(yt_link)
-        except pytube.exceptions.RegexMatchError:
-            raise gr.Error("URL not valid or is empty! Please fix the link or enter one!")
-        except urllib.error.HTTPError as not_ok:
-            raise gr.Error(f"Recieved {not_ok}")
-        except pytube.exceptions.AgeRestrictedError:
-            raise gr.Error("The video you inputted is age-restricted! Please try another link.")
-        #video = yt.streams.get_highest_resolution()
-        #video.download()
-        #video_path = f"{video.default_filename}"
-        #video_clip = VideoFileClip(video_path)
-        #audio_clip = video_clip.audio
-        #audio_clip.write_audiofile("output.mp3")
-        #audio_clip.close()
-        #video_clip.close()
-        for rename in glob.glob("*.mp3"):
-            if os.path.exists(rename):
-                os.rename(rename, "output.mp3")
-        for removalmp4 in glob.glob("*.mp4"):
-            os.remove(removalmp4)
-
-        url = "https://mvsep.com/api/separation/create"
-        files = {
-            "audiofile": open("output.mp3", 'rb')
-        }
-        data = {
-            "api_token": mvsep_key2,
-            "sep_type": sep_dropdown2
-        }
-        r = requests.post(url, files=files, data=data)
-        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:
@@ -322,6 +133,9 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
     gr.HTML(
         "<h1> Welcome to the Cafeteria (formally GDMGS)!</h1>"
     )
+    gr.HTML(
+        "<h2> Currently, no yt functions are working right now, so they have been deleted. Please use this tool to split an audio file, convert an mp4 file to mp3 or wav, or use mvsep."
+    )
     gr.Markdown("## Duplicate this space if you want to make your own changes!")
     gr.HTML(
         """<p style="margin:5px auto;display: flex;justify-content: left;">
@@ -335,35 +149,6 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
         "<h2> This Space's storage is ephemeral, meaning all audio files are visible to you only. I do not have access to any of this, nor would I do anything with it anyway. </h2>"
     )
     with gr.Tabs():
-        with gr.TabItem("Download Video"):
-            with gr.Row():
-                with gr.Column():
-                    with gr.Row():
-                        gr.Markdown("# Currently broken. Use Stacher in the meantime.")
-                        #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?")
-                        #use_ytdlp = gr.Checkbox(False, label="Use yt_dlp instead of pytube?", info="Sometimes Pytube refuses to download a video. If that happens, check this box to download using yt_dlp instead.")
-                        #convertion = gr.Button("Download", variant='primary')
-            #convertion.click(
-            #    fn=download_video,
-            #    inputs=[url, download_as, use_ytdlp],
-            #    outputs=gr.Text(label="Output")
-            #)
-        with gr.TabItem("Split audio files"):
-            with gr.Row():
-                with gr.Column():
-                    with gr.Row():
-                        gr.Markdown("# Offline due to the download video section not working. If you have an audio file however, go to the 'Misc Tools' tab to split that instead.")
-                        #mindur = gr.Number(label="Min duration", minimum=1, maximum=10, value=1)
-                        #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?")
-                        #splitbtn = gr.Button("Split", variant='primary')
-            #splitbtn.click(
-            #    split_audio_from_yt_video,
-            #    inputs=[mindur, maxdur, name_for_split_files, show_amount_of_files_and_file_dur],
-            #    outputs=[gr.Text(label="Output"), gr.File(label="Zipped files")]
-            #)
         with gr.TabItem("Misc tools"):
             with gr.Tab("File splitter"):
                 gr.Markdown("If you would rather split a single WAV or mp3 audio file, use this method instead.")
@@ -395,21 +180,6 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             [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.")
-               #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.Tab("MP4 to mp3/wav converter"):
                 gr.Markdown("If you have an mp4 file, you can convert it to mp3 or wav here. Only click the 'Remove file' button when done.")
                 with gr.Row():
@@ -486,7 +256,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                                     "45 - Bandit v2 (speech, music, effects)"
                                     ],
                                     max_choices=1,
-                                    value="11 - Demucs3 Model B (vocals, drums, bass, other)",
+                                    value="45 - Bandit v2 (speech, music, effects)",
                                     label="Model type:",
                                     interactive=True,
                                     type='index'
@@ -497,73 +267,6 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                                 [mvsep_key, audio_file, sep_dropdown],
                                 [gr.Text(label="Output")]
                             )
-                with gr.Tab("Send Request via yt link"):
-                    gr.Markdown("## Currently broken. Use MVSEP's actual website in the meantime.")
-                    #with gr.Row():
-                    #    with gr.Column():
-                    #        with gr.Row():
-                    #            mvsep_key2 = gr.Textbox(placeholder="Enter your MVSEP API key.", label="API key")
-                    #            yt_link = gr.Textbox(label="YouTube link")
-                    #            sep_dropdown2 = gr.Dropdown(
-                    #                ["0 - spleeter (vocals, music)",
-                    #                "1 - spleeter (vocals, drums, bass, other)",
-                    #                "2 - spleeter (vocals, drums, bass, piano, other)",
-                    #                "3 - unmix XL (vocals, drums, bass, other)",
-                    #                "4 - unmix HQ (vocals, drums, bass, other)",
-                    #                "5 - unmix SD (vocals, drums, bass, other)",
-                    #                "6 - unmix SE (vocals, music)",
-                    #                "7 - MDX A (vocals, drums, bass, other)",
-                    #                "8 - MDX B (vocals, drums, bass, other)",
-                    #                "9 - UVR HQ (vocals, music)",
-                    #                "10 - Demucs3 Model A (vocals, drums, bass, other)",
-                    #                "11 - Demucs3 Model B (vocals, drums, bass, other)",
-                    #                "12 - MDX-B Karaoke (lead/back vocals)",
-                    #                "13 - Demucs2 (vocals, drums, bass, other)",
-                    #                "14 - Zero Shot (Query Based) (LQ)",
-                    #                "15 - Danna sep (vocals, drums, bass, other)",
-                    #                "16 - Byte Dance (vocals, drums, bass, other)",
-                    #                "17 - UVRv5 Demucs (vocals, music)",
-                    #                "18 - MVSep DNR (music, sfx, speech)",
-                    #                "19 - MVSep Vocal Model (vocals, music)",
-                    #                "20 - Demucs4 HT (vocals, drums, bass, other)",
-                    #                "--------------------------------------------------------------------------------------------------",
-                    #                "22 - FoxJoy Reverb Removal (other)",
-                    #                "23 - MDX B (vocals, instrumental)",
-                    #                "24 - MVSep Demucs4HT DNR (dialog, sfx, music)",
-                    #                "25 - MDX23C (vocals, instrumental)",
-                    #                "26 - Ensemble (vocals, instrumental) [PREMIUM ONLY]",
-                    #                "27 - Demucs4 Vocals 2023 (vocals, instrumental)",
-                    #                "28 - Ensemble (vocals, instrumental, bass, drums, other) [PREMIUM ONLY]",
-                    #                "29 - MVSep Piano (piano, other)",
-                    #                "30 - Ensemble All-In (vocals, bass, drums, piano, guitar, lead/back vocals, other) [PREMIUM ONLY]",
-                    #                "31 - MVSep Guitar (guitar, other)",
-                    #                "-------------------------------------------------------------------------------------------------",
-                    #                "33 - Vit Large 23 (vocals, instrum)",
-                    #                "34 - MVSep Crowd removal (crowd, other)",
-                    #                "35 - MVSep MelBand Roformer (vocals, instrum)",
-                    #                "36 - BandIt Plus (speech, music, effects)",
-                    #                "37 - DrumSep (kick, snare, cymbals, toms)",
-                    #                "38 - LarsNet (kick, snare, cymbals, toms, hihat)",
-                    #                "39 - Whisper (extract text from audio)",
-                    #                "40 - BS Roformer (vocals, instrumental)",
-                    #                "41 - MVSep Bass (bass, other)",
-                    #                "42 - MVSep MultiSpeaker (MDX23C)",
-                    #                "43 - MVSEP Multichannel BS (vocals, inst)",
-                    #                "44 - MVSep Drums (drums, other)",
-                    #                "45 - Bandit v2 (speech, music, effects)"
-                    #                ],
-                    #                max_choices=1,
-                    #                value="11 - Demucs3 Model B (vocals, drums, bass, other)",
-                    #                label="Model type:",
-                    #                interactive=True,
-                    #                type='index'
-                    #            )
-                    #            send_req2 = gr.Button("Send request", variant='primary')
-                    #        send_req2.click(
-                    #            mvsep_yt_link_request,
-                    #            [mvsep_key2, sep_dropdown2, yt_link],
-                    #            [gr.Text(label="Output")]
-                    #        )
                 with gr.Tab("Get status of request"):
                     with gr.Row():
                         with gr.Column():
@@ -583,12 +286,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             )
 
         with gr.TabItem("Changelog"):
+            gr.Markdown("v1 - Not the most exciting v1 release. **Removed yt functions as they are no longer working.**")
             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.")
-            gr.Markdown("v0.99.5 - Added bug fixes and Dropdown instead of button for MVSEP.")
-            gr.Markdown("v0.99.4 - Added a button to display the available models for MVSEP.")
-            gr.Markdown("v0.99.3 - Added MVSEP in Misc Tools.")
         
 app.launch()