Update app.py

app.py

@@ -43,7 +43,13 @@ def create_tables():
 
 create_tables()
 
-# Sidebar Navigation
+# Sidebar Navigation with Session State to retain page state
+if "page" not in st.session_state:
+    st.session_state.page = "Home"
+
+def navigate_to(page):
+    st.session_state.page = page
+
 st.sidebar.title("CAFM Navigation")
 
 # Define navigation options
@@ -55,21 +61,19 @@ pages = {
     "Reporting & Analytics": "📊"
 }
 
-selected_page = None
-
 st.sidebar.markdown("---")
 for page, icon in pages.items():
     if st.sidebar.button(f"{icon} {page}"):
-        selected_page = page
+        navigate_to(page)
 
-if selected_page is None:
-    selected_page = "Home"  # Default page
+# Main Page Logic
+selected_page = st.session_state.page
 
 # Home Page
 if selected_page == "Home":
     st.title("Welcome to the Computer-Aided Facility Management (CAFM) System")
     st.image(
-        "https://via.placeholder.com/800x300.png?text=CAFM+System", 
+        "https://via.placeholder.com/800x300.png?text=CAFM+System",
         use_container_width=True
     )
     st.markdown("""
@@ -82,7 +86,7 @@ elif selected_page == "Asset Management":
     st.title("Asset Management")
     st.markdown("Manage and monitor your facility's assets efficiently.")
     
-    with st.form("asset_form", clear_on_submit=True):
+    with st.form("asset_form"):
         site_name = st.text_input("Site Name")
         asset_name = st.text_input("Asset Name")
         asset_type = st.text_input("Asset Type")
@@ -97,6 +101,7 @@ elif selected_page == "Asset Management":
             ''', (site_name, asset_name, asset_type, maintenance_history, warranty, current_condition))
             conn.commit()
             st.success("Asset added successfully!")
+            st.experimental_rerun()
 
     st.subheader("Asset List")
     assets = pd.read_sql_query("SELECT * FROM assets", conn)
@@ -107,7 +112,7 @@ elif selected_page == "Work Order Management":
     st.title("Work Order Management")
     st.markdown("Streamline the management of work orders from creation to completion.")
 
-    with st.form("work_order_form", clear_on_submit=True):
+    with st.form("work_order_form"):
         site_name = st.text_input("Site Name")
         work_order_description = st.text_area("Work Order Description")
         assigned_to = st.text_input("Assigned To")
@@ -120,6 +125,7 @@ elif selected_page == "Work Order Management":
             ''', (site_name, work_order_description, assigned_to, status, datetime.now().strftime("%Y-%m-%d %H:%M:%S")))
             conn.commit()
             st.success("Work order created successfully!")
+            st.experimental_rerun()
 
     st.subheader("Work Orders")
     work_orders = pd.read_sql_query("SELECT * FROM work_orders", conn)
@@ -130,7 +136,7 @@ elif selected_page == "Preventive Maintenance":
     st.title("Preventive Maintenance")
     st.markdown("Plan and manage preventive maintenance tasks to ensure asset longevity.")
 
-    with st.form("maintenance_form", clear_on_submit=True):
+    with st.form("maintenance_form"):
         site_name = st.text_input("Site Name")
         task_name = st.text_input("Task Name")
         schedule_date = st.date_input("Schedule Date")
@@ -143,6 +149,7 @@ elif selected_page == "Preventive Maintenance":
             ''', (site_name, task_name, schedule_date, status))
             conn.commit()
             st.success("Task scheduled successfully!")
+            st.experimental_rerun()
 
     st.subheader("Scheduled Maintenance")
     maintenance = pd.read_sql_query("SELECT * FROM preventive_maintenance", conn)