debug

app.py

@@ -64,16 +64,16 @@ def annotate_mask(sent_id,sent):
     decoded_sent = [tokenizer.decode([token]) for token in input_sent[1:-1]]
     char_nums = [len(word)+2 for word in decoded_sent]
     cols = st.columns(char_nums)
-    mask_locs = []
+    if f'mask_locs_{sent_id}' not in st.session_state:
+        st.session_state[f'mask_locs_{sent_id}'] = []
     for word_id,(col,word) in enumerate(zip(cols,decoded_sent)):
         with col:
             if st.button(word,key=f'word_{sent_id}_{word_id}'):
-                if word_id not in mask_locs:
-                    mask_locs.append(word_id)
+                if word_id not in st.session_state[f'mask_locs_{sent_id}']:
+                    st.session_state[f'mask_locs_{sent_id}'].append(word_id)
                 else:
-                    mask_locs.remove(word_id)
-    st.markdown(show_annotated_sentence(decoded_sent,mask_locs=mask_locs), unsafe_allow_html = True)
-    st.session_state[f'mask_locs_{sent_id}'] = mask_locs
+                    st.session_state[f'mask_locs_{sent_id}'].remove(word_id)
+    st.markdown(show_annotated_sentence(decoded_sent,mask_locs=st.session_state[f'mask_locs_{sent_id}']), unsafe_allow_html = True)
 
 def annotate_options(sent_id,sent):
     st.write(f'Sentence {sent_id}')
@@ -136,10 +136,8 @@ if __name__=='__main__':
             sent_2 = st.session_state['sent_2']
 
             st.write('2. Select sites to mask out and click "Confirm"')
-            with empty():
-                annotate_mask(1,sent_1)
-            with empty():
-                annotate_mask(2,sent_2)
+            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'):