Update Home.py

Home.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import os
 
 # ── 1. CONFIG & ASSETS ────────────────────────────────────────────────────────
 st.set_page_config(page_title="Wercore | App Generator", page_icon="🏢", layout="wide")
@@ -39,7 +40,6 @@ if not st.session_state.get("authenticated", False):
         st.title("🔒 Wercore Portal")
         st.info("Please log in to access the reporting engines.")
         
-        # Change "admin123" to whatever password you want your team to use
         password = st.text_input("Master Password", type="password")
         if st.button("Secure Login", type="primary", use_container_width=True):
             if password == "admin123": 
@@ -54,21 +54,28 @@ st.title("Wercore App Generator")
 st.markdown("Select a reporting engine below to begin generating compliance documentation.")
 st.markdown("---")
 
-# ── 4. NAVIGATION GRID ────────────────────────────────────────────────────────
+# ── 4. NAVIGATION GRID (CRASH-PROOF) ──────────────────────────────────────────
+# This function stops the app from crashing if a file name is misspelled
+def safe_link(file_path, label, icon):
+    if os.path.exists(file_path):
+        st.page_link(file_path, label=label, icon=icon)
+    else:
+        st.warning(f"⚠️ **Missing File:** Please create `{file_path}` to activate this button.")
+
 col1, col2, col3 = st.columns(3)
 
 with col1:
-    st.page_link("pages/1_Water_Tank_Report.py", label="Water Tank Disinfection", icon="💧")
+    safe_link("pages/1_Water_Tank_Report.py", "Water Tank Disinfection", "💧")
     st.write("") # Spacer
-    st.page_link("pages/4_Pipeline_Disinfection.py", label="Pipeline Disinfection", icon="🚰")
+    safe_link("pages/4_Pipeline_Disinfection.py", "Pipeline Disinfection", "🚰")
 
 with col2:
-    st.page_link("pages/2_Pest_Control_Report.py", label="Pest Control Report", icon="🪳")
+    safe_link("pages/2_Pest_Control_Report.py", "Pest Control Report", "🪳")
     st.write("") # Spacer
-    st.page_link("pages/5_Site_Survey_Report.py", label="Site Survey", icon="📋")
+    safe_link("pages/5_Site_Survey_Report.py", "Site Survey", "📋")
 
 with col3:
-    st.page_link("pages/3_Termite_Control.py", label="Termite Control", icon="🐜")
+    safe_link("pages/3_Termite_Control.py", "Termite Control", "🐜")
 
 st.markdown("---")
 st.caption("© 2026 Sterdil Labs")