Update app.py

app.py

@@ -1,46 +1,38 @@
 import os
-import wave
 import json
 import asyncio
 import websockets
-import uuid
+from fastapi import FastAPI, WebSocket
 from vosk import Model, KaldiRecognizer
 
-# Thư mục chứa model
 MODEL_PATH = "model/vosk-model"
-
-# Kiểm tra model đã tải chưa
 if not os.path.exists(MODEL_PATH):
-    raise Exception(f"\u274C Model Vosk không tìm thấy tại {MODEL_PATH}! Kiểm tra lại.")
+    raise Exception("❌ Model Vosk không tìm thấy!")
 
-# Load model
-print("\u2705 Đang tải model Vosk...")
+print("✅ Đang tải model Vosk...")
 model = Model(MODEL_PATH)
 
-async def recognize_audio(websocket, path):
-    print("🔵 Kết nối WebSocket mới...")
+app = FastAPI()
+
+@app.websocket("/ws")
+async def websocket_endpoint(websocket: WebSocket):
+    await websocket.accept()
     rec = KaldiRecognizer(model, 16000)
     result_text = ""
-    
+
     try:
-        async for message in websocket:
-            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.dumps({"text": text}))
-            else:
-                await websocket.send(json.dumps({"error": "Dữ liệu không hợp lệ"}))
-    except websockets.exceptions.ConnectionClosed:
-        print("🔴 Kết nối WebSocket đã đóng")
+        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")
     finally:
         print(f"✅ Văn bản cuối cùng: {result_text.strip()}")
 
-async def main():
-    async with websockets.serve(recognize_audio, "0.0.0.0", 7860):
-        print("🚀 WebSocket STT server đang chạy trên ws://0.0.0.0:7860")
-        await asyncio.Future()  # Giữ server chạy vô thời hạn
-
 if __name__ == "__main__":
-    asyncio.run(main())
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)