fixed: bug fix on GUI

the known bug has been fixed when trying to load a conversation and etc, the bug on the README
This commit is contained in:
– sanz 2023-04-30 09:42:57 +08:00 committed by GitHub
parent 9489dda310
commit 55990be732
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 9 deletions

View File

@ -24,9 +24,9 @@ def load_conversations():
def save_conversations(conversations, current_conversation): def save_conversations(conversations, current_conversation):
updated = False updated = False
for i, conversation in enumerate(conversations): for idx, conversation in enumerate(conversations):
if conversation == current_conversation: if conversation == current_conversation:
conversations[i] = current_conversation conversations[idx] = current_conversation
updated = True updated = True
break break
if not updated: if not updated:
@ -71,19 +71,20 @@ if 'current_conversation' not in st.session_state or st.session_state['current_c
input_placeholder = st.empty() input_placeholder = st.empty()
user_input = input_placeholder.text_input( user_input = input_placeholder.text_input(
'You:', key=f'input_text_{len(st.session_state["current_conversation"]["user_inputs"])}' 'You:', value=st.session_state['input_text'], key=f'input_text_{st.session_state["input_field_key"]}'
) )
submit_button = st.button("Submit") submit_button = st.button("Submit")
if user_input or submit_button: if (user_input and user_input != st.session_state['input_text']) or submit_button:
output = query(user_input, st.session_state['query_method']) output = query(user_input, st.session_state['query_method'])
escaped_output = output.encode('utf-8').decode('unicode-escape') escaped_output = output.encode('utf-8').decode('unicode-escape')
st.session_state.current_conversation['user_inputs'].append(user_input) st.session_state.current_conversation['user_inputs'].append(user_input)
st.session_state.current_conversation['generated_responses'].append(escaped_output) st.session_state.current_conversation['generated_responses'].append(escaped_output)
save_conversations(st.session_state.conversations, st.session_state.current_conversation) save_conversations(st.session_state.conversations, st.session_state.current_conversation)
st.session_state['input_text'] = ''
user_input = input_placeholder.text_input( user_input = input_placeholder.text_input(
'You:', value='', key=f'input_text_{len(st.session_state["current_conversation"]["user_inputs"])}' 'You:', value=st.session_state['input_text'], key=f'input_text_{st.session_state["input_field_key"]}'
) # Clear the input field ) # Clear the input field
# Add a button to create a new conversation # Add a button to create a new conversation
@ -97,10 +98,10 @@ st.session_state['query_method'] = st.sidebar.selectbox("Select API:", options=a
# Sidebar # Sidebar
st.sidebar.header("Conversation History") st.sidebar.header("Conversation History")
for i, conversation in enumerate(st.session_state.conversations): for idx, conversation in enumerate(st.session_state.conversations):
if st.sidebar.button(f"Conversation {i + 1}: {conversation['user_inputs'][0]}", key=f"sidebar_btn_{i}"): if st.sidebar.button(f"Conversation {idx + 1}: {conversation['user_inputs'][0]}", key=f"sidebar_btn_{idx}"):
st.session_state['selected_conversation'] = i st.session_state['selected_conversation'] = idx
st.session_state['current_conversation'] = st.session_state.conversations[i] st.session_state['current_conversation'] = st.session_state.conversations[idx]
if st.session_state['selected_conversation'] is not None: if st.session_state['selected_conversation'] is not None:
conversation_to_display = st.session_state.conversations[st.session_state['selected_conversation']] conversation_to_display = st.session_state.conversations[st.session_state['selected_conversation']]