Close connections properly

handler.py

@@ -97,14 +97,24 @@ class AudioLoop:
                 await self.startup(tools)
 
                 async with asyncio.TaskGroup() as tg:
-                    tg.create_task(self.send_realtime())
-                    tg.create_task(self.receive_audio())
-
-                    # Keep running until canceled
-                    await asyncio.Future()
+                    send_task = tg.create_task(self.send_realtime())
+                    receive_task = tg.create_task(self.receive_audio())
+                    await asyncio.Future()  # Keep running until canceled
 
         except asyncio.CancelledError:
-            pass
+            print("[AudioLoop] Received cancel signal, cleaning up...")
+            # Cancel our tasks if they're still running
+            if 'send_task' in locals():
+                send_task.cancel()
+            if 'receive_task' in locals():
+                receive_task.cancel()
+            # Close Gemini connection if we have one
+            if self.ws and not self.ws.closed:
+                try:
+                    await self.ws.close()
+                except Exception as e:
+                    print(f"[AudioLoop] Error closing Gemini connection: {e}")
+            print("[AudioLoop] Cleanup complete")
         except Exception as e:
             traceback.print_exc()
             raise