Update app.py

app.py

@@ -34,22 +34,20 @@ def save_annotations_as_json(annotated_text, filename):
     with open(filename, 'w', encoding='utf-8') as json_file:
         json.dump(annotated_text, json_file, ensure_ascii=False, indent=4)
 
-# Define the layout
 st.title("Annotation Tool")
+
+# Add Materialize CSS
 st.markdown('<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/materialize/1.0.0/css/materialize.min.css">', unsafe_allow_html=True)
 
-# Input text area
 text = st.text_area("Text")
 if st.button("Annotate"):
     if text:
         segmented_text = tokenize_with_spacy(text)
         annotated_text = annotate_text(segmented_text)
-        
         st.subheader("Segmented Text:")
-        for token in segmented_text:
-            st.write(token)
-        
+        st.write(segmented_text)
         st.subheader("Annotated Text:")
+
         for token in annotated_text:
             st.write(f"Token: {token['token']}")
             st.write(f"POS: {token['pos']}")
@@ -64,14 +62,17 @@ if st.button("Annotate"):
             st.write(f"Case: {', '.join(token['case'])}")
             st.write(f"Gender: {', '.join(token['gender'])}")
             st.write("-----")
+
     else:
         st.warning("Please enter some text.")
 
 if st.button("Save Modifications as JSON"):
     if annotated_text:
+        # Generate a unique file name using a timestamp
         timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
         filename = f'annotations_{timestamp}.json'
         save_annotations_as_json(annotated_text, filename)
         st.success(f"Annotations saved as {filename}")
 
+# Add Materialize JavaScript
 st.markdown('<script src="https://cdnjs.cloudflare.com/ajax/libs/materialize/1.0.0/js/materialize.min.js"></script>', unsafe_allow_html=True)