Update app.py

app.py

@@ -62,9 +62,21 @@ input_gene_list = st.text_input("Input list of HGNC genes (enter comma separated
 gene_list = collect_genes(input_gene_list)
 explainer = shap.TreeExplainer(xgb)
 
+@st.experimental_memo
+def convert_df(df):
+   return df.to_csv(index=False).encode('utf-8')
+
 if len(gene_list) > 1:
     df = df_total[df_total.index.isin(gene_list)]
     st.dataframe(df)
+    csv = convert_df(df)
+    st.download_button(
+       "Press to Download Gene List Prioritisation",
+       csv,
+       "file.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)