forked from TransformerOptimus/SuperAGI
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_gui.py
50 lines (43 loc) · 1.45 KB
/
run_gui.py
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
import os
import sys
import subprocess
from time import sleep
import shutil
def check_command(command, message):
if not shutil.which(command):
print(message)
sys.exit(1)
def run_npm_commands():
os.chdir("gui")
try:
subprocess.run(["npm", "install"], check=True)
except subprocess.CalledProcessError:
print(f"Error during '{' '.join(sys.exc_info()[1].cmd)}'. Exiting.")
sys.exit(1)
os.chdir("..")
def run_server():
api_process = subprocess.Popen(["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"])
os.chdir("gui")
ui_process = subprocess.Popen(["npm", "run", "dev"])
os.chdir("..")
return api_process, ui_process
def cleanup(api_process, ui_process):
print("Shutting down processes...")
api_process.terminate()
ui_process.terminate()
print("Processes terminated. Exiting.")
sys.exit(1)
if __name__ == "__main__":
check_command("node", "Node.js is not installed. Please install it and try again.")
check_command("npm", "npm is not installed. Please install npm to proceed.")
check_command("uvicorn", "uvicorn is not installed. Please install uvicorn to proceed.")
run_npm_commands()
try:
api_process, ui_process = run_server()
while True:
try:
sleep(30)
except KeyboardInterrupt:
cleanup(api_process, ui_process)
except Exception as e:
cleanup(api_process, ui_process)