Update app2.py

app2.py

@@ -1024,25 +1024,26 @@ def respond_to_chat(
     # Initialize chat_history if it's None (Gradio might pass None initially)
     if chat_history is None:
         chat_history = []
-
+    
     if chatbot_data is None or not chatbot_data:
         chat_history.append((message, "Please process some data first using the other tabs before chatting."))
-        return chat_history, chatbot_data, current_filtered_df_state # Return existing state
-
+        return chat_history, chatbot_data, current_filtered_df_state  # Return existing state
+    
     # Append user message to history immediately
-    chat_history.append((message, None)) # Use None as a placeholder for the assistant's response
-
+    chat_history.append((message, None))  # Use None as a placeholder for the assistant's response
+    
     response = ""
     lower_message = message.lower().strip()
-
+    
     # Initialize new_filtered_df_state with the current state to preserve it unless a filter changes it
     new_filtered_df_state = current_filtered_df_state
-
+    
     df = None
     try:
         # Attempt to create a DataFrame from the full chatbot_data for analysis
         # This flattens the structure for easier querying with pandas
         flat_data = []
+        
         def flatten_item(d, parent_key='', sep='_'):
             items = {}
             if isinstance(d, dict):
@@ -1063,7 +1064,7 @@ def respond_to_chat(
             # If d is a primitive (int, str, bool, None), it won't add anything here, which is fine
             # as primitives are handled in the dict/list branches.
             return items
-
+    
         # Process each top-level item in chatbot_data
         for i, item in enumerate(chatbot_data):
             if isinstance(item, dict):
@@ -1072,8 +1073,12 @@ def respond_to_chat(
                 flat_data.append(flat_item)
             # If chatbot_data contains non-dict top-level items, flatten them too
             elif isinstance(item, (list, str, int, float, bool, type(None))):
-                 flat_data.append({'item_value': item}) # Wrap primitives in a dict
-
+                flat_data.append({'item_value': item})  # Wrap primitives in a dict
+    
+    except Exception as e:
+        # Handle exceptions that may occur during processing
+        response = f"An error occurred: {str(e)}"
+        chat_history.append((message, response))  # Append error message to chat history
 
         if flat_data:
             try: