Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,8 +14,11 @@ import json
 
 
 
-# Initialize Firebase
-if not firebase_admin._apps:
+
+@st.cache_data
+def load_data():
+    # your Firebase query or CSV load
+    if not firebase_admin._apps:
 
     # HF codes
     firebase_key = os.environ["FIREBASE_KEY"]
@@ -23,10 +26,18 @@ if not firebase_admin._apps:
     
     # Streamlit, Google Cloud
     #cred = credentials.Certificate("/app/src/firebase_key.json")
+
+    # Initialize Firebase
     firebase_admin.initialize_app(cred)
 
-db = firestore.client()
-st.Page.title="Swiftcom DMS"
+    db = firestore.client()
+    return db
+
+    
+
+
+
+st.page.title="Swiftcom DMS"
 st.set_page_config(layout="wide")
 
 
@@ -34,7 +45,7 @@ st.set_page_config(layout="wide")
 # -------------------------------
 # 🔐 LOGIN SECTION
 # -------------------------------
-#@st.dialog("🔐 Login")
+@st.dialog("🔐 Login")
 def login():
     
     st.title("🔐 Login")
@@ -1357,6 +1368,7 @@ def att_managment_page():
 # 🚀 MAIN APP
 # -------------------------------
 def main():
+    load_data()
     if not st.session_state.get("logged_in"):
         login()
         return