Update app.py

app.py

@@ -15,7 +15,7 @@ import json
 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:
@@ -34,7 +34,16 @@ model = genai.GenerativeModel(
     model_name="gemini-2.0-flash-thinking-exp",
     generation_config=generation_config,
 )
- 
+
+
+# Pandasai gemini
+llm1 = ChatGoogleGenerativeAI(
+    model="gemini-2.0-flash-thinking-exp",
+    temperature=0,
+    max_tokens=None,
+    timeout=None,
+    max_retries=2
+)
 
 def load_data():
     """Load data from CSV files and validate"""
@@ -139,7 +148,7 @@ class StreamLitResponse(ResponseParser):
 
 def generateResponse(prompt, data):
     """Generate response using PandasAI with SmartDataLake"""
-    llm = GoogleGemini(api_key=gemini_api_key)
+    
     
     # Ensure data is a dictionary of DataFrames
     if not isinstance(data, dict) or not all(isinstance(df, pd.DataFrame) for df in data.values()):
@@ -149,7 +158,7 @@ def generateResponse(prompt, data):
     pandas_agent = SmartDatalake(
         list(data.values()),  # Pass list of DataFrames
         config={
-            "llm": llm, 
+            "llm": llm1, 
             "response_parser": StreamLitResponse
         }
     )