Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 from fastapi import FastAPI, WebSocket
+from fastapi.middleware.cors import CORSMiddleware
 from vosk import Model, KaldiRecognizer
 import uvicorn
 
@@ -15,7 +16,16 @@ model = Model(MODEL_PATH)
 # Tạo app FastAPI
 app = FastAPI()
 
-@app.websocket("/ws")
+# Cấu hình CORS để hỗ trợ WebSocket
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Cho phép tất cả nguồn
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.websocket("/")  # WebSocket endpoint phải là "/" trên Hugging Face
 async def websocket_endpoint(websocket: WebSocket):
     await websocket.accept()
     rec = KaldiRecognizer(model, 16000)
@@ -23,20 +33,16 @@ async def websocket_endpoint(websocket: WebSocket):
 
     try:
         while True:
-            message = await websocket.receive()
-            
-            if isinstance(message, bytes):  # Kiểm tra nếu dữ liệu là bytes
-                if rec.AcceptWaveform(message):
-                    result = json.loads(rec.Result())
-                    text = result.get("text", "")
-                    result_text += text + " "
-                    await websocket.send_json({"text": text})
-            else:
-                await websocket.send_json({"error": "Dữ liệu không hợp lệ"})
+            message = await websocket.receive_bytes()  # Nhận bytes từ WebSocket
+            if rec.AcceptWaveform(message):
+                result = json.loads(rec.Result())
+                text = result.get("text", "")
+                result_text += text + " "
+                await websocket.send_json({"text": text})
     except Exception as e:
         print(f"🔴 Kết nối WebSocket đóng: {e}")
     finally:
         print(f"✅ Văn bản cuối cùng: {result_text.strip()}")
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)