Update app.py

app.py

@@ -5,6 +5,7 @@ import numpy as np
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from vosk import Model, KaldiRecognizer
+from flasgger import Swagger
 
 # Thư mục chứa model
 MODEL_PATH = "model/vosk-model"
@@ -20,13 +21,42 @@ model = Model(MODEL_PATH)
 # Khởi tạo Flask app
 app = Flask(__name__)
 CORS(app)
+swagger = Swagger(app)
 
 @app.route("/")
 def home():
+    """API Home
+    ---
+    responses:
+      200:
+        description: API đang chạy
+    """
     return "✅ Vosk STT API đang chạy!"
 
 @app.route("/stt", methods=["POST"])
 def stt():
+    """Chuyển đổi giọng nói thành văn bản (Speech-to-Text)
+    ---
+    consumes:
+      - multipart/form-data
+    parameters:
+      - in: formData
+        name: audio
+        type: file
+        required: true
+        description: File âm thanh WAV mono PCM
+    responses:
+      200:
+        description: Kết quả chuyển đổi văn bản
+        schema:
+          type: object
+          properties:
+            text:
+              type: string
+              example: "Xin chào thế giới"
+      400:
+        description: Lỗi nếu file âm thanh không hợp lệ hoặc không tìm thấy
+    """
     if "audio" not in request.files:
         return jsonify({"error": "Không tìm thấy file audio!"}), 400