Update app.py

app.py

@@ -140,6 +140,9 @@ if len(gene_list) > 1:
     max_display=8
     feature_names=df_shap.columns
     gene_index = df_shap.index
+    feature_order = np.argsort(np.sum(np.abs(shap_values), axis=0)[:-1])
+    feature_order = feature_order[-min(8, len(feature_order)):]
+    col_order = [df.columns[i] for i in feature_order]
     for i in range(1, len(plotly_fig['data']), 2):
         t = gene_index #plotly_fig['data']
         plotly_fig['data'][i]['name'] = ''
@@ -162,7 +165,7 @@ if len(gene_list) > 1:
     plotly_fig['layout']['height']=600
     plotly_fig['layout']['width']=500
     plotly_fig['layout']['xaxis'].update(zeroline=True, showline=True, ticklen=4, showgrid=False)
-    plotly_fig['layout']['yaxis'].update(dict(visible=True))
+    plotly_fig['layout']['yaxis'].update(col_order) #(dict(visible=True))
     plotly_fig.add_trace(colorbar_trace)
     plotly_fig.layout.update(
                              annotations=[dict(