Update app.py

app.py

@@ -735,78 +735,19 @@ def main():
                                     if st.session_state.current_index < total_docs - 1:
                                         st.session_state.current_index += 1
                                         st.rerun()
-                    
-                    elif selected_view == 'Show as Code Editor':
-                        Label = '# 💻 Code editor view - Code is poetry, bugs are typos'
-                        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}')
-                        
-                        # ⬅️➡️ Navigation controls
-                        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()
-                        
-                        # 💾 🗑️ Save and Delete controls
-                        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)
-                                    success, message = update_record(container, updated_doc)
-                                    if success:
-                                        st.success(f"Document {updated_doc['id']} saved successfully.")
-                                        st.session_state.selected_document_id = updated_doc['id']
-                                        st.rerun()
-                                    else:
-                                        st.error(message)
-                                except json.JSONDecodeError as e:
-                                    st.error(f"Invalid JSON: {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)
-                                    response = container.delete_item(
-                                        item=current_doc['id'], 
-                                        partition_key=current_doc['id']
-                                    )
-                                    if response:
-                                        st.success(f"Document {current_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 Editor2':
+                        elif selected_view == 'Show as Code Editor':
                             Label = '# 💻 Code editor view - Code is poetry, bugs are typos'
                             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}')
-
-
+                                                  value=json.dumps(doc, indent=2), 
+                                                  height=300, 
+                                                  key=f'code_editor_{st.session_state.current_index}')
                             
-                            # ⬅️➡️ Navigation and save controls
+                            # ⬅️➡️ Navigation controls
                             col_prev, col_next = st.columns([1, 1])
                             with col_prev:
                                 if st.button("⬅️ Previous", key='prev_code'):
@@ -818,23 +759,39 @@ def main():
                                     if st.session_state.current_index < total_docs - 1:
                                         st.session_state.current_index += 1
                                         st.rerun()
-
-
                             
-                            # 💾 Save functionality - "Save early, save often"
-                            if st.button("💾 Save Changes", key=f'save_button_{st.session_state.current_index}'):
-                                try:
-                                    updated_doc = json.loads(doc_str)
-                                    success, message = update_record(container, updated_doc)
-                                    if success:
-                                        st.success(f"Document {updated_doc['id']} saved successfully.")
-                                        st.session_state.selected_document_id = updated_doc['id']
-                                        st.rerun()
-                                    else:
-                                        st.error(message)
-                                except json.JSONDecodeError as e:
-                                    st.error(f"Invalid JSON: {str(e)} 🚫")
-            
+                            # 💾 🗑️ Save and Delete controls
+                            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)
+                                        success, message = update_record(container, updated_doc)
+                                        if success:
+                                            st.success(f"Document {updated_doc['id']} saved successfully.")
+                                            st.session_state.selected_document_id = updated_doc['id']
+                                            st.rerun()
+                                        else:
+                                            st.error(message)
+                                    except json.JSONDecodeError as e:
+                                        st.error(f"Invalid JSON: {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)
+                                        response = container.delete_item(
+                                            item=current_doc['id'], 
+                                            partition_key=current_doc['id']
+                                        )
+                                        if response:
+                                            st.success(f"Document {current_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)}")
+