Update app3.py

app3.py

@@ -15,13 +15,19 @@ import logging
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
+# Fetch API keys from environment variables
+api_key = os.getenv("OPENAI_API_KEY")
+pandasai_api_key = os.getenv("PANDASAI_API_KEY")
+
+# Check for missing keys
+if not api_key or not pandasai_api_key:
+    raise EnvironmentError(
+        "API keys are missing! Ensure both 'OPENAI_API_KEY' and 'PANDASAI_API_KEY' are set in the environment."
+    )
+
 # Title of the app
 st.title("PandasAI Data Analyzer with RAG")
 
-# Sidebar for API keys
-api_key = st.sidebar.text_input("OpenAI API Key", type="password")
-pandasai_api_key = st.sidebar.text_input("PandasAI API Key", type="password")
-
 # Function to load datasets into session
 def load_dataset_into_session():
     input_option = st.radio(
@@ -83,10 +89,6 @@ load_dataset_into_session()
 if st.session_state.df is not None:
     df = st.session_state.df
     try:
-        # Set API keys in environment variables
-        os.environ["OPENAI_API_KEY"] = api_key
-        os.environ["PANDASAI_API_KEY"] = pandasai_api_key
-
         # Initialize PandasAI Agent
         agent = Agent(df)