Update app.py

app.py

@@ -50,14 +50,9 @@ if st.button("Annotate"):
         df = pd.DataFrame(annotated_text)
         st.dataframe(df)
 
+        if st.button("Save Modifications as JSON"):
+            save_annotations_as_json(df.to_dict(orient='records'), 'annotations.json')
+            st.success("Annotations saved as annotations.json")
+
     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}")
-