Update app.py

app.py

@@ -77,15 +77,8 @@ def initialize_llmchain(vector_db):
     
     memory = ConversationBufferMemory(memory_key="chat_history", output_key='answer', return_messages=True)
     retriever=vector_db.as_retriever()
-    qa_chain = ConversationalRetrievalChain.from_llm(
-        llm,
-        retriever=retriever,
-        chain_type="stuff", 
-        memory=memory,
-        return_source_documents=True,
-        verbose=False,
-    )
-    return qa_chain
+
+    #return qa_chain
 
 
 
@@ -161,13 +154,16 @@ def conversation(qa_chain, message, history):
     
 #document = os.listdir(list_file_obj)
 vector_db, collection_name = initialize_database(list_file_obj)
-qa_chain = initialize_llmchain(vector_db)
+#qa_chain = 
+initialize_llmchain(vector_db)
+qa_chain = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,chain_type="stuff", 
+                                                 memory=memory,return_source_documents=True,verbose=False,)
 print('qa chain and vector_db done')
 
 def demo():
     with gr.Blocks(theme='base') as demo:
         vector_db = gr.State()
-        qa_chain = gr.State()
+        #qa_chain = gr.State()
         collection_name = gr.State()
         
         chatbot = gr.Chatbot(height=300)