|
23 | 23 | NOTIFICATION_TEST, |
24 | 24 | REQUEST_REGION_GRID, |
25 | 25 | UPDATE_AUDIO_ACTIVITY, |
| 26 | + UPDATE_AUDIO_TRANSCRIPTION_STATE, |
26 | 27 | UPDATE_BIRDSEYE_LAYOUT, |
27 | 28 | UPDATE_CAMERA_ACTIVITY, |
28 | 29 | UPDATE_EMBEDDINGS_REINDEX_PROGRESS, |
@@ -61,6 +62,7 @@ def __init__( |
61 | 62 | self.model_state: dict[str, ModelStatusTypesEnum] = {} |
62 | 63 | self.embeddings_reindex: dict[str, Any] = {} |
63 | 64 | self.birdseye_layout: dict[str, Any] = {} |
| 65 | + self.audio_transcription_state: str = "idle" |
64 | 66 | self._camera_settings_handlers: dict[str, Callable] = { |
65 | 67 | "audio": self._on_audio_command, |
66 | 68 | "audio_transcription": self._on_audio_transcription_command, |
@@ -178,6 +180,19 @@ def handle_update_model_state() -> None: |
178 | 180 | def handle_model_state() -> None: |
179 | 181 | self.publish("model_state", json.dumps(self.model_state.copy())) |
180 | 182 |
|
| 183 | + def handle_update_audio_transcription_state() -> None: |
| 184 | + if payload: |
| 185 | + self.audio_transcription_state = payload |
| 186 | + self.publish( |
| 187 | + "audio_transcription_state", |
| 188 | + json.dumps(self.audio_transcription_state), |
| 189 | + ) |
| 190 | + |
| 191 | + def handle_audio_transcription_state() -> None: |
| 192 | + self.publish( |
| 193 | + "audio_transcription_state", json.dumps(self.audio_transcription_state) |
| 194 | + ) |
| 195 | + |
181 | 196 | def handle_update_embeddings_reindex_progress() -> None: |
182 | 197 | self.embeddings_reindex = payload |
183 | 198 | self.publish( |
@@ -264,10 +279,12 @@ def handle_notification_test() -> None: |
264 | 279 | UPDATE_MODEL_STATE: handle_update_model_state, |
265 | 280 | UPDATE_EMBEDDINGS_REINDEX_PROGRESS: handle_update_embeddings_reindex_progress, |
266 | 281 | UPDATE_BIRDSEYE_LAYOUT: handle_update_birdseye_layout, |
| 282 | + UPDATE_AUDIO_TRANSCRIPTION_STATE: handle_update_audio_transcription_state, |
267 | 283 | NOTIFICATION_TEST: handle_notification_test, |
268 | 284 | "restart": handle_restart, |
269 | 285 | "embeddingsReindexProgress": handle_embeddings_reindex_progress, |
270 | 286 | "modelState": handle_model_state, |
| 287 | + "audioTranscriptionState": handle_audio_transcription_state, |
271 | 288 | "birdseyeLayout": handle_birdseye_layout, |
272 | 289 | "onConnect": handle_on_connect, |
273 | 290 | } |
|
0 commit comments