Update app.py

app.py

@@ -7,17 +7,14 @@ import xgboost
 import shap
 st.set_option('deprecation.showPyplotGlobalUse', False)
 
-# CSS to inject contained in a string
-hide_table_row_index = """
+hide_dataframe_row_index = """
             <style>
-            thead tr th:first-child {display:none}
-            tbody th {display:none}
+            .row_heading.level0 {display:none}
+            .blank {display:none}
             </style>
             """
 
-# Inject CSS with Markdown
-st.markdown(hide_table_row_index, unsafe_allow_html=True)
-
+st.markdown(hide_dataframe_row_index, unsafe_allow_html=True)
 seed=42
 
 annotations = pd.read_csv("annotations_dataset.csv")
@@ -109,18 +106,18 @@ else:
 
 input_gene = st.text_input("Input individual HGNC gene:")
 df2 = df_total[df_total.index == input_gene]
-df2_shap = df_total[df_total.index == input_gene]
 df2['Gene'] = df2.index
 st.dataframe(df2)
-df2.drop(columns='XGB_Score', inplace=True)
 
 if input_gene:
+    df2_shap = df_total[df_total.index == input_gene]
+    df2_shap.drop(columns='XGB_Score', inplace=True)
     shap_values = explainer.shap_values(df2_shap)
     shap.getjs()
     force_plot = shap.force_plot(
     explainer.expected_value,
     shap_values,
-    df2_shap , 
+    df2_shap, 
     matplotlib = True,show=False)
     st.pyplot(fig=force_plot)
 else:
@@ -130,5 +127,6 @@ st.markdown("""
 Total Gene Prioritisation Results:
 """)
 
-df_total['Gene'] = df_total.index
+df_total_output = df_total
+df_total_output['Gene'] = df_total_output.index
 st.dataframe(df_total)