Update model/download_model.py

model/download_model.py

@@ -4,7 +4,7 @@ import zipfile
 
 MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-vn-0.4.zip"
 MODEL_DIR = "model"
-MODEL_PATH = os.path.join(MODEL_DIR, "vosk-model-vn")
+MODEL_PATH = os.path.join(MODEL_DIR, "vosk-model-vn-0.4")  # Đổi lại cho đúng với thư mục sau khi giải nén
 
 if not os.path.exists(MODEL_PATH):
     os.makedirs(MODEL_DIR, exist_ok=True)
@@ -32,9 +32,16 @@ if not os.path.exists(MODEL_PATH):
     with zipfile.ZipFile(zip_path, "r") as zip_ref:
         zip_ref.extractall(MODEL_DIR)
 
-    os.rename(os.path.join(MODEL_DIR, "vosk-model-vn-0.4"), MODEL_PATH)
+    # Kiểm tra xem thư mục giải nén có tồn tại không
+    extracted_path = os.path.join(MODEL_DIR, "vosk-model-vn-0.4")
+    if os.path.exists(extracted_path):
+        print(f"✅ Model đã được giải nén tại: {extracted_path}")
+    else:
+        print(f"❌ Lỗi: Model không tồn tại sau khi giải nén! Kiểm tra lại.")
+        exit(1)
+
     os.remove(zip_path)  # Xóa file zip sau khi giải nén
 
     print("✅ Model tiếng Việt đã tải thành công!")
 else:
-    print("✅ Model đã có sẵn, bỏ qua tải về.")
+    print(f"✅ Model đã có sẵn tại {MODEL_PATH}, bỏ qua tải về.")