#!python

import os
import shutil
import subprocess
from importlib.util import find_spec

anchor_found = find_spec("anchor") is not None

conda_path = shutil.which("conda")
mamba_path = shutil.which("mamba")
if mamba_path is None:
    print("No mamba found, installing first...")
    subprocess.call([conda_path, "install", "-c", "conda-forge", "-y", "mamba"], env=os.environ)
package_list = ["montreal-forced-aligner", "kalpy", "kaldi=*=cpu*"]
if anchor_found:
    package_list.append("anchor-annotator")
subprocess.call([mamba_path, "update", "-c", "conda-forge", "-y"] + package_list, env=os.environ)
