Update app.py

app.py

@@ -755,21 +755,23 @@ def main():
                                         
                                         # Get client directly from container's client property
                                         client = container._client
+                                        partitionKey = current_doc['id']
+                                        client = st.session_state.client
                                         
                                         # Set up delete options
                                         options = {
                                             'enableCrossPartitionQuery': True,
-                                            'partitionKey': current_doc['id']
+                                            'partitionKey': partitionKey
                                         }
                                         
                                         # Construct document link
-                                        doc_link = f"dbs/{DATABASE_NAME}/colls/{CONTAINER_NAME}/docs/{current_doc['id']}"
+                                        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 {current_doc['id']} deleted successfully.")
+                                            st.success(f"Document {partitionKey} deleted successfully.")
                                             if st.session_state.current_index > 0:
                                                 st.session_state.current_index -= 1
                                             st.rerun()