diff --git a/mind_palace/app.py b/mind_palace/app.py index 7631e08..5423d3b 100644 --- a/mind_palace/app.py +++ b/mind_palace/app.py @@ -57,6 +57,9 @@ def load_index(model): st.markdown("### Sources") for i, source_node in enumerate(response.source_nodes): with st.expander(f"[{i + 1}] {source_node.node.metadata['citation']}"): - st.write(f"relevancy score: {source_node.score}") + if source_node.metadata["paragraph_number"]: + st.write( + f"paragraph number: {source_node.metadata['paragraph_number']}" + ) st.write("original text:") st.write(source_node.node.get_text().split(":", 1)[1])