app adjusts

app.py

@@ -1,19 +1,48 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, WebSocket
+from contextlib import asynccontextmanager
 import uvicorn
 from whisper_live.server import TranscriptionServer
+import logging
 
-app = FastAPI(title="Whisper Live Server")
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-@app.on_event("startup")
-async def startup_event():
-    # Start the transcription server in the background
-    server = TranscriptionServer()
-    server.run(
-        host="0.0.0.0",
-        port=7860,  # Hugging Face Spaces uses port 7860
-        backend="faster_whisper",  # Using faster_whisper as the backend
-        single_model=True  # Use single model mode for better resource usage
-    )
+# Initialize the transcription server
+transcription_server = TranscriptionServer()
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Any startup configuration can go here
+    yield
+    # Cleanup can go here if needed
+
+app = FastAPI(title="Whisper Live Server", lifespan=lifespan)
+
+@app.websocket("/ws")
+async def websocket_endpoint(websocket: WebSocket):
+    await websocket.accept()
+    try:
+        # Start the transcription server with the WebSocket connection
+        transcription_server.run(
+            host="0.0.0.0",
+            port=7860,  # Hugging Face Spaces uses port 7860
+            backend="faster_whisper",  # Using faster_whisper as the backend
+            single_model=True  # Use single model mode for better resource usage
+        )
+        
+        # Handle WebSocket communication
+        while True:
+            data = await websocket.receive()
+            if data.get("type") == "websocket.disconnect":
+                break
+            # Process the received data through the transcription server
+            # Note: You might need to adapt this part based on your specific needs
+            await websocket.send_text(data)
+    except Exception as e:
+        logger.error(f"WebSocket error: {e}")
+    finally:
+        await websocket.close()
 
 @app.get("/health")
 def health_check():

requirements.txt

@@ -1,9 +1,17 @@
-faster-whisper
-numpy
-websockets
-pyaudio
+fastapi>=0.68.0
+uvicorn>=0.15.0
+websockets>=10.0
+numpy>=1.21.0
+faster-whisper==1.1.0
+PyAudio
+websocket-client
+numba
+kaldialign
 soundfile
-torch
-torchaudio
-fastapi
-uvicorn[standard] 
+tokenizers==0.20.3
+librosa
+openvino
+openvino-genai
+openvino-tokenizers
+optimum
+optimum-intel