Add per-session language proxy

whisperlivekit/session_asr_proxy.py

@@ -0,0 +1,41 @@
+"""Per-session ASR proxy for language override.
+
+Wraps a shared ASR backend so that each WebSocket session can use a
+different transcription language without modifying the shared instance.
+"""
+
+import threading
+
+
+class SessionASRProxy:
+    """Wraps a shared ASR backend with a per-session language override.
+
+    The proxy delegates all attribute access to the wrapped ASR except
+    ``transcribe()``, which temporarily overrides ``original_language``
+    on the shared ASR (under a lock) so the correct language is used.
+
+    Thread-safety: a per-ASR lock serializes ``transcribe()`` calls,
+    which is acceptable because model inference is typically GPU-bound
+    and cannot be parallelized anyway.
+    """
+
+    def __init__(self, asr, language: str):
+        object.__setattr__(self, '_asr', asr)
+        object.__setattr__(self, '_session_language', None if language == "auto" else language)
+        # Attach a shared lock to the ASR instance (created once, reused by all proxies)
+        if not hasattr(asr, '_session_lock'):
+            asr._session_lock = threading.Lock()
+        object.__setattr__(self, '_lock', asr._session_lock)
+
+    def __getattr__(self, name):
+        return getattr(self._asr, name)
+
+    def transcribe(self, audio, init_prompt=""):
+        """Call the backend's transcribe with the session's language."""
+        with self._lock:
+            saved = self._asr.original_language
+            self._asr.original_language = self._session_language
+            try:
+                return self._asr.transcribe(audio, init_prompt=init_prompt)
+            finally:
+                self._asr.original_language = saved