fixed bug

app.py

@@ -218,6 +218,7 @@ with st.container():
                 keys = get_all_keys(schema) 
             comments = handle_comments(comments, keys)
             save_comments(document_id, comments)
+        comments = get_comments(document_id)
         if schema and st.button('Process') :
             if st.session_state.doument_slected_to_chat.split("_")[-1]=="pdf": 
                 data = {"token": st.session_state.token,