Update app.py

app.py

@@ -72,14 +72,22 @@ def plot_spectrogram(wav_file, output_image):
 
     # Create a plot for the spectrogram
     plt.figure(figsize=(10, 4))
-    plt.specgram(audio_data, Fs=sample_rate, NFFT=1024, noverlap=512, cmap='inferno', aspect='auto')
-    plt.title('Spectrogram')
-    plt.xlabel('Time [s]')
-    plt.ylabel('Frequency [Hz]')
+    plt.specgram(audio_data, Fs=sample_rate, NFFT=1024, noverlap=512, cmap='gray', aspect='auto')
+    plt.title('Spectrogram', fontsize=14)
+    plt.xlabel('Time [s]', fontsize=12)
+    plt.ylabel('Frequency [Hz]', fontsize=12)
+
+    # Add a colorbar and adjust its label
+    cbar = plt.colorbar()
+    cbar.set_label('Intensity', fontsize=12)
+
+    # Remove gridlines and ticks for a cleaner look
+    plt.grid(False)
+    plt.xticks([])
+    plt.yticks([])
 
     # Save the plot as an image file
-    plt.colorbar(label='Intensity [dB]')
-    plt.savefig(output_image)
+    plt.savefig(output_image, bbox_inches='tight', pad_inches=0, dpi=300)
     plt.close()
 
 def infer(video_in):