Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from flask import Flask, request, jsonify
|
|
| 8 |
app = Flask(__name__)
|
| 9 |
|
| 10 |
# Kiểm tra và load model Vosk
|
| 11 |
-
MODEL_PATH = "model/vosk-model"
|
| 12 |
if not os.path.exists(MODEL_PATH):
|
| 13 |
raise Exception("❌ Model Vosk không tìm thấy!")
|
| 14 |
print("✅ Đang tải model Vosk...")
|
|
@@ -20,10 +20,17 @@ rec = KaldiRecognizer(model, 16000)
|
|
| 20 |
# API nhận dữ liệu âm thanh và trả kết quả nhận diện giọng nói
|
| 21 |
@app.route('/stt', methods=['POST'])
|
| 22 |
def handle_audio():
|
| 23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
print(f"📩 Nhận dữ liệu âm thanh: {len(audio_data)} bytes")
|
| 25 |
-
|
| 26 |
-
# Chuyển đổi dữ liệu âm thanh thành numpy array
|
| 27 |
audio_np = np.frombuffer(audio_data, dtype=np.int16)
|
| 28 |
|
| 29 |
# Nhận diện giọng nói
|
|
|
|
| 8 |
app = Flask(__name__)
|
| 9 |
|
| 10 |
# Kiểm tra và load model Vosk
|
| 11 |
+
MODEL_PATH = "model/vosk-model" # Thay thế với đường dẫn đến mô hình của bạn
|
| 12 |
if not os.path.exists(MODEL_PATH):
|
| 13 |
raise Exception("❌ Model Vosk không tìm thấy!")
|
| 14 |
print("✅ Đang tải model Vosk...")
|
|
|
|
| 20 |
# API nhận dữ liệu âm thanh và trả kết quả nhận diện giọng nói
|
| 21 |
@app.route('/stt', methods=['POST'])
|
| 22 |
def handle_audio():
|
| 23 |
+
# Kiểm tra xem có tệp âm thanh không
|
| 24 |
+
if 'file' not in request.files:
|
| 25 |
+
return jsonify({"error": "Không có tệp âm thanh trong yêu cầu."}), 400
|
| 26 |
+
|
| 27 |
+
audio_file = request.files['file']
|
| 28 |
+
|
| 29 |
+
# Đọc dữ liệu âm thanh từ tệp
|
| 30 |
+
audio_data = audio_file.read()
|
| 31 |
print(f"📩 Nhận dữ liệu âm thanh: {len(audio_data)} bytes")
|
| 32 |
+
|
| 33 |
+
# Chuyển đổi dữ liệu âm thanh thành numpy array (16-bit mono)
|
| 34 |
audio_np = np.frombuffer(audio_data, dtype=np.int16)
|
| 35 |
|
| 36 |
# Nhận diện giọng nói
|