Update app.py

app.py

@@ -131,6 +131,9 @@ if len(gene_list) > 1:
     df_shap.drop(columns='XGB_Score', inplace=True)
     shap_values = explainer.shap_values(df_shap)
     summary_plot = shap.summary_plot(shap_values, df_shap)
+    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]
     st.caption("SHAP Summary Plot of All Input Genes")
     st.pyplot(fig=summary_plot)
     st.caption("Interactive SHAP Summary Plot of All Input Genes")
@@ -140,9 +143,6 @@ 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'] = ''