AddaxAI/main.py at main · PetervanLunteren/AddaxAI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# 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])