Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import wave
 import json
+import uuid
 import numpy as np
 from flask import Flask, request, jsonify
 from flask_cors import CORS
@@ -12,16 +13,16 @@ MODEL_PATH = "model/vosk-model"
 
 # Kiểm tra model đã tải chưa
 if not os.path.exists(MODEL_PATH):
-    raise Exception(f"❌ Model Vosk không tìm thấy tại {MODEL_PATH}! Kiểm tra lại.")
+    raise Exception(f"\u274C Model Vosk không tìm thấy tại {MODEL_PATH}! Kiểm tra lại.")
 
 # Load model
-print("✅ Đang tải model Vosk...")
+print("\u2705 Đang tải model Vosk...")
 model = Model(MODEL_PATH)
 
 # Khởi tạo Flask app
 app = Flask(__name__)
 CORS(app)
-swagger = Swagger(app)
+Swagger(app)
 
 @app.route("/")
 def home():
@@ -31,7 +32,7 @@ def home():
       200:
         description: API đang chạy
     """
-    return "✅ Vosk STT API đang chạy!"
+    return "\u2705 Vosk STT API đang chạy!"
 
 @app.route("/stt", methods=["POST"])
 def stt():
@@ -61,31 +62,35 @@ def stt():
         return jsonify({"error": "Không tìm thấy file audio!"}), 400
 
     audio_file = request.files["audio"]
-    file_path = "temp.wav"
+    file_path = f"/tmp/{uuid.uuid4()}.wav"  # Lưu vào thư mục tạm để tránh lỗi quyền hạn
     audio_file.save(file_path)
 
-    # Mở file âm thanh
-    wf = wave.open(file_path, "rb")
+    try:
+        # Mở file âm thanh
+        wf = wave.open(file_path, "rb")
 
-    # Kiểm tra file có đúng định dạng WAV mono không
-    if wf.getnchannels() != 1 or wf.getsampwidth() != 2 or wf.getcomptype() != "NONE":
-        os.remove(file_path)
-        return jsonify({"error": "File audio phải là WAV mono PCM!"}), 400
+        # Kiểm tra file có đúng định dạng WAV mono không
+        if wf.getnchannels() != 1 or wf.getsampwidth() != 2 or wf.getcomptype() != "NONE":
+            return jsonify({"error": "File audio phải là WAV mono PCM!"}), 400
 
-    rec = KaldiRecognizer(model, wf.getframerate())
-    result_text = ""
+        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"] + " "
+        while True:
+            data = wf.readframes(4000)
+            if len(data) == 0:
+                break
+            if rec.AcceptWaveform(data):
+                result_text += json.loads(rec.Result())["text"] + " "
 
-    wf.close()
-    os.remove(file_path)  # Xóa file tạm
-
-    return jsonify({"text": result_text.strip()})
+        return jsonify({"text": result_text.strip()})
+    
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+    
+    finally:
+        wf.close()
+        os.remove(file_path)  # Xóa file tạm
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)