Update app.py

app.py

@@ -4,6 +4,8 @@ from annotated_text import annotated_text
 from io import StringIO
 from transformers import AutoTokenizer, AutoModelForTokenClassification
 import os
+from streamlit_text_annotation import text_annotation
+
 os.environ['KMP_DUPLICATE_LIB_OK']='True'
 
 import plotly.express as px
@@ -215,7 +217,7 @@ Dr Jason Sanders"""
 
 with st.sidebar:
     selected_menu = option_menu("Select Option", 
-    ["Upload Document", "Extract Text", "Summarize Document", "Extract Entities","Detected Barriers","Get Answers"], 
+    ["Upload Document", "Extract Text", "Summarize Document", "Extract Entities","Detected Barriers","Get Answers","Annotation Tool"], 
         menu_icon="cast", default_index=0)
     
 
@@ -291,4 +293,66 @@ elif selected_menu == "Get Answers":
         with st.spinner("Finding Answer(s)..."):
             result = pipeline_qa(question=question_text,  context=context)
             st.subheader('Answer')
-            st.text(result['answer'])
+            st.text(result['answer'])
+elif selected_menu == "Annotation Tool":
+    data1 = {
+    "tokens": [
+    {"text": "He", "labels": ["Person"]},
+    {"text": "loves"},
+    {"text": "his"},
+    {"text": "dog", "labels": ["Animal", "Pet"]},
+    ],
+    "labels": [
+    {"text": "Person"},
+    {"text": "Action"},
+    {"text": "Animal"},
+    ]
+    }
+    
+    st.subheader("Display Mode:")
+    left, right = st.columns(2)
+    with left:
+        st.text("Vertical labels:")
+        text_annotation(data1)
+    with right:
+        st.text("Horizontal labels:")
+        data1["labelOrientation"] = "horizontal"
+        text_annotation(data1)
+
+
+    data2 = {
+    "allowEditing": True,
+    "tokens": [
+    {"text": "He", "labels": ["Pronoun", "Person"]},
+    {"text": "loves", "labels": ["Action"]},
+    {"text": "his"},
+    {"text": "dog", "labels": ["Animal"]},
+    ],
+    "labels": [
+    {"text": "Pronoun", "style": {
+    "color": "red",
+    "background-color": "white",
+    "font-size": "8px",
+    "border": "3px dashed red",
+    }},
+    {"text": "Verb", "style": {
+    "color": "green",
+    "background-color": "white",
+    "font-size": "8px",
+    "font-weight": "900",
+    }},
+    {"text": "Noun", "style": {
+    "color": "blue",
+    "background-color": "white",
+    "font-size": "8px",
+    }},
+    {"text": "Person"},
+    {"text": "Animal"},
+    ]
+    }
+    
+    st.subheader("Edit Mode:")
+    data = text_annotation(data2)
+    if data:
+        "Returned data:", data
+