increase sessions cleanup

app.py

@@ -136,6 +136,8 @@ class UserSession:
         self.last_audio_time = None
         self._start_lock = threading.Lock()
         self._task = None  # Track the async task
+        self._websocket = None  # Store websocket for forced closure
+        self._stopped_by_user = False  # Track if user explicitly stopped
     
     @property
     def audio_queue(self):
@@ -297,11 +299,15 @@ async def send_silence(ws, duration=2.0):
 
 async def websocket_handler(session):
     """Connect to WebSocket and handle audio streaming + transcription."""
+    ws = None
     try:
         # Add connection timeout to prevent hanging
         async with asyncio.timeout(10):  # 10 second connection timeout
             ws = await websockets.connect(ws_url)
         
+        # Store websocket reference so it can be closed externally
+        session._websocket = ws
+        
         async with ws:
             await asyncio.wait_for(ws.recv(), timeout=5)
             await ws.send(json.dumps({"type": "session.update", "model": model}))
@@ -393,11 +399,15 @@ async def websocket_handler(session):
         session.status_message = "error"
     finally:
         session.is_running = False
-        # Remove from active sessions
-        with _sessions_lock:
-            _active_sessions.pop(session.session_id, None)
-            active_count = len(_active_sessions)
-        print(f"Session {session.session_id[:8]} ended. Active sessions: {active_count}")
+        session._websocket = None
+        
+        # Only remove and log if not already handled by stop_session
+        if not session._stopped_by_user:
+            with _sessions_lock:
+                removed = _active_sessions.pop(session.session_id, None)
+                active_count = len(_active_sessions)
+            if removed:
+                print(f"Session {session.session_id[:8]} ended. Active sessions: {active_count}")
 
 
 def start_websocket(session):
@@ -481,6 +491,16 @@ def stop_session(session_id):
     if session.is_running:
         session.is_running = False
         session.last_audio_time = None
+        session._stopped_by_user = True  # Mark as user-stopped to avoid duplicate logging
+        
+        # Close the websocket immediately to force cleanup
+        if session._websocket is not None:
+            loop = get_event_loop()
+            try:
+                asyncio.run_coroutine_threadsafe(session._websocket.close(), loop)
+            except Exception:
+                pass  # Ignore errors during close
+            session._websocket = None
         
         # Cancel the running task if any
         if session._task is not None:
@@ -507,6 +527,16 @@ def clear_history(session_id):
     session = ensure_session(session_id)
     session.is_running = False
     session.last_audio_time = None
+    session._stopped_by_user = True  # Mark as user-stopped
+    
+    # Close the websocket immediately
+    if session._websocket is not None:
+        loop = get_event_loop()
+        try:
+            asyncio.run_coroutine_threadsafe(session._websocket.close(), loop)
+        except Exception:
+            pass
+        session._websocket = None
     
     # Cancel the running task if any
     if session._task is not None: