Update app.py

app.py

@@ -8,7 +8,7 @@ from langchain.prompts.chat import ChatPromptTemplate
 
 ## Load Anthropic model and get response
 
-def get_openai_response(question):
+def get_openai_response(question, api_key):
     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):
         ("human", user_template)
     ])
     
-    chat_llm = ChatOpenAI(temperature=0.5)
+    chat_llm = ChatOpenAI(temperature=0.5, openai_api_key=api_key)
     
     chain = chat_prompt | chat_llm
     
@@ -28,8 +28,10 @@ def get_openai_response(question):
 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")
+
 input = st.text_input("Input: ", key=input)
-response = get_openai_response(input)
+response = get_openai_response(input, api_key=OPENAI_API_KEY)
 
 submit = st.button("Ask your question...")
 if submit: