Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,7 +8,7 @@ from langchain.prompts.chat import ChatPromptTemplate
 
 ## Load Anthropic model and get response
 
-def get_openai_response(question, api_key):
+def get_openai_response(question):
     system_template = "You are a helpful coder assistant. When the user asks a question, your task is to write simple Python codes."
     user_template = "{question}"
     chat_prompt = ChatPromptTemplate.from_messages([
@@ -16,7 +16,7 @@ def get_openai_response(question, api_key):
         ("human", user_template)
     ])
     
-    chat_llm = ChatOpenAI(temperature=0.5, api_key=api_key)
+    chat_llm = ChatOpenAI(temperature=0.5)
     
     chain = chat_prompt | chat_llm
     
@@ -29,11 +29,12 @@ st.set_page_config(page_title="Simple Langchain App")
 st.header("Coder Q&A Demo")
 
 OPENAI_API_KEY = st.text_input("OpenAI API Key: ", type="password")
+os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
 
 input = st.text_input("Input: ", key=input)
 
 submit = st.button("Ask your question...")
 if submit:
-    response = get_openai_response(input, api_key=OPENAI_API_KEY)
+    response = get_openai_response(input)
     st.subheader("The response is:")
     st.write(response)