mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2025-12-28 15:17:53 +01:00
59 lines
2.4 KiB
Python
59 lines
2.4 KiB
Python
from prompt_toolkit.styles import Style
|
|
import questionary
|
|
from utils.style import yellow_bold
|
|
import re
|
|
from database.database import save_user_input, get_saved_user_input
|
|
|
|
custom_style = Style.from_dict({
|
|
'question': '#FFFFFF bold', # the color and style of the question
|
|
'answer': '#FF910A bold', # the color and style of the answer
|
|
'pointer': '#FF4500 bold', # the color and style of the selection pointer
|
|
'highlighted': '#63CD91 bold', # the color and style of the highlighted choice
|
|
'instruction': '#FFFF00 bold' # the color and style of the question mark
|
|
})
|
|
|
|
|
|
def remove_ansi_codes(s: str) -> str:
|
|
ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])')
|
|
return ansi_escape.sub('', s)
|
|
|
|
|
|
def styled_select(*args, **kwargs):
|
|
kwargs["style"] = custom_style # Set style here
|
|
return questionary.select(*args, **kwargs).unsafe_ask() # .ask() is included here
|
|
|
|
|
|
def styled_text(project, question, ignore_user_input_count=False, style=None):
|
|
if not ignore_user_input_count:
|
|
project.user_inputs_count += 1
|
|
user_input = get_saved_user_input(project, question)
|
|
if user_input is not None and user_input.user_input is not None and project.skip_steps:
|
|
# if we do, use it
|
|
project.checkpoints['last_user_input'] = user_input
|
|
print(yellow_bold(f'Restoring user input id {user_input.id}: '), end='')
|
|
print(yellow_bold(f'{user_input.user_input}'))
|
|
return user_input.user_input
|
|
|
|
if project.ipc_client_instance is None or project.ipc_client_instance.client is None:
|
|
config = {
|
|
'style': style if style is not None else custom_style,
|
|
}
|
|
question = remove_ansi_codes(question) # Colorama and questionary are not compatible and styling doesn't work
|
|
response = questionary.text(question, **config).unsafe_ask() # .ask() is included here
|
|
else:
|
|
response = print(question, type='user_input_request')
|
|
print(response)
|
|
|
|
if not ignore_user_input_count:
|
|
user_input = save_user_input(project, question, response)
|
|
|
|
print('\n\n', end='')
|
|
return response
|
|
|
|
|
|
def get_user_feedback():
|
|
config = {
|
|
'style': custom_style,
|
|
}
|
|
return questionary.text("How did GPT Pilot do? Were you able to create any app that works? Please write any feedback you have or just press ENTER to exit: ", **config).unsafe_ask()
|