Update app.py

app.py

@@ -48,18 +48,10 @@ if st.button("Annotate"):
 
         # Create a DataFrame from the annotated text
         df = pd.DataFrame(annotated_text)
-
-        # Create an editable table
-        editable_df = df.copy()
-        for col in df.columns:
-            for idx, value in enumerate(editable_df[col]):
-                new_value = st.text_input(f'{col}_{idx}', value=value)
-                editable_df.at[idx, col] = new_value
-
-        st.table(editable_df)
+        st.dataframe(df)
 
         if st.button("Save Modifications as JSON"):
-            save_annotations_as_json(editable_df.to_dict(orient='records'), 'annotations.json')
+            save_annotations_as_json(df.to_dict(orient='records'), 'annotations.json')
             st.success("Annotations saved as annotations.json")
 
     else: