Update app.py

app.py

@@ -1,8 +1,25 @@
 # app.py
 import streamlit as st
-from modules.database import initialize_cosmos_sql_connection, initialize_mongodb_connection, create_user, create_admin_user, create_student_user
+
+from modules.database import (
+    initialize_database_connections, 
+    initialize_cosmos_sql_connection, 
+    initialize_mongodb_connection, 
+    create_user, 
+    create_admin_user, 
+    create_student_user
+)
+if 'db_initialized' not in st.session_state:
+    st.session_state.db_initialized = initialize_database_connections()
+
+if not st.session_state.db_initialized:
+    st.error("Error initializing database connections")
+    st.stop()
+
 from modules.auth import authenticate_user, register_user
+
 from modules.admin_ui import admin_page
+
 from modules.ui import (
     main,
     login_register_page,