main.py - master - UlvHare/AddaxAI_PySide6

main.py
62 строки2 KB
Peter van Lunteren
Change name to AddaxAI (#70)
17 фев 2025, 16:38
17 фев 2025, 16:38bb543eb
# This is a placeholder script for PyInstaller to create the main executable 
# https://addaxdatascience.com/addaxai/
# Created by Peter van Lunteren
# Latest edit by Peter van Lunteren on 12 Jan 2024

import os
import subprocess
import sys
import platform

print("\n")

# check os
system = platform.system()

# os dependent python executables
def get_python_interprator(env_name):
    if system == 'Windows':
        return os.path.join(AddaxAI_files, "envs", f"env-{env_name}", "python.exe")
    else:
        return os.path.join(AddaxAI_files, "envs", f"env-{env_name}", "bin", "python")

# clean path
if getattr(sys, 'frozen', False):
    AddaxAI_files = os.path.dirname(sys.executable)
else:
    AddaxAI_files = os.path.dirname(os.path.abspath(__file__))
    
if AddaxAI_files.endswith("main.app/Contents/MacOS"):
    AddaxAI_files = AddaxAI_files.replace("main.app/Contents/MacOS", "")
    
if AddaxAI_files.endswith(".app/Contents/MacOS"):
    AddaxAI_files = os.path.dirname(os.path.dirname(os.path.dirname(AddaxAI_files)))

# init paths    
GUI_script = os.path.join(AddaxAI_files, "AddaxAI", "AddaxAI_GUI.py")
first_startup_file = os.path.join(AddaxAI_files, "first-startup.txt")

# log
print(f"        AddaxAI_files: {AddaxAI_files}")
print(f"       sys.executable: {sys.executable.replace(AddaxAI_files, '.')}")
print(f"           GUI_script: {GUI_script.replace(AddaxAI_files, '.')}")

# python executable
python_executable = get_python_interprator("base")
print(f"    python_executable: {python_executable.replace(AddaxAI_files, '.')}")

# cuda toolkit
cuda_toolkit_path = os.environ.get("CUDA_HOME") or os.environ.get("CUDA_PATH")
print(f"    cuda_toolkit_path: {cuda_toolkit_path}")

# run the GUI script
print("\nOpening application...") 
if system == 'Windows':
    if sys.executable.endswith("debug.exe"):
        subprocess.run([get_python_interprator("base"), GUI_script])
        input("Press [Enter] to close console window...") # keep console open after closing app
    else:
        subprocess.run([get_python_interprator("base"), GUI_script],
                       creationflags=subprocess.CREATE_NO_WINDOW)
else:
    subprocess.run([get_python_interprator("base"), GUI_script])