Update spectro.py

spectro.py

@@ -182,4 +182,28 @@ def mp3_bytes_from_wav_bytes(wav_bytes: io.BytesIO) -> io.BytesIO:
     sound = pydub.AudioSegment.from_wav(wav_bytes)
     sound.export(mp3_bytes, format="mp3")
     mp3_bytes.seek(0)
-    return mp3_bytes
+    return mp3_bytes
+
+def image_from_spectrogram(spectrogram: np.ndarray, max_volume: float = 50, power_for_image: float = 0.25) -> Image.Image:
+    """
+    Compute a spectrogram image from a spectrogram magnitude array.
+    """
+    # Apply the power curve
+    data = np.power(spectrogram, power_for_image)
+
+    # Rescale to 0-255
+    data = data * 255 / max_volume
+
+    # Invert
+    data = 255 - data
+
+    # Convert to a PIL image
+    image = Image.fromarray(data.astype(np.uint8))
+
+    # Flip Y
+    image = image.transpose(Image.FLIP_TOP_BOTTOM)
+
+    # Convert to RGB
+    image = image.convert("RGB")
+
+    return image