Update app.py

app.py

@@ -49,19 +49,23 @@ def initialize_session_state():
     """Initialize all required session state variables"""
     if 'initialized' not in st.session_state:
         try:
+            # Clear any existing metrics first
+            if 'login_attempts_counter' in st.session_state:
+                del st.session_state['login_attempts_counter']
+            
             # Initialize basic session state variables
             st.session_state.initialized = False
             st.session_state.page = 'login'
             st.session_state.user = None
-            st.session_state.login_attempts = {}  # Dictionary to track login attempts
+            st.session_state.login_attempts = {}
             
             # Initialize database connection
             if 'db' not in st.session_state:
                 st.session_state.db = sqlite3.connect('digital_heir.db', check_same_thread=False)
             
-            # Initialize application state
-            if 'digital_heir' not in st.session_state:
-                st.session_state.digital_heir = init_app()
+            # Initialize metrics
+            from utils.metrics import get_login_attempts_counter
+            st.session_state.login_attempts_counter = get_login_attempts_counter()
             
             # Initialize database tables
             initialize_database()