fix

app.py

@@ -138,6 +138,7 @@ def to_contexts(passages):
 
 
 def qa(
+    openai_api_key: str,
     question: str,
     passages: list,
     model_name: str,
@@ -145,7 +146,7 @@ def qa(
     qa_prompt: str,
     max_tokens=2000,
 ):
-    client = OpenAI()
+    client = OpenAI(api_key=openai_api_key)
     contexts = to_contexts(passages)
     prompt = qa_prompt.format(contexts=contexts, question=question)
     response = client.chat.completions.create(
@@ -164,16 +165,25 @@ def qa(
 
 
 def generate_answer(
-    buf, question, passages, model_name, temperature, qa_prompt, max_tokens
+    openai_api_key,
+    buf,
+    question,
+    passages,
+    model_name,
+    temperature,
+    qa_prompt,
+    max_tokens,
 ):
     buf.write("⏳回答の生成中...")
     texts = ""
     for char in qa(
+        openai_api_key=openai_api_key,
         question=question,
         passages=passages,
         model_name=model_name,
         temperature=temperature,
         qa_prompt=qa_prompt,
+        max_tokens=max_tokens,
     ):
         texts += char
         buf.write(texts)
@@ -287,12 +297,14 @@ def app():
 
         openai_api_key = st.session_state.openai_api_key
         if openai_api_key:
+            openai_api_key = openai_api_key.strip()
             answer_header.subheader("Answer: ")
             openai_model_name = st.session_state.openai_model_name
             temperature = st.session_state.temperature
             qa_prompt = st.session_state.qa_prompt
             max_tokens = st.session_state.max_tokens
             generate_answer(
+                openai_api_key=openai_api_key,
                 buf=answer_text_buffer,
                 question=question,
                 passages=passages,