Archives/GPT-Brain
2
0
Fork 0
mirror of https://github.com/sean1832/GPT-Brain synced 2024-11-18 21:25:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactored for clarity

Browse Source
This commit is contained in:
sean1832 2023-02-10 02:30:51 +11:00
parent 577db13cf2
commit d1b15d09f3
1 changed files with 26 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
web_ui/Seanium_Brain.py
View File

@ -5,9 +5,6 @@ import check_update
import time
import os
# activate session
if 'SESSION_TIME' not in st.session_state:
st.session_state['SESSION_TIME'] = time.strftime("%Y%m%d-%H%H%S")
@ -16,7 +13,7 @@ st.set_page_config(
page_title='Seanium Brain'
)
model_options = ['text-davinci-003', 'text-curie-001','text-babbage-001','text-ada-001']
model_options = ['text-davinci-003', 'text-curie-001', 'text-babbage-001', 'text-ada-001']
header = st.container()
body = st.container()
LOG_PATH = '.user/log'
@ -29,13 +26,15 @@ def create_log():
util.write_file(f'Session {SESSION_TIME}\n\n', CURRENT_LOG_FILE)
return CURRENT_LOG_FILE
def log(content, path=LOG_PATH, seperater_text=''):
def log(content, delimiter=''):
log_file = create_log()
if seperater_text != '':
seperater_text = f'\n\n=============={seperater_text}==============\n'
util.write_file(f'\n{seperater_text + content}', log_file, 'a')
if delimiter != '':
delimiter = f'\n\n=============={delimiter}==============\n'
util.write_file(f'\n{delimiter + content}', log_file, 'a')
def clear_log():
log_file_name = f'log_{SESSION_TIME}.log'
@ -44,21 +43,23 @@ def clear_log():
if not file == log_file_name:
os.remove(os.path.join(root, file))
def save_as():
# download log file
with open(CURRENT_LOG_FILE, 'rb') as f:
bytes = f.read()
content = f.read()
st.download_button(
label="📥download log",
data=bytes,
data=content,
file_name=f'log_{SESSION_TIME}.txt',
mime='text/plain'
)
# sidebar
with st.sidebar:
st.title('Settings')
output_types = st.multiselect('Output Types',['Answer', 'Summary'],default=['Answer'])
output_types = st.multiselect('Output Types', ['Answer', 'Summary'], default=['Answer'])
answer_model = st.selectbox('Answer Model', model_options)
if util.contains(output_types, 'Summary'):
summary_model = st.selectbox('Summary Model', model_options)
@ -72,32 +73,33 @@ with st.sidebar:
chunk_size = st.slider('Chunk Size', 1500, 4500, value=4000)
chunk_count = st.slider('Answer Count', 1, 5, value=1)
if st.button('Clear Log',on_click=clear_log):
if st.button('Clear Log', on_click=clear_log):
st.success('Log Cleared')
with header:
st.title('🧠Seanium Brain')
st.text('This is my personal AI powered brain feeding my own Obsidian notes. Ask anything.')
def execute_brain(question):
def execute_brain(q):
# log question
log(f'\n\n\n\n[{str(time.ctime())}] - QUESTION: {question}')
log(f'\n\n\n\n[{str(time.ctime())}] - QUESTION: {q}')
if check_update.isUpdated():
# if brain-info is updated
brain.build(chunk_size)
st.success('Brain rebuilded!')
st.success('Brain rebuild!')
time.sleep(2)
# thinking on answer
with st.spinner('Thinking on Answer'):
answer = brain.run_answer(question, answer_model, temp, max_tokens, top_p, freq_panl, pres_panl, chunk_count=chunk_count)
answer = brain.run_answer(q, answer_model, temp, max_tokens, top_p, freq_panl, pres_panl,
chunk_count=chunk_count)
if util.contains(output_types, 'Answer'):
# displaying results
st.header('💬Answer')
st.success(answer)
log(answer, seperater_text='ANSWER')
log(answer, delimiter='ANSWER')
# thinking on summary
if util.contains(output_types, 'Summary'):
with st.spinner('Thinking on Summary'):
@ -106,7 +108,8 @@ def execute_brain(question):
# displaying results
st.header('📃Summary')
st.success(summary)
log(summary, seperater_text='SUMMARY')
log(summary, delimiter='SUMMARY')
# main
with body:
@ -117,7 +120,7 @@ with body:
with col2:
if os.path.exists(CURRENT_LOG_FILE):
save_as()
# execute brain calculation
if not question == '' and send:
execute_brain(question)
execute_brain(question)