Update app.py

app.py

@@ -1,23 +1,22 @@
 import os
 import json
-import asyncio
-from fastapi import FastAPI, WebSocket, WebSocketDisconnect
+from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from vosk import Model, KaldiRecognizer
-import uvicorn
+import wave
 
 # Kiểm tra model
 MODEL_PATH = "model/vosk-model"
 if not os.path.exists(MODEL_PATH):
-    raise Exception("\u274c Model Vosk không tìm thấy!")
+    raise Exception("❌ Model Vosk không tìm thấy!")
 
-print("\u2705 Đang tải model Vosk...")
+print("✅ Đang tải model Vosk...")
 model = Model(MODEL_PATH)
 
 # Tạo app FastAPI
 app = FastAPI()
 
-# Cấu hình CORS để hỗ trợ WebSocket
+# Cấu hình CORS để hỗ trợ API
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],  # Cho phép tất cả nguồn
@@ -26,30 +25,26 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-@app.websocket("/ws")  # Đường dẫn WebSocket
-async def websocket_endpoint(websocket: WebSocket):
-    await websocket.accept()
-    rec = KaldiRecognizer(model, 16000)
-    result_text = ""
-
-    print("\U0001F535 Client đã kết nối WebSocket!")
-    
+@app.post("/stt")
+async def speech_to_text(audio: UploadFile = File(...)):
+    """Nhận file âm thanh và trả về văn bản."""
     try:
-        while True:
-            message = await websocket.receive_bytes()  # Nhận dữ liệu
-            if rec.AcceptWaveform(message):
-                result = json.loads(rec.Result())
-                text = result.get("text", "")
-                result_text += text + " "
-                await websocket.send_json({"text": text})  # Gửi kết quả
-                print(f"\U0001F4E4 Gửi kết quả: {text}")
-    except WebSocketDisconnect:
-        print("\U0001F534 Client đã ngắt kết nối!")
+        with wave.open(audio.file, "rb") as wf:
+            if wf.getnchannels() != 1 or wf.getsampwidth() != 2 or wf.getframerate() != 16000:
+                raise HTTPException(status_code=400, detail="❌ File âm thanh phải là WAV mono 16kHz 16-bit.")
+            
+            rec = KaldiRecognizer(model, 16000)
+            while True:
+                data = wf.readframes(4000)
+                if len(data) == 0:
+                    break
+                rec.AcceptWaveform(data)
+            
+            result = json.loads(rec.Result())
+            return {"text": result.get("text", "")}
     except Exception as e:
-        print(f"\u274c Lỗi WebSocket: {e}")
-        await websocket.close(code=1011, reason="Lỗi xử lý dữ liệu")
-    finally:
-        print(f"\u2705 Văn bản cuối cùng: {result_text.strip()}")
+        raise HTTPException(status_code=500, detail=f"❌ Lỗi xử lý âm thanh: {str(e)}")
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)