Adds fix to flush queue on ctrl + c.

This change mitigates missing the cached caption for any final speech when user presses ctrl + c.  Removes redundant np.array() call.

Author: guynich

moonshine/demo/live_captions.py

@@ -1,9 +1,9 @@
 """Live captions from microphone using Moonshine and SileroVAD ONNX models."""
 
+import argparse
 import os
 import sys
 import time
-import argparse
 
 from queue import Queue
 
@@ -67,7 +67,7 @@ def create_input_callback(q):
     def input_callback(data, frames, time, status):
         if status:
             print(status)
-        q.put((data.copy(), status))
+        q.put((data.copy().flatten(), status))
 
     return input_callback
 
@@ -144,12 +144,10 @@ def print_captions(text, new_cached_caption=False):
         print_captions("Ready...")
         try:
             while True:
-                data, status = q.get()
+                chunk, status = q.get()
                 if VERBOSE and status:
                     print(status)
 
-                chunk = np.array(data).flatten()
-
                 speech = np.concatenate((speech, chunk))
                 if not recording:
                     speech = speech[-lookback_size:]
@@ -182,6 +180,12 @@ def print_captions(text, new_cached_caption=False):
         except KeyboardInterrupt:
             stream.close()
 
+            if recording:
+                while not q.empty():
+                    chunk, _ = q.get()
+                    speech = np.concatenate((speech, chunk))
+                end_recording(speech, "<END.>")
+
             print(f"""
 
              model_name :  {model_name}