返回提交历史
Modified
gui/streamlit_chat_app.py
+24
-5
XFEstudio/gpt4free
add searchbar
fe97af4a
代码差异
1 个文件
+24
-5
@@ -1,4 +1,5 @@
1
1
import atexit
2
import Levenshtein
2
3
import os
3
4
import sys
4
5
@@ -64,13 +65,16 @@ if 'input_field_key' not in st.session_state:
64
65
if 'query_method' not in st.session_state:
65
66
st.session_state['query_method'] = query
66
67
68
if 'search_query' not in st.session_state:
69
st.session_state['search_query'] = ''
70
67
71
# Initialize new conversation
68
72
if 'current_conversation' not in st.session_state or st.session_state['current_conversation'] is None:
69
73
st.session_state['current_conversation'] = {'user_inputs': [], 'generated_responses': []}
70
74
71
75
input_placeholder = st.empty()
72
76
user_input = input_placeholder.text_input(
73
'You:', value=st.session_state['input_text'], key=f'input_text_{st.session_state["input_field_key"]}'
77
'You:', value=st.session_state['input_text'], key=f'input_text_-1'#{st.session_state["input_field_key"]}
74
78
)
75
79
submit_button = st.button("Submit")
76
80
@@ -80,7 +84,7 @@ if (user_input and user_input != st.session_state['input_text']) or submit_butto
80
84
81
85
escaped_output = output.encode('utf-8').decode('unicode-escape')
82
86
83
st.session_state.current_conversation['user_inputs'].append(user_input)
87
st.session_state['current_conversation']['user_inputs'].append(user_input)
84
88
st.session_state.current_conversation['generated_responses'].append(escaped_output)
85
89
save_conversations(st.session_state.conversations, st.session_state.current_conversation)
86
90
st.session_state['input_text'] = ''
@@ -99,16 +103,31 @@ st.session_state['query_method'] = st.sidebar.selectbox("Select API:", options=a
99
103
# Proxy
100
104
st.session_state['proxy'] = st.sidebar.text_input("Proxy: ")
101
105
106
# Searchbar
107
search_query = st.sidebar.text_input("Search Conversations:", value=st.session_state.get('search_query', ''), key='search')
108
109
if search_query:
110
filtered_conversations = []
111
for conversation in st.session_state.conversations:
112
if search_query in conversation['user_inputs'][0]:
113
filtered_conversations.append(conversation)
114
115
conversations = sorted(filtered_conversations, key=lambda c: Levenshtein.distance(search_query, c['user_inputs'][0]))
116
sidebar_header = f"Search Results ({len(conversations)})"
117
else:
118
conversations = st.session_state.conversations
119
sidebar_header = "Conversation History"
120
102
121
# Sidebar
103
st.sidebar.header("Conversation History")
122
st.sidebar.header(sidebar_header)
104
123
105
for idx, conversation in enumerate(st.session_state.conversations):
124
for idx, conversation in enumerate(conversations):
106
125
if st.sidebar.button(f"Conversation {idx + 1}: {conversation['user_inputs'][0]}", key=f"sidebar_btn_{idx}"):
107
126
st.session_state['selected_conversation'] = idx
108
127
st.session_state['current_conversation'] = conversation
109
128
110
129
if st.session_state['selected_conversation'] is not None:
111
conversation_to_display = st.session_state.conversations[st.session_state['selected_conversation']]
130
conversation_to_display = conversations[st.session_state['selected_conversation']]
112
131
else:
113
132
conversation_to_display = st.session_state.current_conversation
114
133