install.py

import launch
import platform

# Package check util
# Modified from https://github.com/Bing-su/adetailer/blob/main/install.py
import importlib.util
from importlib.metadata import version
from packaging.version import parse

def is_installed(package: str):
    min_version = "0.0.0"
    max_version = "99999999.99999999.99999999"
    pkg_name = package
    version_check = True
    if "==" in package:
        pkg_name, _version = package.split("==")
        min_version = max_version = _version
    elif "<=" in package:
        pkg_name, _version = package.split("<=")
        max_version = _version
    elif ">=" in package:
        pkg_name, _version = package.split(">=")
        min_version = _version
    else:
        version_check = False
    package = pkg_name
    try:
        spec = importlib.util.find_spec(package)
    except ModuleNotFoundError:
        message = f"is_installed check for {str(package)} failed as error ModuleNotFoundError"
        print(message)
        return False
    if spec is None:
        message = f"is_installed check for {str(package)} failed as 'spec is None'"
        print(message)
        return False
    if not version_check:
        return True
    if package == "google.protobuf":
        package = "protobuf"
    try:
        pkg_version = version(package)
        return parse(min_version) <= parse(pkg_version) <= parse(max_version)
    except Exception as e:
        message = f"is_installed check for {str(package)} failed as error {str(e)}"
        print(message)
        return False
# End of Package check util

if not is_installed("cv2"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install opencv-python", "requirements for opencv")

if not is_installed("tensorflow-cpu"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install tensorflow-cpu", "requirements for tensorflow")

if not is_installed("onnx"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install onnx", "requirements for onnx")

if not is_installed("onnxruntime"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install onnxruntime", "requirements for onnxruntime")

if not is_installed("modelscope==1.9.3"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install modelscope==1.9.3", "requirements for modelscope")

if not is_installed("diffusers==0.18.2"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install diffusers==0.18.2", "requirements for diffusers")

if not is_installed("imageio>=2.29.0"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install imageio>=2.29.0", "requirements for imageio")

# Temporarily pin fsspec==2023.9.2. See https://github.com/huggingface/datasets/issues/6330 for details.
if not is_installed("fsspec==2023.9.2"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install fsspec==2023.9.2", "requirements for fsspec")

# `StableDiffusionXLPipeline` in diffusers requires the invisible-watermark library.
if not launch.is_installed("invisible-watermark"):
    print('Installing requirements for easyphoto-webui')
    launch.run_pip("install invisible-watermark", "requirements for invisible-watermark")

if platform.system() != 'Windows':
    if not is_installed("nvitop"):
        print('Installing requirements for easyphoto-webui')
        launch.run_pip("install nvitop==1.3.0", "requirements for tensorflow")