JaehaerysNL/GPT4FREE
1
0
Fork 0
Code Activity

fix index display for search

Browse source
This commit is contained in:
mache102 2023-05-01 22:57:20 +08:00
parent fe97af4aaf
commit 4853e4bfad
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
gui/streamlit_chat_app.py
View file

@ -108,20 +108,24 @@ search_query = st.sidebar.text_input("Search Conversations:", value=st.session_s
if search_query: if search_query:
filtered_conversations = [] 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]: if search_query in conversation['user_inputs'][0]:
filtered_conversations.append(conversation) 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)})" sidebar_header = f"Search Results ({len(conversations)})"
else: else:
conversations = st.session_state.conversations conversations = enumerate(st.session_state.conversations)
sidebar_header = "Conversation History" sidebar_header = "Conversation History"
# Sidebar # Sidebar
st.sidebar.header(sidebar_header) 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}"): 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['selected_conversation'] = idx
st.session_state['current_conversation'] = conversation st.session_state['current_conversation'] = conversation