Update app.py

app.py

@@ -22,6 +22,8 @@ from langchain.schema import (
 
 st.set_page_config(page_title="CFA Level 1", page_icon="📖")
 
+#Load API Key
+api_key = os.environ["OPENAI_API_KEY"]   
 
 #### sidebar section 1 ####
 with st.sidebar:
@@ -120,10 +122,13 @@ def load_chain():
     
     return qa
 
+chat_history = []
 
 def get_answer(question):
+    '''Generate an answer from the chain'''
+    
     chain = load_chain()
-    result = chain({"query": question})
+    result = chain({"question": question, "chat_history": chat_history})
 
     answer = result["result"]
     
@@ -151,8 +156,6 @@ def get_answer(question):
 
     return answer, pages, extract
 
-##### sidebar section 2 ####
-api_key = os.environ["OPENAI_API_KEY"]   
 
 ##### main ####
 user_input = st.text_input("Your question", "What is an MBS and who are the main issuer and investors of the MBS market?", key="input")