Update page_files/Home.py

page_files/Home.py

@@ -25,6 +25,14 @@ if "visible_cards" not in st.session_state:
 
 VISIBLE_CARDS = st.session_state.visible_cards 
 prop_count = len([c for c in ALL_CARDS if c[2] == "section"])
+# Navigation via session state — fires before any rendering
+if st.session_state.get("_go_extract"):
+    st.session_state["_go_extract"] = False
+    st.switch_page("page_files/Upload_Data.py")
+if st.session_state.get("_go_search"):
+    st.session_state["_go_search"] = False
+    st.switch_page("page_files/Categorized_Search.py")
+
 
 def get_card_icon(title: str, card_type: str) -> str:
     if card_type == "material":
@@ -200,6 +208,14 @@ elif _nav == "search":
 
 
 
+# Navigation handler - must be before any rendering
+if st.session_state.get("_go_extract"):
+    st.session_state["_go_extract"] = False
+    st.switch_page("page_files/Upload_Data.py")
+if st.session_state.get("_go_search"):
+    st.session_state["_go_search"] = False
+    st.switch_page("page_files/Categorized_Search.py")
+
 #  Helper 
 def go_categorized(material=None, search=None):
     if material:
@@ -514,14 +530,6 @@ st.markdown("""
 
 # Showcase section - Databricks style
 
-# Handle navigation via session state — avoids toggle on re-render
-if st.session_state.get("_go_extract"):
-    st.session_state["_go_extract"] = False
-    st.switch_page("page_files/Upload_Data.py")
-if st.session_state.get("_go_search"):
-    st.session_state["_go_search"] = False
-    st.switch_page("page_files/Categorized_Search.py")
-
 st.markdown("""
 <style>
 .st-key-btn_extract button, .st-key-btn_search button {