mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2025-12-28 23:25:46 +01:00
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
# main.py
|
|
from __future__ import print_function, unicode_literals
|
|
import builtins
|
|
import os
|
|
|
|
import sys
|
|
import traceback
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
from utils.style import red
|
|
from utils.custom_print import get_custom_print
|
|
from helpers.Project import Project
|
|
from utils.arguments import get_arguments
|
|
from utils.exit import exit_gpt_pilot
|
|
from logger.logger import logger
|
|
from database.database import database_exists, create_database, tables_exist, create_tables, get_created_apps_with_steps
|
|
|
|
|
|
def init():
|
|
# Check if the "euclid" database exists, if not, create it
|
|
if not database_exists():
|
|
create_database()
|
|
|
|
# Check if the tables exist, if not, create them
|
|
if not tables_exist():
|
|
create_tables()
|
|
|
|
arguments = get_arguments()
|
|
|
|
logger.info('Starting with args: %s', arguments)
|
|
|
|
return arguments
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
# sys.argv.append('--ux-test=' + 'continue_development')
|
|
args = init()
|
|
|
|
builtins.print, ipc_client_instance = get_custom_print(args)
|
|
if '--api-key' in args:
|
|
os.environ["OPENAI_API_KEY"] = args['--api-key']
|
|
if '--get-created-apps-with-steps' in args:
|
|
print({ 'db_data': get_created_apps_with_steps() }, type='info')
|
|
elif '--ux-test' in args:
|
|
from test.ux_tests import run_test
|
|
run_test(args['--ux-test'], args)
|
|
else:
|
|
# TODO get checkpoint from database and fill the project with it
|
|
project = Project(args, ipc_client_instance=ipc_client_instance)
|
|
project.start()
|
|
except KeyboardInterrupt:
|
|
exit_gpt_pilot()
|
|
except Exception as e:
|
|
print(red('---------- GPT PILOT EXITING WITH ERROR ----------'))
|
|
traceback.print_exc()
|
|
print(red('--------------------------------------------------'))
|
|
exit_gpt_pilot(False)
|
|
finally:
|
|
sys.exit(0)
|