show sentence

app.py

@@ -58,6 +58,21 @@ def clear_data():
     for key in st.session_state:
         del st.session_state[key]
 
+def show_annotated_sentence(sent,option_locs=[],mask_locs=[]):
+    disp_style = '"font-family:san serif; color:Black; font-size: 20px"'
+    prefix = f'<p style={disp_style}><span style="font-weight:bold">'
+    style_list = []
+    for i, word in enumerate(sent):
+        if i in mask_locs:
+            style_list.append(f'<span style="color:Red">{word}</span>')
+        elif i in option_locs:
+            style_list.append(f'<span style="color:Blue">{word}</span>')
+        else:
+            style_list.append(f'{word}')
+    disp = ' '.join(style_list)
+    suffix = '</span></p>'
+    return prefix + disp + suffix
+
 if __name__=='__main__':
     wide_setup()
     load_css('style.css')
@@ -101,7 +116,7 @@ if __name__=='__main__':
                             st.session_state['masked_pos_1'].append(word_id)
                         else:
                             st.session_state['masked_pos_1'].remove(word_id)
-            st.write(f'Masked words: {", ".join([decoded_sent[word_id] for word_id in np.sort(st.session_state["masked_pos_1"])])}')
+            st.markdown(show_annotated_sentence(decoded_sent,mask_locs=st.session_state['masked_pos_1']), unsafe_allow_html = True)
 
 
     if st.session_state['page_status']=='analysis':