Update app4.py

app4.py

@@ -76,32 +76,44 @@ if "df" in st.session_state and api_key and pandasai_api_key:
     st.write(df.head())  # Ensure the dataset preview is displayed only once
 
     # Set up PandasAI Agent
-    agent = Agent(df)
+    try:
+        agent = Agent(df)
+        st.info("PandasAI Agent initialized successfully.")
+    except Exception as e:
+        st.error(f"Error initializing PandasAI Agent: {str(e)}")
 
     # Convert dataframe into documents
-    documents = [
-        Document(
-            page_content=", ".join([f"{col}: {row[col]}" for col in df.columns]),
-            metadata={"index": index}
-        )
-        for index, row in df.iterrows()
-    ]
+    try:
+        documents = [
+            Document(
+                page_content=", ".join([f"{col}: {row[col]}" for col in df.columns]),
+                metadata={"index": index}
+            )
+            for index, row in df.iterrows()
+        ]
+        st.info("Documents created successfully for RAG.")
+    except Exception as e:
+        st.error(f"Error creating documents for RAG: {str(e)}")
 
     # Set up RAG
-    embeddings = OpenAIEmbeddings()
-    vectorstore = FAISS.from_documents(documents, embeddings)
-    retriever = vectorstore.as_retriever()
-    qa_chain = RetrievalQA.from_chain_type(
-        llm=ChatOpenAI(),
-        chain_type="stuff",
-        retriever=retriever
-    )
+    try:
+        embeddings = OpenAIEmbeddings()
+        vectorstore = FAISS.from_documents(documents, embeddings)
+        retriever = vectorstore.as_retriever()
+        qa_chain = RetrievalQA.from_chain_type(
+            llm=ChatOpenAI(),
+            chain_type="stuff",
+            retriever=retriever
+        )
+        st.info("RAG setup completed successfully.")
+    except Exception as e:
+        st.error(f"Error setting up RAG: {str(e)}")
 
     # Create tabs
     tab1, tab2, tab3 = st.tabs(["PandasAI Analysis", "RAG Q&A", "Data Visualization"])
 
     with tab1:
-        #st.header("Data Analysis with PandasAI")
+        st.subheader("Data Analysis with PandasAI")
         pandas_question = st.text_input("Ask a question about the dataset (PandasAI):")
         if pandas_question:
             try:
@@ -109,9 +121,12 @@ if "df" in st.session_state and api_key and pandasai_api_key:
                 st.write("PandasAI Answer:", result)
             except Exception as e:
                 st.error(f"PandasAI encountered an error: {str(e)}")
+            finally:
+                st.write("PandasAI intermediate output (if any):")
+                st.write(agent.last_output if hasattr(agent, "last_output") else "No intermediate output available.")
 
     with tab2:
-        st.header("Q&A with RAG")
+        st.subheader("Q&A with RAG")
         rag_question = st.text_input("Ask a question about the dataset (RAG):")
         if rag_question:
             try:
@@ -119,9 +134,15 @@ if "df" in st.session_state and api_key and pandasai_api_key:
                 st.write("RAG Answer:", result)
             except Exception as e:
                 st.error(f"RAG encountered an error: {str(e)}")
+            finally:
+                st.write("RAG Intermediate Status:")
+                st.write({
+                    "retriever": retriever,
+                    "qa_chain": qa_chain
+                })
 
     with tab3:
-        st.header("Data Visualization")
+        st.subheader("Data Visualization")
         viz_question = st.text_input("What kind of graph would you like? (e.g., 'Show a scatter plot of salary vs experience')")
         if viz_question:
             try:
@@ -146,6 +167,9 @@ if "df" in st.session_state and api_key and pandasai_api_key:
                     st.write("Unable to generate the graph. Please try a different query.")
             except Exception as e:
                 st.error(f"An error occurred during visualization: {str(e)}")
+            finally:
+                st.write("Visualization debug details:")
+                st.write({"viz_question": viz_question, "result": result})
 else:
     if not api_key:
         st.warning("Please set the OpenAI API key in environment variables.")