Update app.py

app.py

@@ -335,7 +335,8 @@ def render_dialogue(md_path: str,
 
 
 def render_rupture_markers_widget(index: int, choices: list[str]):
-    """Dynamic list of (rupture_marker_i, line_a_i, line_b_i) with Add/Remove."""
+    """Dynamic list of (rupture_marker_i, line_a_i, line_b_i) with Add/Remove.
+       Works inside a st.form by using form_submit_button instead of callbacks."""
     count_key = f"rupture_count_{index}"
     if count_key not in st.session_state:
         st.session_state[count_key] = 1  # start with one block
@@ -344,7 +345,6 @@ def render_rupture_markers_widget(index: int, choices: list[str]):
 
     st.markdown("**What rupture markers are found?** *(one per row; max 8)*")
 
-    # render the i=1..count rows
     for i in range(1, count + 1):
         col1, col2, col3 = st.columns([3, 1, 1])
 
@@ -362,7 +362,6 @@ def render_rupture_markers_widget(index: int, choices: list[str]):
                 if st.session_state.get(marker_key) is not None else 0
             )
 
-        # line inputs (number or text)
         with col2:
             if RUPTURE_LINE_INPUT_TYPE == "number":
                 st.number_input(f"Lines: {i} – from", min_value=1, step=1, key=a_key)
@@ -377,24 +376,28 @@ def render_rupture_markers_widget(index: int, choices: list[str]):
 
         st.markdown("---")
 
-    # add/remove buttons
+    # Use SUBMIT buttons (legal inside forms) to add/remove rows.
     a_col, b_col = st.columns([1, 1])
     with a_col:
-        st.button(
+        add_clicked = st.form_submit_button(
             "➕ Add another rupture marker",
             key=f"add_rup_{index}",
             disabled=(count >= MAX_RUPTURE_MARKERS),
-            on_click=lambda: st.session_state.__setitem__(count_key, min(MAX_RUPTURE_MARKERS, st.session_state[count_key] + 1))
         )
     with b_col:
-        st.button(
+        remove_clicked = st.form_submit_button(
             "➖ Remove last",
             key=f"remove_rup_{index}",
             disabled=(count <= 1),
-            on_click=lambda: st.session_state.__setitem__(count_key, max(1, st.session_state[count_key] - 1))
         )
 
-    # info text
+    if add_clicked and count < MAX_RUPTURE_MARKERS:
+        st.session_state[count_key] = count + 1
+        st.rerun()
+    if remove_clicked and count > 1:
+        st.session_state[count_key] = count - 1
+        st.rerun()
+
     st.caption(f"{st.session_state[count_key]}/{MAX_RUPTURE_MARKERS} markers")
 
 #################################### Streamlit App ####################################