Update views.py

views.py

@@ -7,6 +7,22 @@ from services import CryptoService, EmailService
 from utils.security import validate_password, check_rate_limit
 from utils.audit import audit_logger
 
+def initialize_services():
+    if hasattr(st.session_state, 'db'):
+        audit_logger.initialize_with_connection(st.session_state.db)
+
+if __name__ == "__main__":
+    st.set_page_config(
+        page_title="Digital Inheritance System",
+        page_icon="🔐",
+        layout="wide"
+    )
+    initialize_session_state()
+    initialize_services()
+    main()
+
+
+
 # Initialize audit logger with connection after database is set up
 def initialize_audit_logger():
     if 'db' in st.session_state: