Update app.py

app.py

@@ -5,7 +5,8 @@ from langchain_core.prompts import FewShotPromptTemplate
 from langchain.prompts.example_selector import LengthBasedExampleSelector
 
 def getLLMResponse(query,age_option,tasktype_option,numberOfWords):
-    llm = OpenAI(model="gpt-3.5-turbo-instruct", temperature=.9)
+
+    llm = OpenAI(model="gpt-3.5-turbo-instruct", temperature=.9, api_key=st.session_state['API_Key'])
 
     examples=[]
     if age_option=="Kids": #Silly and Sweet Kid 
@@ -98,6 +99,11 @@ def getLLMResponse(query,age_option,tasktype_option,numberOfWords):
 
 #Frontend Starts here
 
+if 'API_Key' not in st.session_state:
+    st.session_state['API_Key'] =''
+    
+st.session_state['API_Key']= st.sidebar.text_input("What's your OpenAI API key?",type="password")
+
 st.set_page_config(page_title="Marketing Tool",
                     page_icon='✅',
                     layout='centered',
@@ -119,5 +125,5 @@ numberOfWords= st.slider('Words limit', 1, 200, 25)
 submit = st.button("Generate")
 
 if submit:
-    st.write(getLLMResponse(form_input,age_option,tasktype_option,numberOfWords))
+    st.write(getLLMResponse(form_input,age_option,tasktype_option, numberOfWords))