Update app.py

app.py

@@ -2,9 +2,11 @@ from io import StringIO
 import streamlit as st
 from nltk import Tree
 import stanza
+from annotated_text import annotated_text
 
 st.set_page_config(layout='wide')
 st.title("Clause segmentation")
+
 uploaded_file = st.file_uploader("Upload your text file", type="txt")
 
 
@@ -38,12 +40,20 @@ def sentence_reader():
 
 
 def main():
+    colors = ["#a2d2ff", "#bde0fe", "#ffafcc", "#cdb4db", "#fefae0"]
+    key = 1
     for tree, text in zip(constituency_tree(), sentence_reader()):
-        expander = st.expander(f"{text}")
-        sub_phrases = tree_to_text(tree)
-        for sub_phrase in sub_phrases:
-            expander.write(sub_phrase)
+        with st.expander(f"Sentence{key}", expanded=True):
+            annotated_list = []
+            sub_phrases = tree_to_text(tree)
+            for sub_phrase, color in zip(sub_phrases, colors):
+                annotated_list.append((sub_phrase, color))
+            key += 1
+            st.write(f'{text} \n')
+            annotated_text(*annotated_list)
+            st.write("\n")
+
 
 if __name__ == "__main__":
     if uploaded_file is not None:
-        main()
+        main()