From 08f53b336158d0da44ed7c6e0d6ec2d01ab509a9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E2=80=93=20sanz?= Date: Mon, 1 May 2023 09:10:38 +0800 Subject: Update streamlit_chat_app.py --- gui/streamlit_chat_app.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'gui/streamlit_chat_app.py') diff --git a/gui/streamlit_chat_app.py b/gui/streamlit_chat_app.py index 6abc9caf..bca4b954 100644 --- a/gui/streamlit_chat_app.py +++ b/gui/streamlit_chat_app.py @@ -74,7 +74,6 @@ user_input = input_placeholder.text_input( ) submit_button = st.button("Submit") - if (user_input and user_input != st.session_state['input_text']) or submit_button: output = query(user_input, st.session_state['query_method']) @@ -84,6 +83,7 @@ if (user_input and user_input != st.session_state['input_text']) or submit_butto 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'] = '' + st.session_state['input_field_key'] += 1 # Increment key value for new widget user_input = input_placeholder.text_input( 'You:', value=st.session_state['input_text'], key=f'input_text_{st.session_state["input_field_key"]}' ) # Clear the input field @@ -92,9 +92,8 @@ if (user_input and user_input != st.session_state['input_text']) or submit_butto if st.sidebar.button("New Conversation"): st.session_state['selected_conversation'] = None st.session_state['current_conversation'] = {'user_inputs': [], 'generated_responses': []} - st.session_state['input_field_key'] += 1 - -st.session_state['query_method'] = st.sidebar.selectbox("Select API:", options=avail_query_methods, index=0) + st.session_state['input_field_key'] += 1 # Increment key value for new widget + st.session_state['query_method'] = st.sidebar.selectbox("Select API:", options=avail_query_methods, index=0) # Proxy st.session_state['proxy'] = st.sidebar.text_input("Proxy: ") -- cgit v1.2.3 From 2140fc87cb8615a650711053f0afbbe4b1fd8553 Mon Sep 17 00:00:00 2001 From: mache102 Date: Mon, 1 May 2023 14:43:17 +0800 Subject: Refactor conversation selection --- gui/streamlit_chat_app.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'gui/streamlit_chat_app.py') diff --git a/gui/streamlit_chat_app.py b/gui/streamlit_chat_app.py index 6abc9caf..50d8c5b9 100644 --- a/gui/streamlit_chat_app.py +++ b/gui/streamlit_chat_app.py @@ -105,7 +105,7 @@ st.sidebar.header("Conversation History") 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] + st.session_state['current_conversation'] = conversation if st.session_state['selected_conversation'] is not None: conversation_to_display = st.session_state.conversations[st.session_state['selected_conversation']] @@ -115,4 +115,4 @@ else: if conversation_to_display['generated_responses']: for i in range(len(conversation_to_display['generated_responses']) - 1, -1, -1): message(conversation_to_display["generated_responses"][i], key=f"display_generated_{i}") - message(conversation_to_display['user_inputs'][i], is_user=True, key=f"display_user_{i}") + message(conversation_to_display['user_inputs'][i], is_user=True, key=f"display_user_{i}") \ No newline at end of file -- cgit v1.2.3 From fe97af4aaf65c792930c30bcbddcf5b8758573ad Mon Sep 17 00:00:00 2001 From: mache102 Date: Mon, 1 May 2023 15:23:40 +0800 Subject: add searchbar --- gui/streamlit_chat_app.py | 29 ++++++++++++++++++++++++----- 1 file changed, 24 insertions(+), 5 deletions(-) (limited to 'gui/streamlit_chat_app.py') diff --git a/gui/streamlit_chat_app.py b/gui/streamlit_chat_app.py index 50d8c5b9..c5c960a9 100644 --- a/gui/streamlit_chat_app.py +++ b/gui/streamlit_chat_app.py @@ -1,4 +1,5 @@ import atexit +import Levenshtein import os import sys @@ -64,13 +65,16 @@ if 'input_field_key' not in st.session_state: if 'query_method' not in st.session_state: st.session_state['query_method'] = query +if 'search_query' not in st.session_state: + st.session_state['search_query'] = '' + # Initialize new conversation if 'current_conversation' not in st.session_state or st.session_state['current_conversation'] is None: st.session_state['current_conversation'] = {'user_inputs': [], 'generated_responses': []} input_placeholder = st.empty() user_input = input_placeholder.text_input( - 'You:', value=st.session_state['input_text'], key=f'input_text_{st.session_state["input_field_key"]}' + 'You:', value=st.session_state['input_text'], key=f'input_text_-1'#{st.session_state["input_field_key"]} ) submit_button = st.button("Submit") @@ -80,7 +84,7 @@ if (user_input and user_input != st.session_state['input_text']) or submit_butto 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) save_conversations(st.session_state.conversations, st.session_state.current_conversation) st.session_state['input_text'] = '' @@ -99,16 +103,31 @@ st.session_state['query_method'] = st.sidebar.selectbox("Select API:", options=a # Proxy st.session_state['proxy'] = st.sidebar.text_input("Proxy: ") +# Searchbar +search_query = st.sidebar.text_input("Search Conversations:", value=st.session_state.get('search_query', ''), key='search') + +if search_query: + filtered_conversations = [] + for conversation in st.session_state.conversations: + if search_query in conversation['user_inputs'][0]: + filtered_conversations.append(conversation) + + conversations = sorted(filtered_conversations, key=lambda c: Levenshtein.distance(search_query, c['user_inputs'][0])) + sidebar_header = f"Search Results ({len(conversations)})" +else: + conversations = st.session_state.conversations + sidebar_header = "Conversation History" + # Sidebar -st.sidebar.header("Conversation History") +st.sidebar.header(sidebar_header) -for idx, conversation in enumerate(st.session_state.conversations): +for idx, conversation in enumerate(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'] = conversation if st.session_state['selected_conversation'] is not None: - conversation_to_display = st.session_state.conversations[st.session_state['selected_conversation']] + conversation_to_display = conversations[st.session_state['selected_conversation']] else: conversation_to_display = st.session_state.current_conversation -- cgit v1.2.3 From 4853e4bfad5c3d6a32b8f309f0a2e54e0d35fe3d Mon Sep 17 00:00:00 2001 From: mache102 Date: Mon, 1 May 2023 22:57:20 +0800 Subject: fix index display for search --- gui/streamlit_chat_app.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'gui/streamlit_chat_app.py') diff --git a/gui/streamlit_chat_app.py b/gui/streamlit_chat_app.py index c5c960a9..f80afa80 100644 --- a/gui/streamlit_chat_app.py +++ b/gui/streamlit_chat_app.py @@ -108,20 +108,24 @@ search_query = st.sidebar.text_input("Search Conversations:", value=st.session_s if search_query: filtered_conversations = [] - for conversation in st.session_state.conversations: + indices = [] + for idx, conversation in enumerate(st.session_state.conversations): if search_query in conversation['user_inputs'][0]: filtered_conversations.append(conversation) + indices.append(idx) + + filtered_conversations = list(zip(indices, filtered_conversations)) + conversations = sorted(filtered_conversations, key=lambda x: Levenshtein.distance(search_query, x[1]['user_inputs'][0])) - conversations = sorted(filtered_conversations, key=lambda c: Levenshtein.distance(search_query, c['user_inputs'][0])) sidebar_header = f"Search Results ({len(conversations)})" else: - conversations = st.session_state.conversations + conversations = enumerate(st.session_state.conversations) sidebar_header = "Conversation History" # Sidebar st.sidebar.header(sidebar_header) -for idx, conversation in enumerate(conversations): +for idx, conversation in 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'] = conversation -- cgit v1.2.3 From 1379ec96b98fccc4f89b598149ac9bf4671c3fad Mon Sep 17 00:00:00 2001 From: Vokturz Date: Mon, 1 May 2023 15:44:58 -0400 Subject: added option to remove conversations --- gui/streamlit_chat_app.py | 26 +++++++++++++++++++++----- 1 file changed, 21 insertions(+), 5 deletions(-) (limited to 'gui/streamlit_chat_app.py') diff --git a/gui/streamlit_chat_app.py b/gui/streamlit_chat_app.py index 99e8428c..af3969e6 100644 --- a/gui/streamlit_chat_app.py +++ b/gui/streamlit_chat_app.py @@ -38,6 +38,17 @@ def save_conversations(conversations, current_conversation): os.replace(temp_conversations_file, conversations_file) +def delete_conversation(conversations, current_conversation): + for idx, conversation in enumerate(conversations): + conversations[idx] = current_conversation + break + conversations.remove(current_conversation) + + temp_conversations_file = "temp_" + conversations_file + with open(temp_conversations_file, "wb") as f: + pickle.dump(conversations, f) + + os.replace(temp_conversations_file, conversations_file) def exit_handler(): print("Exiting, saving data...") @@ -118,17 +129,22 @@ if search_query: sidebar_header = f"Search Results ({len(conversations)})" else: - conversations = enumerate(st.session_state.conversations) + conversations = st.session_state.conversations sidebar_header = "Conversation History" # Sidebar st.sidebar.header(sidebar_header) - -for idx, conversation in conversations: - if st.sidebar.button(f"Conversation {idx + 1}: {conversation['user_inputs'][0]}", key=f"sidebar_btn_{idx}"): +sidebar_col1, sidebar_col2 = st.sidebar.columns([5,1]) +for idx, conversation in enumerate(conversations): + if sidebar_col1.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'] = conversation - + if sidebar_col2.button('🗑️', key=f"sidebar_btn_delete_{idx}"): + if st.session_state['selected_conversation'] == idx: + st.session_state['selected_conversation'] = None + st.session_state['current_conversation'] = {'user_inputs': [], 'generated_responses': []} + delete_conversation(conversations, conversation) + st.experimental_rerun() if st.session_state['selected_conversation'] is not None: conversation_to_display = conversations[st.session_state['selected_conversation']] else: -- cgit v1.2.3