Update app.py

app.py

@@ -90,8 +90,8 @@ def load_prompt(book_name, author_name):
     return PROMPT
 
 @st.experimental_singleton(show_spinner=False)
-def load_chain():
-    llm = OpenAI(temperature=0.2)
+def load_chain(openai_api_key=None):
+    llm = OpenAI(temperature=0.2, openai_api_key=openai_api_key)
 
     chain = VectorDBQA.from_chain_type(
         chain_type_kwargs = {"prompt": load_prompt(book_name=BOOK_NAME, author_name=AUTHOR_NAME)},
@@ -104,8 +104,8 @@ def load_chain():
     
     return chain
 
-def get_answer(question):
-    chain = load_chain()
+def get_answer(question, openai_api_key=None):
+    chain = load_chain(openai_api_key=openai_api_key)
     result = chain({"query": question})
 
     answer = result["result"]
@@ -194,7 +194,7 @@ if ask:
         
     with st.spinner(msg):
         try:
-            answer, pages, extract = get_answer(question=user_input)
+            answer, pages, extract = get_answer(question=user_input, openai_api_key=openai_api_key)
             logger.info(f"answer: {answer}")
         except Exception as exc:
             logger.error(exc)