Update app.py

app.py

@@ -42,11 +42,9 @@ def get_param_from_url(param):
 
 # 'local' or 'hf'. hf is for Hugging Face file system but has limits on the number of access per hour
 filesystem = 'hf'
-# path to repo or local file system TODO rename
 input_repo_path = 'datasets/emvecchi/therapy_annotation'
 output_repo_path = 'datasets/emvecchi/therapy_annotation/pilot'
 
-
 to_annotate_file_name = 'to_annotate.csv'  # CSV file to annotate
 COLS_TO_SAVE = ['dialogue_id','dialogue_name','generatedPatient']
 
@@ -409,14 +407,18 @@ def render_rupture_markers_widget(index: int, choices: list[str]):
         b_key = f"rupture_marker_{i}_line_b_{index}"
 
         with col1:
-            st.selectbox(
-                f"Rupture marker {i}",
-                options=choices,
-                key=marker_key,
-                index=(choices.index(st.session_state.get(marker_key))
-                       if st.session_state.get(marker_key) in choices else 0)
-                if st.session_state.get(marker_key) is not None else 0
-            )
+            current = st.session_state.get(marker_key, "")
+            if marker_key in st.session_state:
+                st.selectbox(f"Rupture marker {i}", options=choices, key=marker_key)
+            else:
+                default_idx = choices.index(current) if current in choices else 0
+                st.selectbox(f"Rupture marker {i}", options=choices, key=marker_key, index=default_idx)
+#            st.selectbox(
+#                f"Rupture marker {i}", options=choices, key=marker_key,
+#                index=(choices.index(st.session_state.get(marker_key))
+#                       if st.session_state.get(marker_key) in choices else 0)
+#                if st.session_state.get(marker_key) is not None else 0
+#            )
         with col2:
             if RUPTURE_LINE_INPUT_TYPE == "number":
                 st.number_input("Lines: from", min_value=0, step=1, key=a_key)