Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 import pandasai as PandasAI
-from pandasai import SmartDatalake
+from pandasai import SmartDatalake, SmartDataframe
 from pandasai.responses.response_parser import ResponseParser
 from pandasai.llm import GoogleGemini
 import plotly.express as px
@@ -17,6 +17,7 @@ import os
 from markdown_pdf import MarkdownPdf, Section
 import tempfile
 from langchain_google_genai import ChatGoogleGenerativeAI
+
 # Configure Gemini API
 gemini_api_key = os.environ.get('GOOGLE_API_KEY')
 if not gemini_api_key:
@@ -46,7 +47,7 @@ llm1 = ChatGoogleGenerativeAI(
     max_retries=2
 )
 
-pandas_ai = PandasAI(llm=llm1)
+
 
 def load_data():
     """Load data from CSV files and validate"""
@@ -70,13 +71,18 @@ def load_data():
         return None
 
 
-def my_insights(df):
-    insights = pandas_ai(df)
-    return insights
 
-events_i = my_insights(events_df)
-customer_i = my_insights(customers_df)
-products_i = my_insights(products_df)
+
+events_i = SmartDataframe(events_df,  config={
+            "llm": llm1
+        })
+customer_i = SmartDataframe(customers_df,  config={
+            "llm": llm1
+            
+        })
+products_i = SmartDataframe(products_df,  config={
+            "llm": llm1,
+        })
 #Dashboard
 def create_dashboard(data):
     """Create dashboard visualizations"""