OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2025-12-31 16:40:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

Enabled getting user input from the external process

Browse Source
This commit is contained in:
Zvonimir Sabljic
2023-08-31 08:38:37 +02:00
parent 1418704186
commit bdb4d0dff8
4 changed files with 30 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pilot/utils/questionary.py
View File

@@ -3,6 +3,7 @@ import questionary
from termcolor import colored
from database.database import save_user_input, get_user_input_from_hash_id
from const.ipc import MESSAGE_TYPE
custom_style = Style.from_dict({
'question': '#FFFFFF bold', # the color and style of the question
@@ -28,10 +29,15 @@ def styled_text(project, question):
print(colored(f'{user_input.user_input}', 'yellow', attrs=['bold']))
return user_input.user_input
config = {
'style': custom_style,
}
response = questionary.text(question, **config).unsafe_ask() # .ask() is included here
if project.ipc_client_instance is None or project.ipc_client_instance.client is None:
config = {
'style': custom_style,
}
response = questionary.text(question, **config).unsafe_ask() # .ask() is included here
else:
response = project.log(question, MESSAGE_TYPE['user_input_request'])
project.log(response, MESSAGE_TYPE['verbose'])
user_input = save_user_input(project, question, response)
print('\n\n', end='')