@ -1,6 +1,6 @@
import atexit
import os
import os
import sys
import sys
import atexit
sys . path . append ( os . path . join ( os . path . dirname ( __file__ ) , os . path . pardir ) )
sys . path . append ( os . path . join ( os . path . dirname ( __file__ ) , os . path . pardir ) )
@ -9,9 +9,9 @@ from streamlit_chat import message
from query_methods import query , avail_query_methods
from query_methods import query , avail_query_methods
import pickle
import pickle
conversations_file = " conversations.pkl "
conversations_file = " conversations.pkl "
def load_conversations ( ) :
def load_conversations ( ) :
try :
try :
with open ( conversations_file , " rb " ) as f :
with open ( conversations_file , " rb " ) as f :
@ -31,11 +31,11 @@ def save_conversations(conversations, current_conversation):
break
break
if not updated :
if not updated :
conversations . append ( current_conversation )
conversations . append ( current_conversation )
temp_conversations_file = " temp_ " + conversations_file
temp_conversations_file = " temp_ " + conversations_file
with open ( temp_conversations_file , " wb " ) as f :
with open ( temp_conversations_file , " wb " ) as f :
pickle . dump ( conversations , f )
pickle . dump ( conversations , f )
os . replace ( temp_conversations_file , conversations_file )
os . replace ( temp_conversations_file , conversations_file )
@ -44,10 +44,10 @@ def exit_handler():
# Perform cleanup operations here, like saving data or closing open files.
# Perform cleanup operations here, like saving data or closing open files.
save_conversations ( st . session_state . conversations , st . session_state . current_conversation )
save_conversations ( st . session_state . conversations , st . session_state . current_conversation )
# Register the exit_handler function to be called when the program is closing.
# Register the exit_handler function to be called when the program is closing.
atexit . register ( exit_handler )
atexit . register ( exit_handler )
st . header ( " Chat Placeholder " )
st . header ( " Chat Placeholder " )
if ' conversations ' not in st . session_state :
if ' conversations ' not in st . session_state :
@ -61,7 +61,7 @@ if 'selected_conversation' not in st.session_state:
if ' input_field_key ' not in st . session_state :
if ' input_field_key ' not in st . session_state :
st . session_state [ ' input_field_key ' ] = 0
st . session_state [ ' input_field_key ' ] = 0
if ' query_method ' not in st . session_state :
if ' query_method ' not in st . session_state :
st . session_state [ ' query_method ' ] = query
st . session_state [ ' query_method ' ] = query
@ -69,20 +69,22 @@ if 'query_method' not in st.session_state:
if ' current_conversation ' not in st . session_state or st . session_state [ ' current_conversation ' ] is None :
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 ' : [ ] }
st . session_state [ ' current_conversation ' ] = { ' user_inputs ' : [ ] , ' generated_responses ' : [ ] }
input_placeholder = st . empty ( )
input_placeholder = st . empty ( )
user_input = input_placeholder . text_input ( ' You: ' , key = f ' input_text_ { len ( st . session_state [ " current_conversation " ] [ " user_inputs " ] ) } ' )
user_input = input_placeholder . text_input (
' You: ' , key = f ' input_text_ { len ( st . session_state [ " current_conversation " ] [ " user_inputs " ] ) } '
)
submit_button = st . button ( " Submit " )
submit_button = st . button ( " Submit " )
if user_input or submit_button :
if user_input or submit_button :
output = query ( user_input , st . session_state [ ' query_method ' ] )
output = query ( user_input , st . session_state [ ' query_method ' ] )
escaped_output = output . encode ( ' utf-8 ' ) . decode ( ' unicode-escape ' )
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 )
st . session_state . current_conversation [ ' generated_responses ' ] . append ( escaped_output )
save_conversations ( st . session_state . conversations , st . session_state . current_conversation )
save_conversations ( st . session_state . conversations , st . session_state . current_conversation )
user_input = input_placeholder . text_input ( ' You: ' , value = ' ' , key = f ' input_text_ { len ( st . session_state [ " current_conversation " ] [ " user_inputs " ] ) } ' ) # Clear the input field
user_input = input_placeholder . text_input (
' You: ' , value = ' ' , key = f ' input_text_ { len ( st . session_state [ " current_conversation " ] [ " user_inputs " ] ) } '
) # Clear the input field
# Add a button to create a new conversation
# Add a button to create a new conversation
if st . sidebar . button ( " New Conversation " ) :
if st . sidebar . button ( " New Conversation " ) :
@ -90,11 +92,7 @@ if st.sidebar.button("New Conversation"):
st . session_state [ ' current_conversation ' ] = { ' user_inputs ' : [ ] , ' generated_responses ' : [ ] }
st . session_state [ ' current_conversation ' ] = { ' user_inputs ' : [ ] , ' generated_responses ' : [ ] }
st . session_state [ ' input_field_key ' ] + = 1
st . session_state [ ' input_field_key ' ] + = 1
st . session_state [ ' query_method ' ] = st . sidebar . selectbox (
st . session_state [ ' query_method ' ] = st . sidebar . selectbox ( " Select API: " , options = avail_query_methods , index = 0 )
" Select API: " ,
options = avail_query_methods ,
index = 0
)
# Sidebar
# Sidebar
st . sidebar . header ( " Conversation History " )
st . sidebar . header ( " Conversation History " )