Update app.py

app.py

@@ -445,6 +445,42 @@ def _is_default_value(f: Field, val):
     """Check if current value equals the default for this widget type."""
     return val == INPUT_FIELD_DEFAULT_VALUES.get(f.type)
 
+def prime_rupture_markers_from_saved(index: int, data_collected: dict | None):
+    """
+    Copy saved JSON values (rupture_marker_1, *_line_a, *_line_b) into
+    st.session_state keys used by the widget (…_{index}), and set the count.
+    Runs once per index to avoid clobbering user edits on reruns.
+    """
+    if not data_collected:
+        return
+
+    primed_flag = f"rupture_primed_{index}"
+    if st.session_state.get(primed_flag):
+        return  # already restored for this page
+
+    slots = []
+    for i in range(1, MAX_RUPTURE_MARKERS + 1):
+        m = data_collected.get(f"rupture_marker_{i}")
+        a = data_collected.get(f"rupture_marker_{i}_line_a")
+        b = data_collected.get(f"rupture_marker_{i}_line_b")
+        if any(v is not None for v in (m, a, b)):
+            # widget's selectbox uses "" for "no selection"
+            slots.append(((m or ""), a, b))
+
+    if not slots:
+        return
+
+    # set widget count (at least 1, capped to MAX)
+    count_key = f"rupture_count_{index}"
+    st.session_state[count_key] = min(MAX_RUPTURE_MARKERS, max(1, len(slots)))
+
+    for i, (m, a, b) in enumerate(slots, start=1):
+        st.session_state[f"rupture_marker_{i}_{index}"] = m if m in rupture_choices else (m or "")
+        st.session_state[f"rupture_marker_{i}_line_a_{index}"] = a
+        st.session_state[f"rupture_marker_{i}_line_b_{index}"] = b
+
+    st.session_state[primed_flag] = True
+
 def validate_current_page(fields: List[Field], index: int) -> bool:
     """
     Walk the field tree and verify mandatory inputs are filled.
@@ -586,6 +622,7 @@ def show_field(f: Field, index: int, data_collected):
                 st.checkbox(f.title, key=f.name, value=False)
 
             case 'rupture_markers':
+                prime_rupture_markers_from_saved(index, data_collected)
                 render_rupture_markers_widget(index, rupture_choices)
 
         return