multiclicks

app.py

@@ -123,6 +123,15 @@ def pre_render_images(df,input_sent_id):
         fig_list.append(img)
     return sent_list,fig_list
 
+def update_sent_id(increment_value=0):
+    sent_id = st.session_state.sent_id
+    sent_id += increment_value
+    sent_id = min(len(st.session_state.df)-1,max(0,sent_id))
+    st.session_state.sent_id = sent_id
+
+def initialize_sent_id():
+    st.session_state.sent_id = st.session_state.sent_id_from_slider
+
 
 if __name__=='__main__':
     # Config
@@ -186,8 +195,8 @@ if __name__=='__main__':
 
     if 'df' in st.session_state:
         df = st.session_state.df
-        sent_id = st.sidebar.slider(label='2. Select a position in the chain to start exploring',
-                                    min_value=0,max_value=len(df)-1,value=0)
+        st.sidebar.slider(label='2. Select a position in the chain to start exploring',
+                            min_value=0,max_value=len(df)-1,value=0,key='sent_id_from_slider')
 
         if input_type=='Use one of the example sentences':
             explore_type = st.sidebar.radio('3. Choose the way to explore',options=['In fixed increments','Click through each step','Autoplay'])
@@ -221,12 +230,14 @@ if __name__=='__main__':
                     st.video(f)
         else:
             if explore_type=='In fixed increments':
-                button_labels = ['-500','-100','-10','-1','0','+1','+10','+100','+500']
-                increment = st.sidebar.radio(label='select increment',options=button_labels,index=4)
-                sent_id += int(increment.replace('+',''))
-                sent_id = min(len(df)-1,max(0,sent_id))
+                button_options = [-500,-100,-10,-1,1,10,100,500]
+                button_labels = ['-500','-100','-10','-1','+1','+10','+100','+500']
+                cols = st.sidebar.columns(len(button_options))
+                for col_id,col in enumerate(cols):
+                    with col:
+                        st.button(button_labels[col_id],on_click=update_sent_id,kwargs=dict(increment_value=button_options[col_id]))
             elif explore_type=='Click through each step':
-                sent_id = st.sidebar.number_input(label='step number',value=sent_id)
+                st.session_state.sent_id = st.sidebar.number_input(label='step number',value=sent_id)
 
             x_tsne, y_tsne = df.x_tsne, df.y_tsne
             xscale_unit = (max(x_tsne)-min(x_tsne))/10
@@ -235,6 +246,7 @@ if __name__=='__main__':
             ylims = [(min(y_tsne)//yscale_unit-1)*yscale_unit,(max(y_tsne)//yscale_unit+1)*yscale_unit]
             color_list = sns.color_palette('flare',n_colors=int(len(df)*1.2))
 
+            sent_id = st.session_state.sent_id
             fig = plt.figure(figsize=(5,5),dpi=200)
             ax = fig.add_subplot(1,1,1)
             ax.plot(x_tsne[:sent_id+1],y_tsne[:sent_id+1],linewidth=0.2,color='gray',zorder=1)