Update model/download_model.py

model/download_model.py

@@ -1,46 +1,29 @@
-import os
 import requests
 import zipfile
 
-MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-en-us-0.22.zip"
+MODEL_URL = "https://alphacephei.com/vosk/models/vosk-model-small-en-us-0.15.zip"
+
 MODEL_DIR = "model"
-MODEL_NAME = "vosk-model-en-us-0.22"  # Đổi về model tiếng Anh
-MODEL_PATH = os.path.join(MODEL_DIR, MODEL_NAME)
+MODEL_PATH = os.path.join(MODEL_DIR, "vosk-model")
 
 if not os.path.exists(MODEL_PATH):
     os.makedirs(MODEL_DIR, exist_ok=True)
 
     zip_path = os.path.join(MODEL_DIR, "model.zip")
 
-    print("📥 Đang tải model Vosk tiếng Anh...")
+    print("📥 Đang tải model Vosk...")
     response = requests.get(MODEL_URL, stream=True)
-
-    if response.status_code != 200:
-        print(f"❌ Lỗi tải model! Mã lỗi: {response.status_code}")
-        exit(1)
-
     with open(zip_path, "wb") as f:
         for chunk in response.iter_content(chunk_size=8192):
             f.write(chunk)
 
-    print("📦 Kiểm tra file ZIP...")
-    if not zipfile.is_zipfile(zip_path):
-        print("❌ File tải về không phải là file ZIP hợp lệ!")
-        os.remove(zip_path)
-        exit(1)
-
     print("📦 Giải nén model...")
     with zipfile.ZipFile(zip_path, "r") as zip_ref:
         zip_ref.extractall(MODEL_DIR)
 
-    if os.path.exists(MODEL_PATH):
-        print(f"✅ Model đã được giải nén tại: {MODEL_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.rename(os.path.join(MODEL_DIR, "vosk-model-small-en-us-0.15"), MODEL_PATH)
     os.remove(zip_path)  # Xóa file zip sau khi giải nén
 
-    print("✅ Model tiếng Anh đã tải thành công!")
+    print("✅ Model đã tải thành công!")
 else:
-    print(f"✅ Model đã có sẵn tại {MODEL_PATH}, bỏ qua tải về.")
+    print("✅ Model đã có sẵn, bỏ qua tải về.")