diff options
Diffstat (limited to 'gui/streamlit_chat_app.py')
| -rw-r--r-- | gui/streamlit_chat_app.py | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/gui/streamlit_chat_app.py b/gui/streamlit_chat_app.py index 3d18333a..fc5c8d8e 100644 --- a/gui/streamlit_chat_app.py +++ b/gui/streamlit_chat_app.py @@ -24,9 +24,9 @@ def load_conversations(): def save_conversations(conversations, current_conversation): updated = False - for i, conversation in enumerate(conversations): + for idx, conversation in enumerate(conversations): if conversation == current_conversation: - conversations[i] = current_conversation + conversations[idx] = current_conversation updated = True break if not updated: @@ -71,19 +71,22 @@ if 'current_conversation' not in st.session_state or st.session_state['current_c input_placeholder = st.empty() 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") -if user_input or submit_button: - output = query(user_input, st.session_state['query_method'], st.session_state['proxy']) + +if (user_input and user_input != st.session_state['input_text']) or submit_button: + output = query(user_input, st.session_state['query_method']) + escaped_output = output.encode('utf-8').decode('unicode-escape') st.session_state.current_conversation['user_inputs'].append(user_input) st.session_state.current_conversation['generated_responses'].append(escaped_output) save_conversations(st.session_state.conversations, st.session_state.current_conversation) + st.session_state['input_text'] = '' 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 # Add a button to create a new conversation @@ -100,10 +103,10 @@ st.session_state['proxy'] = st.sidebar.text_input("Proxy: ") # Sidebar st.sidebar.header("Conversation History") -for i, conversation in enumerate(st.session_state.conversations): - if st.sidebar.button(f"Conversation {i + 1}: {conversation['user_inputs'][0]}", key=f"sidebar_btn_{i}"): - st.session_state['selected_conversation'] = i - st.session_state['current_conversation'] = st.session_state.conversations[i] +for idx, conversation in enumerate(st.session_state.conversations): + if st.sidebar.button(f"Conversation {idx + 1}: {conversation['user_inputs'][0]}", key=f"sidebar_btn_{idx}"): + st.session_state['selected_conversation'] = idx + st.session_state['current_conversation'] = st.session_state.conversations[idx] if st.session_state['selected_conversation'] is not None: conversation_to_display = st.session_state.conversations[st.session_state['selected_conversation']] |