4_spectrum.py

# Learning to visualize audio
# Mr. Poznanski == Poz - nan - ski == Pause - NaN - Ski

# Now, we are going to draw the waveform as a graph

import numpy as np
import PySimpleGUI as sg
import pyaudio

sg.theme("BluePurple")

FFT_BINS = 200

layout = [
    [sg.Graph(canvas_size=(500,300),
              graph_bottom_left=(0, 0),
              graph_top_right=(1,1), key="graph")],
    [sg.Graph(canvas_size=(500, 300),
              graph_bottom_left=(0, 0),
              graph_top_right=(FFT_BINS, 1), key="bars")],
]

window = sg.Window("Visualization", layout, finalize=True)
graph = window["graph"]
bars = window["bars"]

# Setup the microphone
RATE = 44100  # (sampling rate) number of frames per second
CHANNELS = 1
CHUNK = 1000  # signal is split into CHUNK number of frames
FORMAT = pyaudio.paInt16

p = pyaudio.PyAudio()
stream = p.open(format=FORMAT,
                channels=CHANNELS,
                rate=RATE,
                input=True,
                frames_per_buffer=CHUNK)

while True:
    event, values = window.read(timeout=10)

    if event == sg.WIN_CLOSED:
        break

    raw = stream.read(CHUNK, exception_on_overflow=False)
    data = np.frombuffer(raw, dtype=np.int16)
    data = data.astype(np.float32) / 65535.0

    graph.erase()
    points = []

    for x, y in enumerate(data):
        points.append((x / CHUNK, y * 2 + 0.5))

    graph.draw_lines(points, color="red", width=1)

    fft = np.fft.rfft(data, n=FFT_BINS)
    fft = np.abs(fft) * 0.2

    bars.erase()

    for i in range(1, FFT_BINS // 2):
        bars.draw_rectangle((2 * i, 0.0), (2 * (i+1), fft[i]), fill_color="green", line_width=0)

stream.close()
window.close()