summaryrefslogtreecommitdiff
path: root/style.py
diff options
context:
space:
mode:
authorSchark <jordan@schark.online>2023-12-05 14:33:42 -0800
committerSchark <jordan@schark.online>2023-12-05 14:33:42 -0800
commit90e2e98f420b7b213ff16c1376a2905f10addb5f (patch)
treee352b92ed9ec39fd7a83166aeba8178c196d1f00 /style.py
parent45cfb6e15597676f3f9be4c76a6ddc78076e2748 (diff)
downloadcli-gpt-90e2e98f420b7b213ff16c1376a2905f10addb5f.tar.gz
cli-gpt-90e2e98f420b7b213ff16c1376a2905f10addb5f.zip
Moving API key location to 'key', file restructure
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()