Update app.py

app.py

@@ -6,6 +6,7 @@ from collections import defaultdict
 import json
 import copy
 import plotly.express as px
+from find_splitting_words import find_dividing_words
 
 from dataset_loading import load_local_qrels, load_local_corpus, load_local_queries
 
@@ -168,6 +169,10 @@ 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(relevant_docs_text)
+
+            st.markdown(splitting_words)
+
             container.subheader(f"Relevant Documents ({len(list(qrels[str(inst_num)].keys()))})")
             current_checkboxes = []
             for (docid, title, text) in doc_texts: