valeriylo commited on
Commit
cd053e7
1 Parent(s): 2ec105c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -73,7 +73,7 @@ def get_conversation_chain(vectorstore, model_name):
73
  return_source_documents=True
74
  )
75
 
76
- return conversation_chain
77
 
78
 
79
  def handle_userinput(user_question):
@@ -124,6 +124,5 @@ with st.sidebar:
124
  vectorstore = get_vectorstore(text_chunks)
125
 
126
  # create conversation chain
127
- st.session_state.conversation = get_conversation_chain(vectorstore, model_name)
128
- conversation = get_conversation_chain(vectorstore, model_name)
129
- st.text_area(conversation['source_documents'][0])
 
73
  return_source_documents=True
74
  )
75
 
76
+ return conversation_chain, conversation_chain['source_documents'][0]
77
 
78
 
79
  def handle_userinput(user_question):
 
124
  vectorstore = get_vectorstore(text_chunks)
125
 
126
  # create conversation chain
127
+ st.session_state.conversation, retrieved_docs = get_conversation_chain(vectorstore, model_name)
128
+ st.text_area(retrieved_docs)