Update app.py

app.py

@@ -711,8 +711,7 @@ 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)
@@ -753,25 +752,30 @@ def main():
                                 if st.button("🗑️ Delete", key=f'delete_button_{st.session_state.current_index}'):
                                     try:
                                         current_doc = json.loads(doc_str)
+                                        
+                                        # Get client directly from container's client property
+                                        client = container._client
+                                        
+                                        # Set up delete options
                                         options = {
                                             'enableCrossPartitionQuery': True,
-                                            'partitionKey': '/Id'
+                                            'partitionKey': current_doc['id']
                                         }
-                                        response = container.delete_item(
-                                            item=current_doc['id'],
-                                            partition_key=current_doc['id'],
-                                            enable_cross_partition_query=True
-                                        )
+                                        
+                                        # Construct document link
+                                        doc_link = f"dbs/{DATABASE_NAME}/colls/{CONTAINER_NAME}/docs/{current_doc['id']}"
+                                        
+                                        # Delete using client directly
+                                        response = client.DeleteItem(doc_link, options)
+                                        
                                         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)}")
-                        
-
-
+                                        st.error(f"Error deleting document: {str(e)}")                        
+                                        
                                     
                         elif selected_view == 'Show as Edit, Save, Run AI':
                             Label = '# ✏️ Edit and save view - Edit with wisdom, save with precision'