fix "not working on HF Spaces but working locally"

app/main.py

@@ -5,6 +5,27 @@ from app.asr_worker import create_recognizer, stream_audio, finalize_stream
 
 app = FastAPI()
 
+@app.websocket("/ws")
+async def websocket_endpoint(websocket: WebSocket):
+    await websocket.accept()
+    print("[INFO] WebSocket connection accepted.")
+    stream = recognizer.create_stream()
+
+    try:
+        while True:
+            data = await websocket.receive_bytes()
+            print(f"[DEBUG] Received {len(data)} bytes")
+            result, rms = stream_audio(data, stream, recognizer)
+            await websocket.send_json({
+                "partial": result,
+                "volume": min(rms * 5.0, 1.0)
+            })
+    except Exception as e:
+        print(f"[ERROR] {e}")
+        final = finalize_stream(stream, recognizer)
+        await websocket.send_json({"final": final})
+        await websocket.close()
+
 app.mount("/static", StaticFiles(directory="app/static"), name="static")
 
 recognizer = create_recognizer()

app/static/index.html

@@ -9,7 +9,7 @@
   <p>Partial: <span id="partial"></span></p>
   <p>Final: <b id="final"></b></p>
   <script>
-    const ws = new WebSocket("ws://" + location.host + "/ws");
+    const ws = new WebSocket("wss://" + location.host + "/ws");
     const vol = document.getElementById("vol");
     const partial = document.getElementById("partial");
     const finalText = document.getElementById("final");