Update app.py

app.py

@@ -101,7 +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.session_state.page = "Asset Management"
 
     st.subheader("Asset List")
     assets = pd.read_sql_query("SELECT * FROM assets", conn)
@@ -125,7 +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.session_state.page = "Work Order Management"
 
     st.subheader("Work Orders")
     work_orders = pd.read_sql_query("SELECT * FROM work_orders", conn)
@@ -149,7 +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.session_state.page = "Preventive Maintenance"
 
     st.subheader("Scheduled Maintenance")
     maintenance = pd.read_sql_query("SELECT * FROM preventive_maintenance", conn)