update.py

import subprocess


def do(cmd):  # https://stackoverflow.com/a/62986640
    try:
        run = subprocess.run(cmd, shell=True)
        run.check_returncode()
        return run
    except subprocess.CalledProcessError as e:
        try:
            print(
                e.stderr.decode().strip(),
                e.returncode,
            )
        except Exception:
            print(e)
            pass
        raise e


def try_install(requirements, name=None):
    try:
        print(f"Installing {name or requirements} dependencies...")
        do(f"pip install -r {requirements}")
        print(f"Successfully installed {name or requirements} dependencies")
    except Exception:
        print(f"Failed to install {name or requirements} dependencies")


def is_node_installed():
    try:
        print("Checking if node is installed...")
        do("node --version")
        return True
    except Exception:
        print("Node is not installed, skipping node_modules setup")
        return False


def setup_node_modules():
    try:
        print("Installing node_modules...")
        do("cd react-ui && npm install")
        print("Successfully installed node_modules")
        print("Building react-ui...")
        do("cd react-ui && npm run build")
        print("Successfully built react-ui")
    except Exception:
        print("Failed to install node_modules")


def check_if_torch_has_cuda():
    try:
        print("Checking if torch has CUDA...")
        do("python check_cuda.py")
        return True
    except Exception:
        return False


def main():
    print("Updating dependencies...")
    try_install(
        "requirements_audiocraft_only.txt --no-deps", "musicgen, audiocraft 1/2"
    )
    try_install("requirements_audiocraft_deps.txt", "musicgen, audiocraft 2/2")
    try_install(
        "requirements_bark_hubert_quantizer.txt",
        "Bark Voice Clone, bark-hubert-quantizer",
    )
    try_install("requirements_rvc.txt", "RVC")
    # hydracore fix because of fairseq
    do("pip install hydra-core==1.3.2")

    if is_node_installed():
        setup_node_modules()

    if check_if_torch_has_cuda():
        print("Torch has CUDA, skipping reinstall")
    else:
        print(
            "Torch does not have CUDA, assuming CPU mode, installing CPU version of PyTorch"
        )
        do(
            "conda install --force-reinstall -y -k pytorch torchvision torchaudio cpuonly git -c pytorch"
        )


if __name__ == "__main__":
    main()