Update app.py

app.py

@@ -174,38 +174,14 @@ def update_record(container, updated_record):
         return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
 # 🗑️ Delete record - Saying goodbye to data (it's not you, it's me)
-def delete_record(container, id):
+def delete_record(container, record):
     try:
-        # First query to verify document exists and get partition key
-        query = "SELECT c.id FROM c WHERE c.id = @id"
-        params = [{"name": "@id", "value": id}]
-        
-        # Execute query with cross partition enabled
-        items = list(container.query_items(
-            query=query,
-            parameters=params,
-            enable_cross_partition_query=True
-        ))
-        
-        if not items:
-            return False, f"Record with id {id} not found. 🕵️‍♂️"
-            
-        # Get partition key value from the query result
-        partition_key = items[0]['id']
-        
-        # Create partition key instance
-        pk = PartitionKey(partition_key)
-        
-        # Delete the document using correct partition key
-        container.delete_item(item=id, partition_key=pk)
-        return True, f"Successfully deleted record with id: {id} 🗑️"
-            
-    except exceptions.CosmosResourceNotFoundError:
-        return False, f"Record with id {id} not found. 🕵️‍♂️"
+        container.delete_item(item=record['id'], partition_key=record['id'])
+        return True, f"Record with id {record['id']} successfully deleted. 🗑️"
     except exceptions.CosmosHttpResponseError as e:
         return False, f"HTTP error occurred: {str(e)} 🚨"
     except Exception as e:
-        return False, f"An unexpected error occurred: {str(e)} 😱"
+        return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
         
 # 💾 Save to Cosmos DB - Preserving data for future generations (or just until the next update)
@@ -737,7 +713,7 @@ def main():
                                         st.rerun()
                         
                         elif selected_view == 'Show as Code Editor':
-                            Label = '# 💻 Code editor view - Code is poetry, bugs are typos'
+                            Label = '# 💻 Code editor view'
                             st.markdown(Label)
                             total_docs = len(documents)
                             doc = documents[st.session_state.current_index]
@@ -747,7 +723,6 @@ def main():
                                                   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'):
@@ -760,7 +735,6 @@ def main():
                                         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}'):
@@ -780,17 +754,16 @@ def main():
                                 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.")
+                                        success, message = delete_record(container, current_doc)
+                                        if success:
+                                            st.success(message)
                                             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)}")
+                                        else:
+                                            st.error(message)
+                                    except json.JSONDecodeError as e:
+                                        st.error(f"Invalid JSON: {str(e)} 🚫")
                     
 
 
@@ -847,14 +820,24 @@ def main():
                                             search_glossary(json.dumps(editable_doc, indent=2))
                                     
                                     with col_delete:
-                                        if st.button("🗑️ Delete Record", key=f'delete_button_{idx}'):
-                                            # Use only the 'id' to delete the record
-                                            success, message = delete_record(container, id=doc.get('id', ''))
-                                            if success:
-                                                st.success(message)
-                                                st.rerun()  # Refresh the page after deletion
-                                            else:
-                                                st.error(message)
+                                        #if st.button("🗑️ Delete Record", key=f'delete_button_{idx}'):
+
+
+                                        if st.button("🗑️ Delete", key=f'delete_button_{st.session_state.current_index}'):
+                                            try:
+                                                current_doc = json.loads(doc_str)
+                                                success, message = delete_record(container, current_doc)
+                                                if success:
+                                                    st.success(message)
+                                                    if st.session_state.current_index > 0:
+                                                        st.session_state.current_index -= 1
+                                                    st.rerun()
+                                                else:
+                                                    st.error(message)
+                                            except json.JSONDecodeError as e:
+                                                st.error(f"Invalid JSON: {str(e)} 🚫")
+
+                        
                                                 
                         elif selected_view == 'Clone Document':
                             st.markdown("#### Clone a document:")