Update backup12.app.py

backup12.app.py

@@ -684,13 +684,100 @@ 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'
                             st.markdown(Label)
                             total_docs = len(documents)
+                            
+                            if total_docs == 0:
+                                st.warning("No documents available.")
+                                return
+                                
+                            doc = documents[st.session_state.current_index]
+                            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}'):
+                                    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 json.JSONDecodeError:
+                                        st.error("Invalid JSON format. Please check your edits.")
+                                    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.get("id")
+                                        
+                                        if not doc_id:
+                                            st.error("Document ID not found.")
+                                            return
+                                        
+                                        # Confirm deletion
+                                        if 'confirm_delete' not in st.session_state:
+                                            st.session_state.confirm_delete = False
+                                        
+                                        if not st.session_state.confirm_delete:
+                                            if st.button("⚠️ Click to confirm deletion", key=f'confirm_delete_{st.session_state.current_index}'):
+                                                st.session_state.confirm_delete = True
+                                                st.rerun()
+                                        else:
+                                            try:
+                                                # Delete the document
+                                                container.delete_item(item=doc_id, partition_key=doc_id)
+                                                
+                                                # Update the session state
+                                                st.session_state.confirm_delete = False
+                                                
+                                                # Update the current index if necessary
+                                                if total_docs > 1:
+                                                    if st.session_state.current_index == total_docs - 1:
+                                                        st.session_state.current_index = max(0, total_docs - 2)
+                                                    documents.pop(st.session_state.current_index)
+                                                else:
+                                                    st.session_state.current_index = 0
+                                                    documents.clear()
+                                                
+                                                st.success(f"Document {doc_id} deleted successfully.")
+                                                st.rerun()
+                                                
+                                            except Exception as e:
+                                                st.error(f"Error deleting document: {str(e)}")
+                                                st.session_state.confirm_delete = False
+                                        
+                                    except json.JSONDecodeError:
+                                        st.error("Invalid JSON format. Please check the document.")
+                                    except Exception as e:
+                                        st.error(f"Error processing deletion: {str(e)}")
+                        
+                        elif selected_view == 'Show as Code Editor - Old':
+                            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", 
@@ -736,6 +823,8 @@ def main():
                                             st.rerun()
                                     except Exception as e:
                                         st.error(f"Error deleting document: {str(e)}")                                        
+
+                        
                         
                         
                         elif selected_view == 'Show as Run AI':
@@ -761,44 +850,44 @@ def main():
                                                          key=f'doc_str_{idx}')
                         
                                     # Save and AI operations columns
-                                    col_save, col_ai, col_delete = st.columns(3)
-                                    
-                                    with col_save:
-                                        if st.button("💾 Save Changes", key=f'save_runai_{idx}'):
-                                            try:
-                                                updated_doc = json.loads(doc_str)
-                                                # Reinsert ID and name from editable fields
-                                                updated_doc['id'] = editable_id
-                                                updated_doc['name'] = editable_name
-                                                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_ai:
-                                        if st.button("🤖 Run AI", key=f'run_with_ai_button_{idx}'):
-                                            # Your existing AI processing code here
-                                            values_with_space = []
-                                            def extract_values2(obj):
-                                                if isinstance(obj, dict):
-                                                    for k, v in obj.items():
-                                                        extract_values2(v)
-                                                elif isinstance(obj, list):
-                                                    for item in obj:
-                                                        extract_values2(item)
-                                                elif isinstance(obj, str):
-                                                    if ' ' in obj:
-                                                        values_with_space.append(obj)
-                                            
-                                            extract_values2(doc)
-                                            for term in values_with_space:
-                                                display_glossary_entity(term)
-                                                search_glossary(term)
 
-                        
+                                    if st.button("🤖 Run AI", key=f'run_with_ai_button_{idx}'):
+                                        # Your existing AI processing code here
+                                        values_with_space = []
+                                        def extract_values2(obj):
+                                            if isinstance(obj, dict):
+                                                for k, v in obj.items():
+                                                    extract_values2(v)
+                                            elif isinstance(obj, list):
+                                                for item in obj:
+                                                    extract_values2(item)
+                                            elif isinstance(obj, str):
+                                                if ' ' in obj:
+                                                    values_with_space.append(obj)
+                                        
+                                        extract_values2(doc)
+                                        for term in values_with_space:
+                                            display_glossary_entity(term)
+                                            search_glossary(term)
+
+                                    if st.button("💾 Save Changes", key=f'save_runai_{idx}'):
+                                        try:
+                                            updated_doc = json.loads(doc_str)
+                                            # Reinsert ID and name from editable fields
+                                            updated_doc['id'] = editable_id
+                                            updated_doc['name'] = editable_name
+                                            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)}")
+
+
+
+
+                                                
                         elif selected_view == 'Clone Document':
                             st.markdown("#### 📄 Clone Document (Save As)")