freddyaboulton HF staff commited on
Commit
9cc847c
1 Parent(s): 3ce0cf6
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -133,6 +133,16 @@ class OpenAIHandler(StreamHandler):
133
  except StopIteration:
134
  self._generator = self.generator()
135
  return None
 
 
 
 
 
 
 
 
 
 
136
 
137
  def shutdown(self) -> None:
138
  if self.connection:
@@ -140,6 +150,7 @@ class OpenAIHandler(StreamHandler):
140
  self.quit.set()
141
  if self.thread:
142
  self.thread.join(timeout=5)
 
143
 
144
 
145
  def update_chatbot(chatbot: list[dict], response: ResponseAudioTranscriptDoneEvent):
 
133
  except StopIteration:
134
  self._generator = self.generator()
135
  return None
136
+
137
+ def reset_state(self):
138
+ """Reset connection state for new recording session"""
139
+ self.connection = None
140
+ self.args_set.clear()
141
+ self.quit.clear()
142
+ self.connected.clear()
143
+ self.thread = None
144
+ self._generator = None
145
+ self.current_session = None
146
 
147
  def shutdown(self) -> None:
148
  if self.connection:
 
150
  self.quit.set()
151
  if self.thread:
152
  self.thread.join(timeout=5)
153
+ self.reset_state()
154
 
155
 
156
  def update_chatbot(chatbot: list[dict], response: ResponseAudioTranscriptDoneEvent):