Update app.py

app.py

@@ -94,12 +94,24 @@ class GeminiHandler(AsyncAudioVideoStreamHandler):
                 turn = self.session.receive()
                 try:
                     async for response in turn:
+                        # Check if data exists before trying to process it as audio
                         if data := response.data:
                             audio = np.frombuffer(data, dtype=np.int16).reshape(1, -1)
-                        self.audio_queue.put_nowait(audio)
+                            self.audio_queue.put_nowait(audio) # Only put if audio was created
+                        # You might want to handle other parts of the response here
+                        # e.g., response.text, response.tool_code, etc.
+                        # For now, we just ensure we don't crash if data is None.
+
                 except websockets.exceptions.ConnectionClosedOK:
                     print("connection closed")
                     break
+                except Exception as e:
+                    # Catch other potential errors during response processing
+                    print(f"Error processing response: {e}")
+                    # Depending on the error, you might want to break or continue
+                    # For now, let's break to prevent infinite loops on persistent errors
+                    break
+
 
     async def video_receive(self, frame: np.ndarray):
         self.video_queue.put_nowait(frame)
@@ -124,8 +136,17 @@ class GeminiHandler(AsyncAudioVideoStreamHandler):
         _, array = frame
         array = array.squeeze()
         audio_message = encode_audio(array)
-        if self.session:
-            await self.session.send(input=audio_message)
+        # Add a check to ensure the session is still active before sending
+        if self.session and not self.session._ws.closed: # Check if session exists and websocket is not closed
+            try:
+                await self.session.send(input=audio_message)
+            except websockets.exceptions.ConnectionClosedOK:
+                print("Attempted to send on a closed connection.")
+            except Exception as e:
+                print(f"Error sending audio message: {e}")
+        else:
+            print("Session not active, cannot send audio message.")
+
 
     async def emit(self):
         array = await wait_for_item(self.audio_queue, 0.01)