Update app.py

app.py

@@ -1,10 +1,10 @@
 import os
 import json
-import asyncio
-import websockets
 from fastapi import FastAPI, WebSocket
 from vosk import Model, KaldiRecognizer
+import uvicorn
 
+# Kiểm tra model
 MODEL_PATH = "model/vosk-model"
 if not os.path.exists(MODEL_PATH):
     raise Exception("❌ Model Vosk không tìm thấy!")
@@ -12,6 +12,7 @@ if not os.path.exists(MODEL_PATH):
 print("✅ Đang tải model Vosk...")
 model = Model(MODEL_PATH)
 
+# Tạo app FastAPI
 app = FastAPI()
 
 @app.websocket("/ws")
@@ -22,17 +23,20 @@ async def websocket_endpoint(websocket: WebSocket):
 
     try:
         while True:
-            message = await websocket.receive_bytes()
-            if rec.AcceptWaveform(message):
-                result = json.loads(rec.Result())
-                text = result.get("text", "")
-                result_text += text + " "
-                await websocket.send_json({"text": text})
-    except Exception:
-        print("🔴 Kết nối WebSocket đóng")
+            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ệ"})
+    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__":
-    import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)