import os
import subprocess
import sys

requset_path = os.path.join(os.path.dirname(__file__), "./requirements.txt")


def install_requirements():
    try:
        # 检查是否已经安装了所有依赖项
        with open(requset_path) as f:
            requirements = f.read().splitlines()

        for requirement in requirements:
            try:
                # 提取纯包名，去除版本号和其他特殊字符
                package_name = (
                    requirement.split("==")[0]
                    .split(">")[0]
                    .split("<")[0]
                    .split("~")[0]
                    .strip()
                )
                # print(f"{package_name} 正在检查...")
                if package_name == "uuid7":
                    __import__("uuid_extensions")
                else:
                    __import__(package_name)
            except ImportError:
                print(f"{requirement} 未安装，正在安装...")
                subprocess.check_call(
                    [sys.executable, "-m", "pip", "install", requirement]
                )

        print("WeiLin-Comfyui-Tools ==> 所有依赖项已安装。")
    except Exception as e:
        print(f"WeiLin-Comfyui-Tools ==> 安装依赖项时出错: {e}")