Create app.py

app.py

@@ -0,0 +1,49 @@
+import os
+import wave
+import json
+import numpy as np
+from flask import Flask, request, jsonify
+from flask_cors import CORS
+from vosk import Model, KaldiRecognizer
+
+# 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
+if not os.path.exists(MODEL_PATH):
+    raise Exception("Model Vosk không tìm thấy!")
+
+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
+    wf = wave.open(file_path, "rb")
+    rec = KaldiRecognizer(model, wf.getframerate())
+
+    result_text = ""
+    while True:
+        data = wf.readframes(4000)
+        if len(data) == 0:
+            break
+        if rec.AcceptWaveform(data):
+            result_text += json.loads(rec.Result())["text"] + " "
+
+    # Trả về kết quả STT
+    return jsonify({"text": result_text.strip()})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860, debug=True)