Consolidate clean up tasks for voice audio resources

packages/embed-react/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@humeai/voice-embed-react",
-  "version": "0.2.0-beta.3",
+  "version": "0.2.0-beta.4",
   "description": "",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",

packages/embed/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@humeai/voice-embed",
-  "version": "0.2.0-beta.3",
+  "version": "0.2.0-beta.4",
   "description": "",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",

packages/react/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@humeai/voice-react",
-  "version": "0.2.0-beta.3",
+  "version": "0.2.0-beta.4",
   "description": "",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",

packages/react/src/lib/VoiceProvider.tsx

@@ -200,6 +200,8 @@ export const VoiceProvider: FC<VoiceProviderProps> = ({
 
   const config = props;
 
+  const micCleanUpFnRef = useRef<null | (() => void)>(null);
+
   const player = useSoundPlayer({
     onError: (message) => {
       updateError({ type: 'audio_error', message });
@@ -213,6 +215,28 @@ export const VoiceProvider: FC<VoiceProviderProps> = ({
     },
   });
 
+  const handleResourceCleanup = useCallback(() => {
+    player.stopAll();
+    if (micCleanUpFnRef.current !== null) {
+      micCleanUpFnRef.current();
+    }
+    if (clearMessagesOnDisconnect) {
+      messageStore.clearMessages();
+    }
+    toolStatus.clearStore();
+    setIsPaused(false);
+
+    if (status.value !== 'error') {
+      setStatus({ value: 'disconnected' });
+    }
+  }, [
+    player,
+    clearMessagesOnDisconnect,
+    messageStore,
+    toolStatus,
+    status.value,
+  ]);
+
   const { streamRef, getStream, permission: micPermission } = useEncoding();
 
   const client = useVoiceClient({
@@ -257,9 +281,10 @@ export const VoiceProvider: FC<VoiceProviderProps> = ({
       (event) => {
         stopTimer();
         messageStore.createDisconnectMessage(event);
+        handleResourceCleanup();
         onClose.current?.(event);
       },
-      [messageStore, stopTimer],
+      [messageStore, stopTimer, handleResourceCleanup],
     ),
     onToolCall: props.onToolCall,
   });
@@ -295,6 +320,10 @@ export const VoiceProvider: FC<VoiceProviderProps> = ({
     ),
   });
 
+  useEffect(() => {
+    micCleanUpFnRef.current = mic.stop;
+  }, [mic]);
+
   const { clearQueue } = player;
 
   const pauseAssistant = useCallback(() => {
@@ -373,21 +402,8 @@ export const VoiceProvider: FC<VoiceProviderProps> = ({
     if (client.readyState !== VoiceReadyState.CLOSED) {
       client.disconnect();
     }
-    player.stopAll();
-    mic.stop();
-    if (clearMessagesOnDisconnect) {
-      messageStore.clearMessages();
-    }
-    toolStatus.clearStore();
-    setIsPaused(false);
-  }, [
-    client,
-    player,
-    mic,
-    clearMessagesOnDisconnect,
-    toolStatus,
-    messageStore,
-  ]);
+    handleResourceCleanup();
+  }, [client, handleResourceCleanup]);
 
   const disconnect = useCallback(
     (disconnectOnError?: boolean) => {