Input for user OpenAI API key

app.py

@@ -1,4 +1,5 @@
 import time
+import os
 from functools import lru_cache
 
 import pandas as pd
@@ -20,6 +21,10 @@ st.title("Auto-PySQLify")
 st.markdown("> #### NYC-AI-Hackathon April, 23 2023", unsafe_allow_html=True)
 st.write("_Data Analysis_ Tool")
 
+if "OPENAI_API_KEY" not in os.environ:
+    os.environ["OPENAI_API_KEY"] = st.text_input("OpenAI API Key", type="password")
+
+
 p = st.write
 print = st.write
 display = st.write