Stale layout

src/streamlit_app.py

@@ -80,6 +80,17 @@ def _on_set_changed():
         st.toast(f"Switched to set {curr}. Cleared current pack & deck.")
 
 
+def _pick_card(card: str):
+    # save undo before mutating
+    push_undo()
+    st.session_state["deck"].append(card)
+
+    # clear the whole pack and any transient widget values
+    st.session_state["pack"].clear()
+
+    # stop this run so nothing else re-renders with stale UI
+
+
 
 if "model" not in st.session_state:
     st.session_state.model = load_model()  # your class
@@ -297,11 +308,12 @@ with tabs[0]:
                     """
                     c2.markdown(tooltip_html, unsafe_allow_html=True)
                     with c3:
-                        if st.button("Pick", key=f"pick_btn_{i}", use_container_width=True, help="Add to deck & clear pack"):
-                            push_undo()
-                            st.session_state["deck"].append(card)
-                            st.session_state["pack"] = [] 
-                            st.rerun()
+                        st.button("Pick", 
+                                key=f"pick_btn_{i}", 
+                                use_container_width=True, 
+                                help="Add to deck & clear pack",
+                                on_click=_pick_card, args=(card,),
+                        )
             else:
                 st.caption("Pack is empty.")