check changelog tab

app.py

@@ -1,4 +1,4 @@
-import gradio as gr, glob, os, auditok, zipfile, wave, pytube.exceptions, librosa, time
+import gradio as gr, glob, os, auditok, zipfile, wave, pytube.exceptions, librosa, time, librosa, librosa.display, matplotlib.pyplot as plt, numpy as np
 from pytube import YouTube
 from moviepy.editor import VideoFileClip
 
@@ -131,15 +131,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 get_average_pitch(audio_file):
-    if audio_file == None:
-        raise gr.Error("Audio file cannot be empty!")
-    else:
-        y, sr = librosa.load(audio_file, sr=None)
-        pitches, magnitudes = librosa.piptrack(y=y, sr=sr)
-        mean_pitch = pitches.mean()
-        return f"Average pitch: {mean_pitch:.2f} Hz"
-
 def all_in_one_inator(ytvideo, download_yt_video_as, min_duration, max_duration, name_for_outputted_split_files, progress=gr.Progress()):
     if download_as == "mp3":
         gr.Warning("MP3 is experimental, especially with this, so caution is advised.")
@@ -227,6 +218,18 @@ def check_for_remaining_wav_or_mp3_files(which_filetype):
     audio_files = glob.glob(f"*.{which_filetype}")
     return f"There are {len(audio_files)} leftover files."
 
+def display_audio_spectrogram(audio_file):
+    y, sr = librosa.load(audio_file)
+    d = librosa.amplitude_to_db(np.abs(librosa.stft(y)), ref=np.max)
+    plt.figure(figsize=(12, 8))
+    librosa.display.specshow(d, sr=sr, x_axis="time", y_axis="log")
+    plt.colorbar(format="%+2.0f db")
+    plt.title("Spectrogram")
+    output = "spectrogram.png"
+    plt.savefig(output)
+    plt.close()
+    return output
+
 with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") as app:
     gr.HTML(
         "<h1> Welcome to the GDMGS! (GlobalDatasetMaker Gradio Space) </h1>"
@@ -295,19 +298,6 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             [audiofileuploader, mindur2, maxdur2, name_for_split_files2],
                             [gr.Text(label="Output"), gr.File(label="Zipped files")]
                         )
-            with gr.Tab("Pitch analyzer"):
-                gr.Markdown("Upload a wav file here, and this will determine the average pitch.")
-                gr.HTML("<h1> Zip files and MP3 files are not supported as of now.")
-                with gr.Row():
-                    with gr.Column():
-                            with gr.Row():
-                                upload = gr.File(file_count='single', file_types=[".wav"], label="WAV file")
-                            analyze = gr.Button("Analyze", variant='primary')
-                    analyze.click(
-                        get_average_pitch,
-                        [upload],
-                        [gr.Text(label="Result")]
-                    )
             with gr.Tab("All-in-one downloader and splitter"):
                 gr.Markdown("This is very experimental and may break or change in the future. This essentially combines both the first 2 tabs into an all-in-one script.")
                 with gr.Row():
@@ -349,7 +339,19 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             which_filetype,
                             gr.Text(label="Result")
                         )
+            with gr.Tab("Audio spectrogram"):
+                gr.Markdown("Insert a wav file here and this will show the spectrogram for it.")
+                with gr.Row():
+                    with gr.Column():
+                        filetoanalyze = gr.File(file_count='single', file_types=[".wav"], label="WAV file")
+                        analyzebtn = gr.Button("Display", variant='primary')
+                    analyzebtn.click(
+                        display_audio_spectrogram,
+                        filetoanalyze,
+                        gr.Image(label="Spectrogram result", show_download_button=False, scale=2)
+                    )
         with gr.TabItem("Changelog"):
+            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.")