Update app.py

app.py

@@ -366,37 +366,37 @@ def render_rupture_markers_widget(index: int, choices: list[str]):
                 if st.session_state.get(marker_key) is not None else 0
             )
         with col2:
-            if RUPTURE_LINE_INPUT_TYPE == "number":
-                st.number_input(f"Lines: from", min_value=0, step=1, key=a_key)
-            else:
-                st.text_input(f"Lines: from", key=a_key, value=st.session_state.get(a_key, ""))
+            st.number_input("Lines: from", min_value=0, step=1, key=a_key) \
+                if RUPTURE_LINE_INPUT_TYPE == "number" \
+                else st.text_input("Lines: from", key=a_key, value=st.session_state.get(a_key, ""))
         with col3:
-            if RUPTURE_LINE_INPUT_TYPE == "number":
-                st.number_input(f"Lines: to", min_value=0, step=1, key=b_key)
-            else:
-                st.text_input(f"Lines: to", key=b_key, value=st.session_state.get(b_key, ""))
+            st.number_input("Lines: to", min_value=0, step=1, key=b_key) \
+                if RUPTURE_LINE_INPUT_TYPE == "number" \
+                else st.text_input("Lines: to", key=b_key, value=st.session_state.get(b_key, ""))
 
         st.markdown("---")
 
     a_col, b_col = st.columns([1, 1])
+
+    def _inc():
+        st.session_state[count_key] = min(MAX_RUPTURE_MARKERS, st.session_state[count_key] + 1)
+
+    def _dec():
+        st.session_state[count_key] = max(1, st.session_state[count_key] - 1)
+
     with a_col:
-        add_clicked = st.form_submit_button(
+        st.form_submit_button(
             "➕ Add another rupture marker",
             disabled=(count >= MAX_RUPTURE_MARKERS),
+            on_click=_inc,
         )
     with b_col:
-        remove_clicked = st.form_submit_button(
+        st.form_submit_button(
             "➖ Remove last",
             disabled=(count <= 1),
+            on_click=_dec,
         )
 
-    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 ####################################