check changelog tab

app.py

@@ -1,30 +1,48 @@
-import gradio as gr, glob, os, auditok, zipfile, wave, pytube.exceptions, librosa, time, librosa, librosa.display, matplotlib.pyplot as plt, numpy as np, urllib.error
+import gradio as gr, glob, os, auditok, zipfile, wave, pytube.exceptions, librosa, time, librosa, librosa.display, matplotlib.pyplot as plt, numpy as np, urllib.error, traceback, yt_dlp
 from pytube import YouTube
 from moviepy.editor import VideoFileClip
 
-def download_video(url, download_as):
-    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 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:
@@ -247,6 +265,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                     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?")
+                        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,
@@ -336,27 +355,8 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                         gr.Image(label="Spectrogram result", show_download_button=False, scale=2)
                     )
         with gr.TabItem("Changelog"):
+            gr.Markdown("v0.98.5 - Added an option to download yt link via yt_dlp if pytube cant download it.")
             gr.Markdown("v0.98.2 - Added new upcoming features tab.")
-            gr.Markdown("v0.98.1 - Reverted back due to none of the new features working apart from removing SR analyzer and adding new error checks.")
-            gr.Markdown("v0.97a - Fixed issue with spectrogram tab if no WAV file was selected.")
-            gr.Markdown("v0.97 - Added new spectrogram tool in Misc Tools and removed pitch analyzer as it did not fit with the purpose of this Space.")
-            gr.Markdown("v0.96 - Added new remaining files tool in Misc Tools.")
-            gr.Markdown("v0.95 - Fixed issue with mp3 files not downloading audio properly.")
-            gr.Markdown("v0.94a - Fixed issue with existing output.wav or output.mp3 files clashing with the split audio files with addition of the new tool.")
-            gr.Markdown("v0.94 - Added new tool: YouTube-to-audio.")
-            gr.Markdown("v0.93 - Removed obsolete warnings and fixed issue with all-in-one if output.mp3 or output.wav doesnt exist.")
-            gr.Markdown("v0.92 - Added all-in-one tab under Misc Tools.")
-            gr.Markdown("v0.91 - Added mp3 file support for single file splitting, and also fixed bug if neither output.wav or output.mp3 exists.")
-            gr.Markdown("v0.90a - Fixed bug that if 'show_amount_of_files_and_file_dur' was False, split wav files would not be deleted.")
-            gr.Markdown("v0.90 - Added mp3 support for downloading a Youtube video.")
-            gr.Markdown("v0.85 - Fixed bug in pitch analyzer if no audio file was given.")
-            gr.Markdown("v0.80 - Added new tool: Pitch Analyzer.")
-            gr.Markdown("v0.75 - Fixed bug that would cause split wav files to be packaged with the previously split wav files.")
-            gr.Markdown("v0.74 - Added new tool: WAV file splitter.")
-            gr.Markdown("v0.73 - Added Misc Tools tab and new Sample Rate analyzer tool.")
-            gr.Markdown("v0.70 - Fixed bug if no URL was passed or if the URL was invalid.")
-            gr.Markdown("v0.65 - Fixed bug if user tried to split an audio file when 'output.wav' didnt exist.")
-            gr.Markdown("v0.60 - Initial push to Huggingface Space.")
         with gr.TabItem("Upcoming features"):
             gr.HTML(
                 """