Update app.py

app.py

@@ -3,7 +3,6 @@ import firebase_admin
 from firebase_admin import credentials, auth, db
 import pandas as pd
 import plotly.express as px
-from datetime import datetime
 
 # Initialize Firebase
 try:
@@ -29,7 +28,7 @@ def login():
                 user = auth.get_user_by_email(email)
                 st.session_state['user'] = user
                 st.success("Logged in successfully!")
-                st.experimental_rerun()
+                st.rerun()  # Use st.rerun() instead of st.experimental_rerun()
             except Exception as e:
                 st.error("Invalid credentials. Please try again.")
             return submit_button
@@ -75,13 +74,11 @@ def main():
 
     st.title("💰 Financial Dashboard")
     
-    # Add logout button
     # Add logout button
     if st.sidebar.button("Logout"):
         st.session_state.pop('user', None)
         st.rerun()  # Updated line
 
-    
     # Get user data
     user = st.session_state['user']
     df = get_user_transactions(user.uid)
@@ -95,15 +92,19 @@ def main():
     months = sorted(df['month'].unique())
     selected_months = st.sidebar.multiselect("Select Months", months, default=months)
     
-    # Transaction type filter
+    # Transaction type filter with "All" option
     transaction_types = sorted(df['transactionType'].unique())
-    selected_types = st.sidebar.multiselect("Transaction Types", transaction_types, default=transaction_types)
-    
+    transaction_types.insert(0, "All")  # Add "All" option at the beginning
+    selected_types = st.sidebar.multiselect("Transaction Types", transaction_types, default=transaction_types[1:])
+
     # Filter data
-    filtered_df = df[
-        (df['month'].isin(selected_months)) &
-        (df['transactionType'].isin(selected_types))
-    ]
+    if "All" in selected_types:
+        filtered_df = df[df['month'].isin(selected_months)]
+    else:
+        filtered_df = df[
+            (df['month'].isin(selected_months)) &
+            (df['transactionType'].isin(selected_types))
+        ]
     
     # Dashboard layout
     col1, col2, col3 = st.columns(3)
@@ -182,4 +183,4 @@ def main():
         )
 
 if __name__ == "__main__":
-    main()
+    main()