summaryrefslogtreecommitdiff
path: root/style.py
diff options
context:
space:
mode:
Diffstat (limited to 'style.py')
-rw-r--r--style.py57
1 files changed, 0 insertions, 57 deletions
diff --git a/style.py b/style.py
deleted file mode 100644
index ad616d8..0000000
--- a/style.py
+++ /dev/null
@@ -1,57 +0,0 @@
-import sys
-import time
-
-from prompt_toolkit import PromptSession, print_formatted_text, prompt
-from prompt_toolkit.formatted_text import HTML
-from prompt_toolkit.lexers import PygmentsLexer
-from prompt_toolkit.styles import Style
-from pygments.lexers import PythonLexer
-from pygments.styles.native import NativeStyle
-from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
-from prompt_toolkit.history import FileHistory
-from prompt_toolkit.completion import WordCompleter
-
-class StyleLog:
-
- styler = None
-
- style = Style.from_dict({
- 'input': 'bg:#000000 #00ff00',
- 'assistant': 'bg:#000000 #7777ff',
- 'system': 'bg:#000000 #ff00ff',
- })
-
- def __init__(self):
- self.styler = PromptSession(lexer=PygmentsLexer(PythonLexer), auto_suggest=AutoSuggestFromHistory(), history=FileHistory('history.txt'))
-
- def prompt(self, role: str, message: str):
- if role == 'assistant':
- print_formatted_text(HTML(f"<assistant>Assistant: </assistant>%s") % (message, ), style = self.style)
- elif role == 'user':
- user_input = prompt(
- [
- ('class:input', "\nInput: "),
- ('', '')
- ],
- style = self.style
- )
- return user_input
- elif role == 'system':
- print_formatted_text(HTML(f'<system>System:</system> {message}'), style = self.style)
- elif role == 'none':
- print_formatted_text(HTML(f'{message}'), style = self.style)
- return
-
- def show_ellipsis(self, api_call_done):
- loop = True
- while loop:
- for i in range(0, 4):
- if api_call_done.is_set():
- loop = False
- sys.stdout.write('\r' + ' ' * 4 + '\r')
- break
- time.sleep(1)
- sys.stdout.write('.')
- sys.stdout.flush()
- sys.stdout.write('\r' + ' ' * 4 + '\r')
- sys.stdout.flush()