v2.6.2 - easy windows install

Author: BBC-Esq

src/setup.py

@@ -0,0 +1,66 @@
+import sys
+import subprocess
+import os
+import ctypes
+
+def user_confirmation(message):
+    return ctypes.windll.user32.MessageBoxW(0, message, "Confirmation", 1) == 1
+
+def check_python_version_and_confirm():
+    major, minor = map(int, sys.version.split()[0].split('.')[:2])
+    if major < 3 or (major == 3 and minor < 10):
+        ctypes.windll.user32.MessageBoxW(0, "Python 3.10 or higher is required for this installation.", "Python Version Error", 0)
+        return False
+    else:
+        return user_confirmation(f"Python version {sys.version.split()[0]} detected. Click OK to proceed with the installation, or Cancel to stop.")
+
+def check_cuda_version():
+    try:
+        cuda_version_output = subprocess.check_output(["nvcc", "--version"]).decode('utf-8')
+        if "release" in cuda_version_output:
+            cuda_version = cuda_version_output.split("release ")[1].split(",")[0]
+            return cuda_version
+        else:
+            return None
+    except FileNotFoundError:
+        return None
+
+def display_cuda_message():
+    cuda_version = check_cuda_version()
+    if cuda_version is None:
+        proceed_without_cuda = ctypes.windll.user32.MessageBoxW(0, "No CUDA installation detected. Would you like to proceed with a CPU-only installation?", "CUDA Check", 1) == 1
+        return proceed_without_cuda
+    elif cuda_version == "11.8":
+        return ctypes.windll.user32.MessageBoxW(0, "You have the correct CUDA version (11.8). Click OK to proceed.", "CUDA Check", 1) == 1
+    else:
+        update_cuda = ctypes.windll.user32.MessageBoxW(0, f"Incorrect version of CUDA installed (Version: {cuda_version}). Would you like to proceed with a CPU-only installation?", "CUDA Check", 1) == 1
+        return update_cuda
+
+def manual_installation_confirmation():
+    if not user_confirmation("Have you installed Git? Click OK to confirm, or Cancel to exit installation."):
+        return False
+    if not user_confirmation("Have you installed Git Large File Storage? Click OK to confirm, or Cancel to exit installation."):
+        return False
+    if not user_confirmation("Have you installed Pandoc? Click OK to confirm, or Cancel to exit installation."):
+        return False
+    return True
+
+def setup_windows_installation():
+    if not check_python_version_and_confirm():
+        return
+    if not manual_installation_confirmation():
+        return
+    cuda_installed = display_cuda_message()
+    os.system("python -m pip install --upgrade pip")
+    if cuda_installed:
+        if user_confirmation("Click OK for Nvidia GPU-acceleration support or Cancel for CPU only."):
+            os.system("pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118")
+        else:
+            os.system("pip install torch torchvision torchaudio")
+    else:
+        os.system("pip install torch torchvision torchaudio")
+    os.system("pip install -r requirements.txt")
+    print("Installation completed successfully.")
+    print("Run 'Python gui.py' to start program.")
+
+setup_windows_installation()