change app.py

app.py

@@ -522,11 +522,14 @@ def auto_start_recording(session):
         if session.is_running:
             return get_transcription_html(session.transcription_text, session.status_message, session.current_wpm)
         
-        # Check if we've hit max concurrent sessions
+        # Check if we've hit max concurrent sessions - kill all if so
         with _sessions_lock:
-            if len(_active_sessions) >= MAX_CONCURRENT_SESSIONS:
-                session.status_message = "error"
-                return get_transcription_html("Server at capacity. Please try again later.", "error", "")
+            at_capacity = len(_active_sessions) >= MAX_CONCURRENT_SESSIONS
+        
+        if at_capacity:
+            kill_all_sessions()
+            session.status_message = "error"
+            return get_transcription_html("Server reset due to capacity. Please click the microphone to restart.", "error", "")
         
         session.transcription_text = ""
         session.word_timestamps = []
@@ -625,18 +628,21 @@ def clear_history(session_id):
 
 def process_audio(audio, session_id):
     """Process incoming audio and queue for streaming."""
-    # Check capacity - if at max, kill ALL sessions to reset
+    # Check capacity - if at or above max, kill ALL sessions to reset
     with _sessions_lock:
-        at_capacity = len(_active_sessions) >= MAX_CONCURRENT_SESSIONS
+        current_count = len(_active_sessions)
+        # Check if THIS user is already active (they can continue)
+        is_active_user = session_id and any(s.session_id == session_id for s in _active_sessions.values())
     
-    if at_capacity:
+    # Kill all if at capacity AND this is a new user trying to join
+    # OR if we've exceeded capacity (shouldn't happen, but safety check)
+    if current_count > MAX_CONCURRENT_SESSIONS or (current_count >= MAX_CONCURRENT_SESSIONS and not is_active_user):
         kill_all_sessions()
-        # Return error to this user - they can retry immediately
         return get_transcription_html(
             "Server reset due to capacity. Please click the microphone to restart.", 
             "error", 
             ""
-        ), None  # Return None to force fresh session on retry
+        ), None
     
     # Always ensure we have a valid session first
     try: