add options to turn features off

app.py

@@ -174,7 +174,13 @@ with st.sidebar:
     st.header("Upload qrels")
     qrels_file = st.file_uploader("Choose a file", key="qrels")
     qrels = load_local_qrels(qrels_file)
-
+    # add a checkbox to turn off highlighting
+    st.header("Highlighting Off")
+    highlighting_off = st.checkbox("Turn off highlighting", key="highlighting_off")
+    # add a checkbox to turn off word suggestions
+    st.header("Word Suggestions Off")
+    word_suggestions_off = st.checkbox("Turn off word suggestions", key="word_suggestions_off")
+    
     ## make sure all qids in qrels are in queries and write out a warning if not
     if queries is not None and qrels is not None:
         missing_qids = set(qrels.keys()) - set(queries.keys()) | set(queries.keys()) - set(qrels.keys())
@@ -271,7 +277,12 @@ if corpus is not None and queries is not None and qrels is not None:
             # relevant
             relevant_docs = list(qrels[str(inst_num)].keys())[:n_relevant_docs]
             doc_texts = [(doc_id, corpus[doc_id]["title"] if "title" in corpus[doc_id] else "", corpus[doc_id]["text"]) for doc_id in relevant_docs]
-            splitting_words = find_dividing_words([item[1] + " " + item[2] for item in doc_texts])
+            
+            if word_suggestions_off:
+                splitting_words = []
+            else:
+                splitting_words = find_dividing_words([item[1] + " " + item[2] for item in doc_texts])
+
 
             # make a selectbox of these splitting words (allow multiple)
             container.subheader("Splitting Words")
@@ -284,7 +295,7 @@ if corpus is not None and queries is not None and qrels is not None:
             highlighted_texts = []
             highlighted_titles = []
             for (docid, title, text) in tqdm.tqdm(doc_texts):
-                if not len(splitting_word_select):
+                if not len(splitting_word_select) or highlighting_off:
                     highlighted_texts.append(text)
                     highlighted_titles.append(title)
                     continue