fix word index

app.py

@@ -167,7 +167,7 @@ def show_changed_site():
     prev_step_id = st.session_state.prev_step_id
     curr_sent = df.cleaned_sentence.loc[step_id].split(' ')
     prev_sent = df.cleaned_sentence.loc[prev_step_id].split(' ')
-    locs = [df.next_sample_loc.to_list()[step_id-1]] if 'next_sample_loc' in df else (
+    locs = [df.next_sample_loc.to_list()[step_id-1]]-1 if 'next_sample_loc' in df else (
         [i for i in range(len(curr_sent)) if curr_sent[i] not in prev_sent]
     )
     disp_style = '"font-family:san serif; color:Black; font-size: 20px"'
@@ -261,7 +261,7 @@ if __name__=='__main__':
         if 'step_id' not in st.session_state:
             st.session_state.prev_step_id = 0
             st.session_state.step_id = 0
-            
+
 
         explore_type = st.sidebar.radio(
             '2. Choose how to explore the chain',