Stale layout

src/streamlit_app.py

@@ -214,7 +214,7 @@ def undo_last():
 # --- Tab 1: Draft -------------------------------------------------------
 draft_area = st.empty()
 with tabs[0]:
-    with draft_area:
+    with draft_area.container():
         if st.session_state.get("set_code") == "EOE":
             st.warning("The model is not trained for this set. Results may be inaccurate.", icon="⚠️")
         
@@ -248,7 +248,7 @@ with tabs[0]:
                 h1.markdown("**Card**")
                 h2.markdown("**Remove?**")
                 for i, card in enumerate(st.session_state["deck"]):
-                    name_col, rm_col = st.columns([6, 3], gap="small")
+                    name_col, rm_col = st.columns([6, 2], gap="small")
                     name_col.write(card)
                     with rm_col:
                         if st.button("❌", key=f"rm-deck-{i}", use_container_width=True):
@@ -277,6 +277,7 @@ with tabs[0]:
                     pick, logits, scores = rank_cards(deck, pack)
                 except Exception as e:
                     st.error(f"Error calculating card rankings: {e}")
+                    pick, logits, scores = None, {}, {}
 
                 st.button("🗑️ Clear pack", on_click=lambda: st.session_state.update(pack=[]), use_container_width=True)
                 h1, h2, h3 = st.columns([6, 2, 3])
@@ -297,7 +298,7 @@ with tabs[0]:
                     logit = row["Logits"]
 
 
-                    c1, c2, c3 = st.columns([6, 2, 3], gap="small")
+                    c1, c2, c3 = st.columns([6, 2, 2], gap="small")
                     c1.write(card)
                     
                     tooltip_html = f"""
@@ -307,7 +308,7 @@ with tabs[0]:
                     """
                     c2.markdown(tooltip_html, unsafe_allow_html=True)
                     with c3:
-                        st.button("Pick", 
+                        st.button("✔️", 
                                 key=f"pick_btn_{i}", 
                                 use_container_width=True, 
                                 help="Add to deck & clear pack",