Upload install_modules.py

install_modules.py

@@ -0,0 +1,138 @@
+import subprocess
+import sys
+import os
+import urllib.request
+
+def colored(text, color):
+    """
+    Функция для вывода цветного текста в консоль.
+    """
+    colors = {
+        'red': '\033[91m',
+        'green': '\033[92m',
+        'yellow': '\033[93m',
+        'blue': '\033[94m',
+        'magenta': '\033[95m',
+        'cyan': '\033[96m',
+        'white': '\033[97m',
+    }
+    return colors.get(color, '') + text + '\033[0m'
+
+def uninstall_module(module_name, python_path):
+    """
+    Функция для деинсталляции модуля.
+    """
+    print(colored(f"Удаление модуля {module_name}...", 'red'))
+    pip_path = os.path.join(python_path, "Scripts", "pip.exe")
+    subprocess.run([
+        pip_path,
+        "uninstall",
+        "-y",
+        module_name
+    ])
+
+def download_file(url, filename):
+    """
+    Функция для загрузки файла с помощью urllib.request.
+    """
+    print(colored(f"Загрузка файла {filename}...", 'red'))
+    try:
+        with urllib.request.urlopen(url) as response, open(filename, 'wb') as f:
+            f.write(response.read())
+        print(colored(f"Файл {filename} успешно загружен.", 'red'))
+    except Exception as e:
+        print(colored(f"Ошибка при загрузке файла: {e}", 'red'))
+
+def install_module(filename, python_path):
+    """
+    Функция для установки модуля из файла .whl.
+    """
+    print(colored(f"Установка модуля из {filename}...", 'red'))
+    pip_path = os.path.join(python_path, "Scripts", "pip.exe")
+    subprocess.run([
+        pip_path,
+        "install",
+        filename
+    ])
+
+def install_from_index(module_name, index_url, python_path):
+    """
+    Функция для установки модуля из указанного индекса.
+    """
+    print(colored(f"Установка модуля {module_name} с индекса {index_url}...", 'red'))
+    pip_path = os.path.join(python_path, "Scripts", "pip.exe")
+    subprocess.run([
+        pip_path,
+        "install",
+        "--extra-index-url",
+        index_url,
+        module_name
+    ])
+
+if __name__ == "__main__":
+    # Получение пути к текущей директории
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+
+    # Создание полного пути к папке Python
+    python_path = os.path.join(current_dir, "python_embeded")
+    
+    # Определение путей к git
+    git_path = os.path.join(current_dir, "git")
+    git_bin_path = os.path.join(git_path, "bin")
+    git_libexec_path = os.path.join(git_path, "libexec")
+
+    # Обновление переменной PATH
+    os.environ['PATH'] = f"{git_bin_path};{git_libexec_path};{os.environ['PATH']}"
+
+    # Установка Comfy
+    print(colored("Установка Comfy", 'red'))
+    git_exe_path = os.path.join(git_bin_path, "git.exe")
+    
+    subprocess.run([
+        git_exe_path,
+        "clone",
+        "https://github.com/comfyanonymous/ComfyUI"
+    ])
+    
+    # Обновление
+    modules_to_uninstall = ["numpy", "onnxruntime", "onnxruntime-gpu", "torch", "torchvision", "torchaudio"]
+    for module in modules_to_uninstall:
+        uninstall_module(module, python_path)
+
+    # Обновление модуля pip
+    print(colored("Установка модуля numpy==1.26.4...", 'red'))
+    pip_path = os.path.join(python_path, "Scripts", "pip.exe")
+    subprocess.run([
+        pip_path,  # Добавлен путь к pip.exe
+        "install",
+        "--upgrade",
+        "pip"
+    ])
+    
+    # Установка numpy==1.26.4
+    print(colored("Установка модуля numpy==1.26.4...", 'red'))
+    pip_path = os.path.join(python_path, "Scripts", "pip.exe")
+    subprocess.run([
+        pip_path,  # Добавлен путь к pip.exe
+        "install",
+        "numpy==1.26.4"
+    ])
+
+    # Загрузка файла insightface-0.7.3-cp311-cp311-win_amd64.whl
+    url = "https://github.com/Gourieff/Assets/raw/main/Insightface/insightface-0.7.3-cp311-cp311-win_amd64.whl"
+    filename = "insightface-0.7.3-cp311-cp311-win_amd64.whl"
+    download_file(url, filename)
+
+    # Установка модуля из файла .whl
+    install_module(filename, python_path)
+
+    # Установка onnxruntime-gpu из сетевого ресурса
+    onnxruntime_index = "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"
+    install_from_index("onnxruntime-gpu", onnxruntime_index, python_path)
+
+    pytorch_index = "https://download.pytorch.org/whl/cu124"
+    install_from_index("torch", pytorch_index, python_path)
+    install_from_index("torchvision", pytorch_index, python_path)
+    install_from_index("torchaudio", pytorch_index, python_path)
+
+    print(colored("Установка завершена.", 'green'))