Update app.py

app.py

@@ -114,7 +114,13 @@ if uploaded_file:
             all_anomalies['monetary value'] = all_anomalies['monetary value'].apply(lambda x: f"{x:.2f}")
             all_anomalies.sort_values(by=['monetary value'], ascending=False, inplace=True)
             all_anomalies = all_anomalies.fillna('').astype(str)
-            st.dataframe(st.session_state['all_anomalies'].head())
+            
+            # Store in session state
+            st.session_state['all_anomalies'] = all_anomalies
+        
+    # Display the dataframe if it exists in session state
+    if 'all_anomalies' in st.session_state:
+        st.dataframe(st.session_state['all_anomalies'].head())
     else:
         st.warning("Ainda não há dados de anomalias para exibir. Por favor, carregue um arquivo.")
 
@@ -145,7 +151,10 @@ def response(user_question, table_data):
     return response_pt
 
 # Streamlit interface
-st.dataframe(st.session_state['all_anomalies'].head())
+if 'all_anomalies' in st.session_state:
+    st.dataframe(st.session_state['all_anomalies'].head())
+else:
+    st.warning("Ainda não há dados de anomalias para exibir. Por favor, carregue um arquivo.")
 
 # Chat history
 if 'history' not in st.session_state: