Update app.py

app.py

@@ -32,13 +32,20 @@ model = genai.GenerativeModel(
     model_name="gemini-2.0-flash-thinking-exp",
     generation_config=generation_config,
 )
+ 
 
 def load_data():
-    """Load data from CSV files"""
+    """Load data from CSV files and validate"""
     try:
         events_df = pd.read_csv("Delta-Events.csv")
         customers_df = pd.read_csv("delta_customers.csv")
         products_df = pd.read_csv("Customer_Products.csv")
+        
+        # Validate data
+        if events_df.empty or customers_df.empty or products_df.empty:
+            st.error("One or more data files are empty.")
+            return None
+        
         return {
             'events': events_df,
             'customers': customers_df,
@@ -182,8 +189,14 @@ 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()):
+        st.error("Invalid data format. Expected a dictionary of DataFrames.")
+        return None
+    
     pandas_agent = SmartDatalake(
-        data.values(),
+        list(data.values()),  # Pass list of DataFrames
         config={
             "llm": llm, 
             "response_parser": StreamLitResponse
@@ -225,7 +238,6 @@ def handle_userinput(question, data):
             })
             
             result = generateResponse(question, data)
-            # ... [Keep the rest of handle_userinput same as before]
             if isinstance(result, dict):
                 response_type = result.get('type', 'text')
                 response_value = result.get('value')
@@ -252,7 +264,7 @@ def handle_userinput(question, data):
                     "content": str(result)
                 })
         else:
-            st.write("No data loaded.")
+            st.error("No valid data available for analysis.")
     except Exception as e:
         st.error(f"Error processing input: {e}")
 
@@ -266,7 +278,6 @@ def main():
     if "data" not in st.session_state:
         st.session_state.data = load_data()
 
-
     # Create tabs
     tab_dashboard, tab_chat, tab_reports = st.tabs(["📊 Dashboard", "💬 Chat", "📈 Reports"])
 
@@ -280,22 +291,21 @@ def main():
     # Chat Tab
     with tab_chat:
         st.title("AI Data Analyst")
-        # ... [Keep chat interface similar but update handle_userinput calls to use st.session_state.data]
         chat_container = st.container()
         with chat_container:
             for message in st.session_state.chat_history:
                 with st.chat_message(message["role"]):
                     render_chat_message(message)
-        # Chat input
+        
         user_question = st.chat_input("Ask a question about your data:")
         if user_question:
             handle_userinput(user_question, st.session_state.data)
-            # Update chat container immediately after processing the input
             chat_container.empty()
             with chat_container:
                 for message in st.session_state.chat_history:
                     with st.chat_message(message["role"]):
                         render_chat_message(message)
+
     # Reports Tab
     with tab_reports:
         st.title("Custom Reports")
@@ -333,16 +343,15 @@ def main():
                     report = response.text
                     
                     # PDF Generation and display
-                    # ... [Keep the PDF generation code from original]
                     try:
-                        pdf_bytes = generate_pdf(report)  # Use generate_pdf
+                        pdf_bytes = generate_pdf(report)
                         st.download_button(
                             label="Download Report as PDF",
                             data=pdf_bytes,
                             file_name="report.pdf",
                             mime="application/pdf"
                         )
-                        st.markdown(report)  # Display the report below the download button
+                        st.markdown(report)
                     except Exception as e:
                         st.error(f"Error generating PDF: {e}")
                         st.markdown(report)