Update app.py

app.py

@@ -712,6 +712,7 @@ def main():
                                         st.session_state.current_index += 1
                                         st.rerun()
                         
+                        
                         elif selected_view == 'Show as Code Editor':
                             Label = '# 💻 Code editor view'
                             st.markdown(Label)
@@ -735,6 +736,59 @@ def main():
                                         st.session_state.current_index += 1
                                         st.rerun()
                             
+                            col_save, col_delete = st.columns([1, 1])
+                            with col_save:
+                                if st.button("💾 Save Changes", key=f'save_button_{st.session_state.current_index}'):
+                                    try:
+                                        updated_doc = json.loads(doc_str)
+                                        response = container.upsert_item(body=updated_doc)
+                                        if response:
+                                            st.success(f"Document {updated_doc['id']} saved successfully.")
+                                            st.session_state.selected_document_id = updated_doc['id']
+                                            st.rerun()
+                                    except Exception as e:
+                                        st.error(f"Error saving document: {str(e)}")
+                            
+                            with col_delete:
+                                if st.button("🗑️ Delete", key=f'delete_button_{st.session_state.current_index}'):
+                                    try:
+                                        current_doc = json.loads(doc_str)
+                                        doc_id = current_doc['id']
+                                        response = container.delete_item(item=doc_id, partition_key=doc_id)
+                                        if response:
+                                            st.success(f"Document {doc_id} deleted successfully.")
+                                            if st.session_state.current_index > 0:
+                                                st.session_state.current_index -= 1
+                                            st.rerun()
+                                    except Exception as e:
+                                        st.error(f"Error deleting document: {str(e)}")
+                        
+                                        
+
+                        
+                        elif selected_view == 'Show as Code Editor 3':
+                            Label = '# 💻 Code editor view'
+                            st.markdown(Label)
+                            total_docs = len(documents)
+                            doc = documents[st.session_state.current_index]
+                            st.markdown(f"#### Document ID: {doc.get('id', '')}")
+                            doc_str = st.text_area("Edit Document", 
+                                                  value=json.dumps(doc, indent=2), 
+                                                  height=300, 
+                                                  key=f'code_editor_{st.session_state.current_index}')
+                            
+                            col_prev, col_next = st.columns([1, 1])
+                            with col_prev:
+                                if st.button("⬅️ Previous", key='prev_code'):
+                                    if st.session_state.current_index > 0:
+                                        st.session_state.current_index -= 1
+                                        st.rerun()
+                            with col_next:
+                                if st.button("➡️ Next", key='next_code'):
+                                    if st.session_state.current_index < total_docs - 1:
+                                        st.session_state.current_index += 1
+                                        st.rerun()
+                            
                             col_save, col_delete = st.columns([1, 1])
                             with col_save:
                                 if st.button("💾 Save Changes", key=f'save_button_{st.session_state.current_index}'):