diff --git a/initial_file_creator.py b/initial_file_creator.py index 217e5e2..60e9eb9 100644 --- a/initial_file_creator.py +++ b/initial_file_creator.py @@ -13,5 +13,14 @@ def create(): {'note_dir': '', 'delimiter': '', 'append_mode': False, 'force_mode': False}) print(f'brain memo file created: {user_dir}/brain-memo.json') + util.create_file_not_exist(f'{user_dir}/input.txt', '') + print(f'input file created: {user_dir}/input.txt') + + if not os.path.exists(f'{user_dir}/API-KEYS.txt'): + print('Create API profile...') + api_key = input('Enter your OpenAI API key: ') + util.create_file_not_exist(f'{user_dir}/API-KEYS.txt', api_key) + print(f'API key file created: {user_dir}/API-KEYS.txt') + create() diff --git a/modules/utilities.py b/modules/utilities.py index 6b15fe0..12346fb 100644 --- a/modules/utilities.py +++ b/modules/utilities.py @@ -60,9 +60,9 @@ def create_path_not_exist(path): os.makedirs(directory) -def create_file_not_exist(path): +def create_file_not_exist(path, content=''): if not os.path.exists(path): - write_file('', path) + write_file(content, path) def read_file(filepath, delimiter='', force=False): diff --git a/setup.bat b/setup.bat index 11212ab..f7931f6 100644 --- a/setup.bat +++ b/setup.bat @@ -5,14 +5,12 @@ python -m venv venv echo Virtural environment created successfully! ping 127.0.0.1 -n 2 > NUL - echo Activating Virtural environment! call .\venv\Scripts\activate echo updating pip python -m pip install --upgrade pip - pip3 install -r requirements.txt ping 127.0.0.1 -n 2 > NUL echo Virtual requirements installed successfully! @@ -22,14 +20,6 @@ echo Creating OpenAI API keys profile... REM if .user\ not exist, create one if not exist .user\ (md .user\) -REM Create API KEY file -if not exist .user\API-KEYS.txt ( - set /p API_KEYS=[Enter your API keys]: - echo %API_KEYS%> .user\API-KEYS.txt - echo API key written to file! -) - - REM copy example prompt if not exist .user\prompt (md .user\prompt) xcopy "example_prompt\*.*" ".user\prompt" /s /i @@ -37,17 +27,9 @@ xcopy "example_prompt\*.*" ".user\prompt" /s /i REM wait 2 tick ping 127.0.0.1 -n 2 > NUL -REM create input txt file -if not exist .user\input.txt ( - echo.> .user\input.txt - echo input file created! -) - - -python web_ui/initial_file_creator.py +python initial_file_creator.py echo Setup complete! Exiting... - pause \ No newline at end of file