error handling

app.py

@@ -333,6 +333,13 @@ def start_websocket(session):
     # Cleanup happens in websocket_handler's finally block
 
 
+def ensure_session(session):
+    """Ensure we have a valid UserSession instance (not the lambda factory)."""
+    if session is None or callable(session):
+        return UserSession()
+    return session
+
+
 def auto_start_recording(session):
     """Automatically start the transcription service when audio begins."""
     # Protect against startup races: Gradio can call `process_audio` concurrently.
@@ -361,6 +368,7 @@ def auto_start_recording(session):
 
 def clear_history(session):
     """Stop the websocket connection and clear all history."""
+    session = ensure_session(session)
     session.is_running = False
     session.last_audio_time = None
     
@@ -387,6 +395,7 @@ def clear_history(session):
 
 def process_audio(audio, session):
     """Process incoming audio and queue for streaming."""
+    session = ensure_session(session)
     try:
         # Quick return if audio is None
         if audio is None:
@@ -439,7 +448,11 @@ def process_audio(audio, session):
         return get_transcription_html(session.transcription_text, session.status_message, session.current_wpm)
     except Exception as e:
         print(f"Error processing audio: {e}")
-        return get_transcription_html(session.transcription_text, session.status_message, session.current_wpm)
+        # Safely get session attributes with fallbacks
+        text = getattr(session, 'transcription_text', '') if not callable(session) else ''
+        status = getattr(session, 'status_message', 'error') if not callable(session) else 'error'
+        wpm = getattr(session, 'current_wpm', '') if not callable(session) else ''
+        return get_transcription_html(text, status, wpm)
 
 
 def _safe_queue_put(q, item):