import os
import subprocess

MODEL_PATH = "model"
MODEL_URL = "https://huggingface.co/vosk-models/vosk-small-en-us/resolve/main"

# Danh sách file cần tải
files = ["am", "conf", "graph", "ivector", "rescore"]

if not os.path.exists(MODEL_PATH):
    os.makedirs(MODEL_PATH)

for file in files:
    file_url = f"{MODEL_URL}/{file}.zip"
    file_path = os.path.join(MODEL_PATH, f"{file}.zip")
    
    if not os.path.exists(file_path):
        print(f"Đang tải {file_url} ...")
        subprocess.run(["wget", "-O", file_path, file_url])
        subprocess.run(["unzip", "-o", file_path, "-d", MODEL_PATH])
        os.remove(file_path)

print("Tải model thành công!")