Update app3.py

app3.py

@@ -11,6 +11,7 @@ from langchain.chains import RetrievalQA
 from langchain.schema import Document
 import os
 import logging
+import re
 
 # Configure logging
 logging.basicConfig(level=logging.DEBUG)
@@ -84,7 +85,7 @@ else:
         try:
             # Initialize PandasAI Agent
             llm = OpenAI(api_key=pandasai_api_key, max_tokens=1500, timeout=60)
-            agent = Agent(df, llm=llm)
+            agent = Agent(llm)  # Corrected initialization
 
             # Convert the DataFrame into documents for RAG
             documents = [
@@ -152,7 +153,7 @@ else:
                             viz_code = code_match.group(1)
                             logger.debug(f"Extracted visualization code: {viz_code}")
 
-                            # Safeguard: Modify and validate code for Plotly
+                            # Modify and validate code for Plotly
                             viz_code = viz_code.replace('plt.', 'px.')
                             exec(viz_code)  # Execute the visualization code
                             st.plotly_chart(fig)