fixed bugs

app.py

@@ -79,7 +79,8 @@ def run_tsne(chain):
     return tsne
 
 def clear_df():
-    del st.session_state['df']
+    if 'df' in st.session_state:
+        del st.session_state['df']
 
 #@st.cache(show_spinner=False)
 def plot_fig(df,sent_id,xlims,ylims,color_list):
@@ -165,14 +166,15 @@ if __name__=='__main__':
 
     if input_type=='Use one of the example sentences':
         sentence = st.sidebar.selectbox("Select the inital sentence",
-                                ('About 170 campers attend the camps each week.',
+                                ('--Please select one from below--',
+                                'About 170 campers attend the camps each week.',
                                 'She grew up with three brothers and ten sisters.'))
-        if sentence=='About 170 campers attend the camps each week.':
-            sentence_num = 6
-        else:
-            sentence_num = 8
-
-        st.session_state.df = load_data(sentence_num)
+        if sentence!='--Please select one from below--':
+            if sentence=='About 170 campers attend the camps each week.':
+                sentence_num = 6
+            elif sentence=='She grew up with three brothers and ten sisters.':
+                sentence_num = 8
+            st.session_state.df = load_data(sentence_num)
 
     else:
         sentence = st.sidebar.text_input('Type down your own sentence here.',on_change=clear_df)
@@ -255,12 +257,12 @@ if __name__=='__main__':
                     st.write(decoded_sent[-1])
                 for word_id,(col,word) in enumerate(zip(cols[1:-1],decoded_sent[1:-1])):
                     with col:
-                        if st.button(word):
+                        if st.button(word,key=f'word_{word_id}'):
                             probs = mask_prob(model,mask_id,input_sent,word_id+1)
                             _,candidates_df = sample_words(probs, word_id+1, input_sent)
                             st.table(candidates_df)
             else:
-                disp_style = '"font-family:san serif; color:Black; font-size: 25px; font-weight:bold"'
+                disp_style = '"font-family:san serif; color:Black; font-size: 20px; font-weight:bold"'
                 if explore_type=='Click through each step' and input_type=='Use your own initial sentence' and sent_id>0 and 'finished_sampling' in st.session_state:
                     sampled_loc = df.next_sample_loc.to_list()[sent_id-1]
                     disp_sent_before = f'<p style={disp_style}>'+' '.join(decoded_sent[1:sampled_loc])