Update app.py

app.py

@@ -15,7 +15,15 @@ print(os.system('ls -l repo_directory/')) # testing purpose
 
 
 # from repo_directory.network18.app import main
-openai.api_key= os.environ.get("OPENAI_API_KEY")
-from repo_directory.PersonalityChatbot.app import *
-
-main()
+if "openai_key" not in st.session_state:
+    with st.form("API key"):
+        key = st.text_input("OpenAI Key", value="", type="password")
+        if st.form_submit_button("Submit"):
+            st.session_state.openai_key = key
+            openai.api_key = key
+            st.session_state.prompt_history = []
+            st.success('Saved API key for this session.')
+
+    if "openai_key" in st.session_state:
+        from repo_directory.PersonalityChatbot.app import *
+        main()