Update app3.py

app3.py

@@ -3,6 +3,7 @@ import pandas as pd
 import plotly.express as px
 from datasets import load_dataset
 from pandasai import Agent
+from pandasai.llm.openai import OpenAI
 from langchain_community.embeddings.openai import OpenAIEmbeddings
 from langchain_community.vectorstores import FAISS
 from langchain_openai import ChatOpenAI
@@ -32,8 +33,11 @@ if missing_keys:
         f"The following API key(s) are missing: {missing_keys_str}. Please set them in the environment."
     )
 
+logger.debug(f"OPENAI_API_KEY: {api_key}")
+logger.debug(f"PANDASAI_API_KEY: {pandasai_api_key}")
+
 # Title of the app
-st.title("Data Analyzer")
+st.title("PandasAI Data Analyzer with RAG")
 
 # Function to load datasets into session
 def load_dataset_into_session():
@@ -62,7 +66,6 @@ def load_dataset_into_session():
         if st.button("Load Hugging Face Dataset"):
             try:
                 dataset = load_dataset(dataset_name, split="train", trust_remote_code=True)
-                # Convert Hugging Face dataset to Pandas DataFrame
                 if hasattr(dataset, "to_pandas"):
                     st.session_state.df = dataset.to_pandas()
                 else:
@@ -97,7 +100,8 @@ if st.session_state.df is not None:
     df = st.session_state.df
     try:
         # Initialize PandasAI Agent
-        agent = Agent(df)
+        llm = OpenAI(api_key=pandasai_api_key)
+        agent = Agent(df, llm=llm)
 
         # Convert DataFrame to documents for RAG
         documents = [
@@ -132,9 +136,13 @@ if st.session_state.df is not None:
             if pandas_question:
                 try:
                     result = agent.chat(pandas_question)
-                    st.write("PandasAI Answer:", result)
+                    if result:
+                        st.write("PandasAI Answer:", result)
+                    else:
+                        st.warning("PandasAI returned no result. Try another question.")
                 except Exception as e:
                     st.error(f"Error during PandasAI Analysis: {e}")
+                    logger.error(f"PandasAI Analysis error: {e}")
 
         # Tab 2: RAG Q&A
         with tab2:
@@ -146,6 +154,7 @@ if st.session_state.df is not None:
                     st.write("RAG Answer:", result)
                 except Exception as e:
                     st.error(f"Error during RAG Q&A: {e}")
+                    logger.error(f"RAG Q&A error: {e}")
 
         # Tab 3: Data Visualization
         with tab3:
@@ -157,7 +166,6 @@ if st.session_state.df is not None:
                 try:
                     result = agent.chat(viz_question)
 
-                    # Extract Python code for visualization
                     import re
 
                     code_pattern = r"```python\n(.*?)\n```"
@@ -165,9 +173,8 @@ if st.session_state.df is not None:
 
                     if code_match:
                         viz_code = code_match.group(1)
-                        # Replace matplotlib (plt) code with Plotly (px)
                         viz_code = viz_code.replace("plt.", "px.")
-                        exec(viz_code)  # Execute the visualization code
+                        exec(viz_code)
                         st.plotly_chart(fig)
                     else:
                         st.warning("Could not generate a graph. Try a different query.")