Persist live session receive loop by wrapping session.receive iterator in a while True loop to support multi-turn dialogues

backend/main.py

@@ -323,36 +323,43 @@ async def websocket_live_endpoint(websocket: WebSocket):
                     ws_log("Client WebSocket disconnected (WebSocketDisconnect).")
                 except Exception as e:
                     ws_log(f"[WebSocket Proxy Client -> Gemini] Error: {e}")
+                finally:
+                    ws_log("receive_from_client loop exited.")
 
             async def send_to_client():
                 try:
-                    async for response in session.receive():
-                        server_content = response.server_content
-                        if server_content is not None:
-                            model_turn = server_content.model_turn
-                            if model_turn is not None:
-                                for part in model_turn.parts:
-                                    if part.inline_data is not None:
-                                        # Stream PCM audio output back to client as Base64
-                                        audio_b64 = base64.b64encode(part.inline_data.data).decode('utf-8')
-                                        await websocket.send_json({
-                                            "type": "audio",
-                                            "data": audio_b64
-                                        })
-                                    elif part.text is not None:
-                                        # Stream text transcription back to client
-                                        ws_log(f"Streaming text chunk from Gemini: {part.text}")
-                                        await websocket.send_json({
-                                            "type": "text",
-                                            "data": part.text
-                                        })
-                            
-                            # Handle turn completion (model finished speaking)
-                            if server_content.turn_complete:
-                                ws_log("Gemini sent turn_complete.")
-                                await websocket.send_json({"type": "turn_complete"})
+                    while True:
+                        async for response in session.receive():
+                            server_content = response.server_content
+                            if server_content is not None:
+                                model_turn = server_content.model_turn
+                                if model_turn is not None:
+                                    for part in model_turn.parts:
+                                        if part.inline_data is not None:
+                                            # Stream PCM audio output back to client as Base64
+                                            audio_b64 = base64.b64encode(part.inline_data.data).decode('utf-8')
+                                            await websocket.send_json({
+                                                "type": "audio",
+                                                "data": audio_b64
+                                            })
+                                        elif part.text is not None:
+                                            # Stream text transcription back to client
+                                            ws_log(f"Streaming text chunk from Gemini: {part.text}")
+                                            await websocket.send_json({
+                                                "type": "text",
+                                                "data": part.text
+                                            })
+                                
+                                # Handle turn completion (model finished speaking)
+                                if server_content.turn_complete:
+                                    ws_log("Gemini sent turn_complete.")
+                                    await websocket.send_json({"type": "turn_complete"})
+                        # Avoid tight loop if iterator finishes instantly
+                        await asyncio.sleep(0.1)
                 except Exception as e:
                     ws_log(f"[WebSocket Proxy Gemini -> Client] Error: {e}")
+                finally:
+                    ws_log("send_to_client loop exited.")
 
             # Run both tasks concurrently and terminate when the first one finishes
             done, pending = await asyncio.wait(