Update app.py

app.py

@@ -4,6 +4,7 @@ from annotated_text import annotated_text
 from io import StringIO
 from transformers import AutoTokenizer, AutoModelForTokenClassification
 from text_extractor import *
+from text_annotatator import *
 import os
 from streamlit_text_annotation import text_annotation
 
@@ -178,65 +179,24 @@ elif selected_menu == "Get Answers":
             result = pipeline_qa(question=question_text,  context=context)
             st.subheader('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"},
-    ]
-    }
+   
+    display_only_data = get_display_only_data()
+    editable_data = get_editable_data()
     
     st.subheader("Display Mode:")
     left, right = st.columns(2)
     with left:
         st.text("Vertical labels:")
-        text_annotation(data1)
+        text_annotation(display_only_data )
     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"},
-    ]
-    }
-    
+        text_annotation(display_only_data )
+
+
     st.subheader("Edit Mode:")
-    data = text_annotation(data2)
+    data = text_annotation(editable_data)
     if data:
-        "Returned data:", data
-    
+        "Returned data:", data