import numpy as np
import matplotlib.pyplot as plt

# Load the Mel spectrogram data
spectrogram = np.load("./testdata/rust_jfk.npy")
# Plot the spectrogram
plt.figure(figsize=(10, 4))
plt.imshow(spectrogram, aspect="auto", origin="lower", cmap="viridis")
plt.title("Mel Spectrogram")
plt.ylabel("Mel Filter Banks")
plt.xlabel("Frame (Time)")
plt.colorbar(label="Magnitude")
plt.show()