返回提交历史
Modified
gui/streamlit_chat_app.py
+8
-4
XFEstudio/gpt4free
fix index display for search
4853e4bf
代码差异
1 个文件
+8
-4
@@ -108,20 +108,24 @@ search_query = st.sidebar.text_input("Search Conversations:", value=st.session_s
108
108
109
109
if search_query:
110
110
filtered_conversations = []
111
for conversation in st.session_state.conversations:
111
indices = []
112
for idx, conversation in enumerate(st.session_state.conversations):
112
113
if search_query in conversation['user_inputs'][0]:
113
114
filtered_conversations.append(conversation)
115
indices.append(idx)
116
117
filtered_conversations = list(zip(indices, filtered_conversations))
118
conversations = sorted(filtered_conversations, key=lambda x: Levenshtein.distance(search_query, x[1]['user_inputs'][0]))
114
119
115
conversations = sorted(filtered_conversations, key=lambda c: Levenshtein.distance(search_query, c['user_inputs'][0]))
116
120
sidebar_header = f"Search Results ({len(conversations)})"
117
121
else:
118
conversations = st.session_state.conversations
122
conversations = enumerate(st.session_state.conversations)
119
123
sidebar_header = "Conversation History"
120
124
121
125
# Sidebar
122
126
st.sidebar.header(sidebar_header)
123
127
124
for idx, conversation in enumerate(conversations):
128
for idx, conversation in conversations:
125
129
if st.sidebar.button(f"Conversation {idx + 1}: {conversation['user_inputs'][0]}", key=f"sidebar_btn_{idx}"):
126
130
st.session_state['selected_conversation'] = idx
127
131
st.session_state['current_conversation'] = conversation