Update app4.py

app4.py

@@ -12,7 +12,7 @@ import os
 # Set title
 st.title("Data Analyzer")
 
-# Add fields to input API keys via the sidebar
+# API keys
 api_key = os.getenv("OPENAI_API_KEY")
 pandasai_api_key = os.getenv("PANDASAI_API_KEY")
 
@@ -33,7 +33,6 @@ def load_dataset_into_session():
             try:
                 st.session_state.df = pd.read_csv(file_path)
                 st.success(f"File loaded successfully from '{file_path}'!")
-                st.dataframe(st.session_state.df.head(10))
             except Exception as e:
                 st.error(f"Error loading dataset from the repo directory: {e}")
 
@@ -51,7 +50,6 @@ def load_dataset_into_session():
                 else:
                     st.session_state.df = pd.DataFrame(dataset)
                 st.success(f"Hugging Face Dataset '{dataset_name}' loaded successfully!")
-                st.dataframe(st.session_state.df.head(10))
             except Exception as e:
                 st.error(f"Error loading Hugging Face dataset: {e}")
 
@@ -62,7 +60,6 @@ def load_dataset_into_session():
             try:
                 st.session_state.df = pd.read_csv(uploaded_file)
                 st.success("File uploaded successfully!")
-                st.dataframe(st.session_state.df.head(10))
             except Exception as e:
                 st.error(f"Error reading uploaded file: {e}")
 
@@ -76,7 +73,7 @@ if "df" in st.session_state and api_key and pandasai_api_key:
 
     df = st.session_state.df
     st.write("Dataset Preview:")
-    st.write(df.head())
+    st.write(df.head())  # Ensure the dataset preview is displayed only once
 
     # Set up PandasAI Agent
     agent = Agent(df)
@@ -107,15 +104,21 @@ if "df" in st.session_state and api_key and pandasai_api_key:
         st.header("Data Analysis with PandasAI")
         pandas_question = st.text_input("Ask a question about the dataset (PandasAI):")
         if pandas_question:
-            result = agent.chat(pandas_question)
-            st.write("PandasAI Answer:", result)
+            try:
+                result = agent.chat(pandas_question)
+                st.write("PandasAI Answer:", result)
+            except Exception as e:
+                st.error(f"PandasAI encountered an error: {str(e)}")
 
     with tab2:
         st.header("Q&A with RAG")
         rag_question = st.text_input("Ask a question about the dataset (RAG):")
         if rag_question:
-            result = qa_chain.run(rag_question)
-            st.write("RAG Answer:", result)
+            try:
+                result = qa_chain.run(rag_question)
+                st.write("RAG Answer:", result)
+            except Exception as e:
+                st.error(f"RAG encountered an error: {str(e)}")
 
     with tab3:
         st.header("Data Visualization")
@@ -142,8 +145,7 @@ if "df" in st.session_state and api_key and pandasai_api_key:
                 else:
                     st.write("Unable to generate the graph. Please try a different query.")
             except Exception as e:
-                st.write(f"An error occurred: {str(e)}")
-                st.write("Please try asking in a different way.")
+                st.error(f"An error occurred during visualization: {str(e)}")
 else:
     if not api_key:
         st.warning("Please set the OpenAI API key in environment variables.")