Update app.py

app.py

@@ -16,8 +16,8 @@ model = Model(MODEL_PATH)
 
 # Khởi tạo Flask app
 app = Flask(__name__)
-CORS(app)  # Cho phép CORS để React app truy cập
-Swagger(app)  # Khởi tạo Swagger cho tài liệu API
+CORS(app)
+Swagger(app)
 
 @app.route("/")
 def home():
@@ -55,7 +55,6 @@ def stt():
       500:
         description: Lỗi server nội bộ
     """
-    # Kiểm tra trường 'file' để khớp với React
     if "file" not in request.files:
         return jsonify({"error": "Không tìm thấy file âm thanh! Vui lòng gửi trường 'file'."}), 400
 
@@ -72,7 +71,13 @@ def stt():
     wav_path = None
     wf = None
     try:
-        # Chuyển đổi WebM sang WAV mono PCM bằng ffmpeg
+        # Kiểm tra xem FFmpeg có sẵn không
+        try:
+            ffmpeg.probe(webm_path)
+        except Exception as e:
+            return jsonify({"error": f"FFmpeg không khả dụng: {str(e)}"}), 500
+
+        # Chuyển đổi WebM sang WAV mono PCM
         wav_path = tempfile.mktemp(suffix=".wav")
         stream = ffmpeg.input(webm_path)
         stream = ffmpeg.output(
@@ -80,7 +85,8 @@ def stt():
             wav_path,
             acodec="pcm_s16le",  # PCM 16-bit signed little-endian
             ac=1,               # Mono
-            ar=16000            # Tần số mẫu 16kHz, phù hợp với Vosk
+            ar=16000,           # Tần số mẫu 16kHz, phù hợp với Vosk
+            format="wav"
         )
         ffmpeg.run(stream, overwrite_output=True, quiet=True)
 
@@ -104,7 +110,6 @@ def stt():
                 result = json.loads(rec.Result())
                 result_text += result.get("text", "") + " "
             else:
-                # Thêm kết quả từng phần để cải thiện độ chính xác
                 partial_result = json.loads(rec.PartialResult())
                 if partial_result.get("partial", ""):
                     result_text += partial_result["partial"] + " "
@@ -116,7 +121,8 @@ def stt():
         return jsonify({"text": final_text})
 
     except ffmpeg.Error as e:
-        return jsonify({"error": f"Lỗi chuyển đổi âm thanh từ WebM sang WAV: {str(e.stderr.decode())}"}), 500
+        error_message = e.stderr.decode('utf-8') if e.stderr else str(e)
+        return jsonify({"error": f"Lỗi chuyển đổi âm thanh từ WebM sang WAV: {error_message}"}), 500
     except Exception as e:
         return jsonify({"error": f"Lỗi xử lý âm thanh: {str(e)}"}), 500