chrisclark · November 27, 2023 14:29 · Nov 27, 2023
diff --git a/speech_visualization.py b/speech_visualization.py
@@ -0,0 +1,72 @@
+%matplotlib qt
+
+import pyaudio
+import numpy as np
+import matplotlib.pyplot as plt
+from queue import Queue
+import threading
+import time
+from IPython.display import display, clear_output
+
+FORMAT = pyaudio.paInt16
+CHANNELS = 1
+RATE = 44100
+CHUNK = 1024
+
+audio_queue = Queue()
+
+def flush_queue():
+    while not audio_queue.empty():
+        audio_queue.get()
+
+def audio_stream():
+    def audio_callback(in_data, frame_count, time_info, status):
+        audio_queue.put(np.frombuffer(in_data, dtype=np.int16))
+        return (in_data, pyaudio.paContinue)
+
+    audio = pyaudio.PyAudio()
+
+    stream = audio.open(format=FORMAT, channels=CHANNELS,
+                        rate=RATE, input=True,
+                        frames_per_buffer=CHUNK,
+                        input_device_index=AUDIO_DEVICE,
+                        stream_callback=audio_callback)
+
+    stream.start_stream()
+
+    while stream.is_active():
+        time.sleep(0.1)
+
+    stream.stop_stream()
+    stream.close()
+    audio.terminate()
+
+# Start audio stream in a separate thread
+thread = threading.Thread(target=audio_stream)
+thread.daemon = True
+thread.start()
+
+# Initialize Matplotlib plot
+fig, ax = plt.subplots()
+x = np.arange(0, 2 * CHUNK, 2)
+line, = ax.plot(x, np.zeros(CHUNK))
+ax.set_ylim(-16000, 16000)
+ax.set_xlim(0, CHUNK)
+ax.axis('off')
+
+def update_plot():
+    try:
+        while True:
+            if not audio_queue.empty():
+                data = audio_queue.get()
+                flush_queue()
+                line.set_ydata(data)
+                ax.relim()  # Recompute the ax.dataLim
+                ax.autoscale_view()  # Update axes limits
+                fig.canvas.draw()  # Redraw the figure
+                fig.canvas.flush_events()  # Process GUI events
+            time.sleep(0.1)
+    except KeyboardInterrupt:
+        pass
+
+update_plot()