Update app.py

app.py

@@ -6,6 +6,7 @@ import streamlit.components.v1 as components
 from pandasai import PandasAI
 from pandasai.llm.openai import OpenAI
 from streamlit_chat import message
+import os
 
 # Adjust the width of the Streamlit page
 st.set_page_config(page_title="Data Visualization ⭐", layout="wide")
@@ -86,7 +87,7 @@ if uploaded_file is not None:
     components.html(pyg_html, height=1000, scrolling=True)
 
     # Instantiate a LLM
-    OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
+    OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
     llm = OpenAI(api_token=OPENAI_API_KEY)
     pandas_ai = PandasAI(llm)