Merge pull request #254 from Proteobench/highlight_column_fix

Fix highlight column being uploaded
Proteobench · Feb 16, 2024 · 838e0c9 · 838e0c9
2 parents 3aa64ee + 05c6a9c
commit 838e0c9
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/webinterface/pages/DDA_Quant_ion.py b/webinterface/pages/DDA_Quant_ion.py
@@ -248,9 +248,9 @@ def _main_page(self):
                 filter_df_numquant_nr_prec, min_quant=default_val_slider
             )
 
-            if HIGHLIGHT_LIST not in st.session_state.keys():
+            if HIGHLIGHT_LIST not in st.session_state.keys() and "Highlight" not in all_datapoints.columns:
                 all_datapoints.insert(0, "Highlight", [False] * len(all_datapoints.index))
-            else:
+            elif "Highlight" not in all_datapoints.columns:
                 all_datapoints.insert(0, "Highlight", st.session_state[HIGHLIGHT_LIST])
 
             st.markdown(
@@ -635,8 +635,12 @@ def generate_results(
                 st.session_state[SUBMIT] = True
                 user_comments = self.user_input["comments_for_submission"]
                 if not LOCAL_DEVELOPMENT:
+                    submit_df = st.session_state[ALL_DATAPOINTS]
+                    if "Highlight" in submit_df.columns:
+                        submit_df.drop("Highlight", inplace=True)
+
                     pr_url = IonModule().clone_pr(
-                        st.session_state[ALL_DATAPOINTS],
+                        submit_df,
                         params,
                         st.secrets["gh"]["token"],
                         username="Proteobot",