From 3ba5cfffe60828a451c941ad6384f2213fc20e03 Mon Sep 17 00:00:00 2001
From: Randy Mackay <rmackay9@yahoo.com>
Date: Tue, 6 Aug 2024 16:05:05 +0900
Subject: [PATCH] chat: use queue to send stop-recording-cmd

---
 MAVProxy/modules/mavproxy_chat/chat_voice_to_text.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/MAVProxy/modules/mavproxy_chat/chat_voice_to_text.py b/MAVProxy/modules/mavproxy_chat/chat_voice_to_text.py
index d03076b882..eb77824c08 100644
--- a/MAVProxy/modules/mavproxy_chat/chat_voice_to_text.py
+++ b/MAVProxy/modules/mavproxy_chat/chat_voice_to_text.py
@@ -15,6 +15,7 @@
     print("chat: failed to import pyaudio, wave or openai.  See https://ardupilot.org/mavproxy/docs/modules/chat.html")
     exit()
 
+from MAVProxy.modules.lib import multiproc
 
 class chat_voice_to_text():
     def __init__(self):
@@ -22,6 +23,7 @@ def __init__(self):
         self.client = None
         self.assistant = None
         self.stop_recording = False
+        self.external_cmd_queue = multiproc.Queue()
 
     # set the OpenAI API key
     def set_api_key(self, api_key_str):
@@ -66,6 +68,10 @@ def record_audio(self):
             data = stream.read(1024)
             frames.append(data)
             curr_time = time.time()
+            queue_item = self.external_cmd_queue.get_nowait()
+            if queue_item == "stop":
+                print("chat: voice-to-text got stop command!")
+                break
 
         # Stop and close the stream
         stream.stop_stream()
@@ -85,6 +91,7 @@ def record_audio(self):
     def stop_record_audio(self):
         print("chat: voice-to-text will stop recording!")
         self.stop_recording = True
+        self.external_cmd_queue.put("stop")
 
     # convert audio to text
     # returns transcribed text on success or None if failed