Update app.py

app.py

@@ -752,31 +752,20 @@ 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
-                                        partitionKey = current_doc['id']
-                                        client = st.session_state.client
-                                        
-                                        # Set up delete options
-                                        options = {
-                                            'enableCrossPartitionQuery': True,
-                                            'partitionKey': partitionKey
-                                        }
-                                        
-                                        # Construct document link
-                                        doc_link = f"dbs/{DATABASE_NAME}/colls/{CONTAINER_NAME}/docs/{partitionKey}"
-                                        
-                                        # Delete using client directly
-                                        response = client.DeleteItem(doc_link, options)
-                                        
-                                        if response:
-                                            st.success(f"Document {partitionKey} deleted successfully.")
+                                        # Direct deletion using container method with id and partition key
+                                        delete = container.delete_item(current_doc["id"], current_doc["id"])
+                                        if delete:
+                                            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'