Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
 import wave
 import json
-import uuid
 import tempfile
 import ffmpeg
 from flask import Flask, request, jsonify
@@ -67,10 +66,24 @@ def stt():
         webm_path = temp_webm_file.name
         audio_file.save(webm_path)
 
+    # Kiểm tra kích thước tệp
+    if os.path.getsize(webm_path) < 100:
+        os.remove(webm_path)
+        return jsonify({"error": "Tệp âm thanh quá nhỏ hoặc rỗng!"}), 400
+
     # Đường dẫn file WAV tạm thời sau khi chuyển đổi
     wav_path = None
     wf = None
     try:
+        # Kiểm tra tệp WebM bằng ffprobe trước khi chuyển đổi
+        try:
+            probe = ffmpeg.probe(webm_path)
+            if 'streams' not in probe or not any(s['codec_type'] == 'audio' for s in probe['streams']):
+                raise ValueError("Tệp không chứa luồng âm thanh hợp lệ!")
+        except ffmpeg.Error as e:
+            error_message = e.stderr.decode('utf-8') if e.stderr else str(e)
+            return jsonify({"error": f"Lỗi kiểm tra tệp WebM: {error_message}"}), 500
+
         # Chuyển đổi WebM sang WAV mono PCM
         wav_path = tempfile.mktemp(suffix=".wav")
         stream = ffmpeg.input(webm_path)
@@ -82,6 +95,7 @@ def stt():
             ar=16000,           # Tần số mẫu 16kHz, phù hợp với Vosk
             format="wav"
         )
+        # Thêm cờ -vn để bỏ qua video nếu có và -y để ghi đè
         ffmpeg.run(stream, overwrite_output=True, quiet=True, capture_stdout=True, capture_stderr=True)
 
         # Mở file WAV đã chuyển đổi