new update, check changelog tab

app.py

@@ -214,7 +214,7 @@ def mvsep_api_request(mvsep_key, audio_file, sep_dropdown):
     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."
+    return f"Request sent successfully! Your hash is: {hash_val}\n\nUse the 'Get Status' tab to check the status of your request."
 
 def mvsep_check_request(hash_textbox):
     try:
@@ -246,11 +246,45 @@ 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:
+            yt = YouTube(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 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."
+        
 
 
 with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") as app:
     gr.HTML(
-        "<h1> Welcome to the Cafeteria! (formally GDMGS)</h1>"
+        "<h1> Welcome to the Cafeteria (formally GDMGS)!</h1>"
     )
     gr.Markdown("## Duplicate this space if you want to make your own changes!")
     gr.HTML(
@@ -259,10 +293,10 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
         </p>"""
     )
     gr.Markdown(
-        "This Space will create a dataset for you and use MVSEP to isolate vocals (EXPERIMENTAL), all automatically. **Please be warned that due to not having a GPU on this Space, some steps might take longer to complete.**"
+        "This Space will create a dataset for you and use MVSEP to isolate vocals, all automatically. **Please be warned that due to not having a GPU on this Space, some steps might take longer to complete.**"
     )
     gr.HTML(
-        "<h2> This Space's storage is ephemeral, meaning once you reload this space, all audio files will be lost. </h2>"
+        "<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"):
@@ -343,8 +377,8 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             None
                         )
             with gr.Tab("MVSEP"):
-                gr.Markdown("**VERY EXPERIMENTAL!** Use MVSEP to isolate audio.\n\n**You will be required to input your API key, but it will not be saved ever, I don't use anything saved here for bad intentions, nor would I have access to it regardless.**")
-                with gr.Tab("Send Request"):
+                gr.Markdown("Use MVSEP to isolate audio.\n\n**You will be required to input your API key, but it will not be saved ever, I don't use anything saved here for bad intentions, nor would I have access to it regardless.**")
+                with gr.Tab("Send Request via audio file"):
                     with gr.Row():
                         with gr.Column():
                             with gr.Row():
@@ -405,6 +439,67 @@ 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"):
+                    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)"],
+                                    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():
@@ -424,8 +519,9 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             )
 
         with gr.TabItem("Changelog"):
+            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. This is VERY EXPERIMENTAL and there will be bugs present.")
+            gr.Markdown("v0.99.3 - Added MVSEP in Misc Tools.")
         
 app.launch()