Update app.py

app.py

@@ -69,19 +69,20 @@ def convert_df(df):
 if len(gene_list) > 1:
     df = df_total[df_total.index.isin(gene_list)]
     st.dataframe(df)
+    df_shap = df
     df['Gene'] = df.index
     output = df[['Gene', 'XGB_Score']]
     csv = convert_df(output)
     st.download_button(
-       "Download Gene List Prioritisation",
+       "Download Gene Prioritisation",
        csv,
        "bp_gene_prioritisation.csv",
        "text/csv",
        key='download-csv'
     )
-    df.drop(columns='XGB_Score', inplace=True)
-    shap_values = explainer.shap_values(df)
-    summary_plot = shap.summary_plot(shap_values, df, show=False)
+    df_shap.drop(columns='XGB_Score', inplace=True)
+    shap_values = explainer.shap_values(df_shap)
+    summary_plot = shap.summary_plot(shap_values, df_shap, show=False)
     st.caption("SHAP Summary Plot of All Input Genes")
     st.pyplot(fig=summary_plot)