debug

app.py

@@ -84,14 +84,14 @@ def annotate_options(sent_id,sent):
     option_locs = []
     for word_id,(col,word) in enumerate(zip(cols,decoded_sent)):
         with col:
-            if st.button(word,key=f'word_{word_id}'):
+            if st.button(word,key=f'word_{sent_id}_{word_id}'):
                 if word_id not in option_locs:
                     option_locs.append(word_id)
                 else:
                     option_locs.remove(word_id)
     st.markdown(show_annotated_sentence(decoded_sent,option_locs=option_locs,
                                         mask_locs=st.session_state[f'mask_locs_{sent_id}']), unsafe_allow_html = True)
-    return option_locs
+    st.session_state[f'option_locs_{sent_id}'] = option_locs
 
 def show_annotated_sentence(sent,option_locs=[],mask_locs=[]):
     disp_style = '"font-family:san serif; color:Black; font-size: 20px"'
@@ -138,6 +138,8 @@ if __name__=='__main__':
             st.write('2. Select sites to mask out and click "Confirm"')
             annotate_mask(1,sent_1)
             annotate_mask(2,sent_2)
+            st.write(st.session_state['mask_locs_1'])
+            st.write(st.session_state['mask_locs_2'])
             if st.button('Confirm'):
                 st.session_state['page_status'] = 'annotate_options'
                 main_area.empty()