debug

app.py

@@ -68,12 +68,13 @@ def annotate_mask(sent_id,sent):
         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 st.button(word,key=f'word_mask_{sent_id}_{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:
                     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)
+    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}')
@@ -81,17 +82,18 @@ def annotate_options(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)
-    option_locs = []
+    if f'option_locs_{sent_id}' not in st.session_state:
+        st.session_state[f'option_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 option_locs:
-                    option_locs.append(word_id)
+            if st.button(word,key=f'word_option_{sent_id}_{word_id}'):
+                if word_id not in st.session_state[f'option_locs_{sent_id}']:
+                    st.session_state[f'option_locs_{sent_id}'].append(word_id)
                 else:
-                    option_locs.remove(word_id)
-    st.markdown(show_annotated_sentence(decoded_sent,option_locs=option_locs,
+                    st.session_state[f'option_locs_{sent_id}'].remove(word_id)
+    st.markdown(show_annotated_sentence(decoded_sent,
+                                        option_locs=st.session_state[f'option_locs_{sent_id}'],
                                         mask_locs=st.session_state[f'mask_locs_{sent_id}']), unsafe_allow_html = True)
-    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"'