Update app.py

app.py

@@ -6,34 +6,24 @@ from flask import Flask, request, jsonify
 from flask_cors import CORS
 from vosk import Model, KaldiRecognizer
 
-# Tạo Flask app
+# Khởi tạo Flask app
 app = Flask(__name__)
 CORS(app)
 
-# Tải model Vosk (dùng model nhỏ để tiết kiệm tài nguyên)
-MODEL_PATH = "model"  # Thư mục chứa model Vosk
+# Đường dẫn model Vosk
+MODEL_PATH = "model"
 if not os.path.exists(MODEL_PATH):
-    raise Exception("Model Vosk không tìm thấy!")
+    raise Exception("Model Vosk không tìm thấy! Đảm bảo đã tải về model đúng thư mục.")
 
 model = Model(MODEL_PATH)
 
-@app.route("/")
-def home():
-    return "Vosk STT API đang chạy!"
-
-@app.route("/stt", methods=["POST"])
-def stt():
-    if "audio" not in request.files:
-        return jsonify({"error": "Không tìm thấy file audio!"}), 400
-
-    audio_file = request.files["audio"]
-    file_path = "temp.wav"
-    audio_file.save(file_path)
-
-    # Xử lý file audio
+def process_audio(file_path):
+    """ Xử lý file audio và trả về văn bản nhận dạng """
     wf = wave.open(file_path, "rb")
-    rec = KaldiRecognizer(model, wf.getframerate())
+    if wf.getnchannels() != 1:
+        return {"error": "Chỉ hỗ trợ file WAV mono (1 kênh)"}
 
+    rec = KaldiRecognizer(model, wf.getframerate())
     result_text = ""
     while True:
         data = wf.readframes(4000)
@@ -42,8 +32,28 @@ def stt():
         if rec.AcceptWaveform(data):
             result_text += json.loads(rec.Result())["text"] + " "
 
-    # Trả về kết quả STT
-    return jsonify({"text": result_text.strip()})
+    wf.close()
+    os.remove(file_path)  # Xóa file sau khi xử lý
+    return {"text": result_text.strip()}
+
+@app.route("/")
+def home():
+    return "✅ Vosk STT API đang chạy! Gửi file âm thanh đến /stt để nhận dạng."
+
+@app.route("/stt", methods=["POST"])
+def stt():
+    if "audio" not in request.files:
+        return jsonify({"error": "Không tìm thấy file audio!"}), 400
+    
+    audio_file = request.files["audio"]
+    if not audio_file.filename.endswith(".wav"):
+        return jsonify({"error": "Chỉ hỗ trợ file WAV!"}), 400
+    
+    file_path = "temp.wav"
+    audio_file.save(file_path)
+    
+    result = process_audio(file_path)
+    return jsonify(result)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host="0.0.0.0", port=7860, debug=True)