Update app.py

app.py

@@ -79,6 +79,12 @@ explainer = shap.TreeExplainer(xgb)
 def convert_df(df):
    return df.to_csv(index=False).encode('utf-8')
 
+features = df[['Gene','XGB_Score', 'mousescore_Exomiser',
+ 'SDI', 'Liver_GTExTPM',  'pLI_ExAC',
+ 'HIPred',
+ 'Cells - EBV-transformed lymphocytes_GTExTPM',
+ 'Pituitary_GTExTPM',
+ 'IPA_BP_annotation']]
 
 if len(gene_list) > 1:
     df = df_total[df_total.index.isin(gene_list)]
@@ -104,7 +110,7 @@ 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)
-    summary_plot = shap_summary_plot(shap_values, features=df, feature_names=df.columns, max_display=8)
+    summary_plot = shap_summary_plot(shap_values, features=features, feature_names=features.columns, max_display=8)
     st.pyplot(fig=summary_plot)
     st.caption("SHAP Summary Plot of All Input Genes")
     feature_order = np.argsort(np.sum(np.abs(shap_values), axis=0)[:-1])
@@ -128,7 +134,7 @@ st.dataframe(df2)
 
 if input_gene:
     if ' ' in input_gene or ',' in input_gene:
-        st.write('Please input only a single HGNC gene name.')
+        st.write('Input Error: Please input only a single HGNC gene name with no white spaces or commas.')
     else:
         df2_shap = df_total[df_total.index == input_gene]
         df2_shap.drop(columns='XGB_Score', inplace=True)
@@ -146,7 +152,7 @@ else:
     pass
 
 st.markdown("""
-Total Gene Prioritisation Results:
+### Total Gene Prioritisation Results:
 """)
 
 df_total_output = df_total