add options to turn features off

app.py

@@ -180,6 +180,10 @@ with st.sidebar:
     # 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")
+    # use only Qrels with relevance 2
+    st.header("Use only Qrels with relevance 2")
+    use_only_relevance_2 = st.checkbox("Use only Qrels with relevance 2", key="use_only_relevance_2")
+    
     
     ## 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:
@@ -193,6 +197,13 @@ with st.sidebar:
                 if qid in queries:
                     del queries[qid]
 
+        if use_only_relevance_2:
+            # remove all qrels that are not relevance 2
+            for qid, doc_rels in qrels.items():
+                qrels[qid] = {docid: rel for docid, rel in doc_rels.items() if rel == "2"}
+            # remove all queries that have no qrels
+            queries = {qid: text for qid, text in queries.items() if qid in qrels}
+
 
         data = []
         for key, value in qrels.items():