Update app.py

app.py

@@ -516,7 +516,17 @@ def preprocess_text(text):
     text = text.strip()
     return text
 
-   
+# 🗑️ Delete record - Saying goodbye to data (with just the id)
+def delete_record(container, id):
+    try:
+        container.delete_item(item=id, partition_key=id)
+        return True, f"Successfully deleted record with id: {id} 🗑️"
+    except exceptions.CosmosResourceNotFoundError:
+        return False, f"Record with id {id} not found. It may have been already 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: {traceback.format_exc()} 😱"   
  
                 
 # 🎭 Main function - "All the world's a stage, and all the code merely players" -Shakespeare, probably
@@ -790,10 +800,9 @@ def main():
                                                          height=300, 
                                                          key=f'doc_str_{idx}')
 
-
-                                    
                                     # 💾🤖 Save and AI operations
                                     col_save, col_ai, col_delete = st.columns(3)
+                                    
                                     with col_save:
                                         if st.button("💾 Save Changes", key=f'save_button_{idx}'):
                                             try:
@@ -808,20 +817,20 @@ def main():
                                                     st.error(message)
                                             except json.JSONDecodeError as e:
                                                 st.error(f"Invalid JSON: {str(e)} 🚫")
-
+                                    
                                     with col_ai:
                                         if st.button("🤖 Run AI", key=f'run_with_ai_button_{idx}'):
                                             search_glossary(json.dumps(editable_doc, indent=2))
-                                   
+                                    
                                     with col_delete:
                                         if st.button("🗑️ Delete Record", key=f'delete_button_{idx}'):
-                                            success, message = delete_record(container, doc.get('name', ''), doc.get('id', ''))
+                                            # 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)
-
                                                 
                         elif selected_view == 'Clone Document':
                             st.markdown("#### Clone a document:")
@@ -857,6 +866,8 @@ def main():
                                                 st.error("Failed to create new document")
                                         except Exception as e:
                                             st.error(f"Error creating document: {str(e)}")
+
+                                            
                         elif selected_view == 'New Record':
                             st.markdown("#### Create a new document:")
                             
@@ -912,58 +923,6 @@ def main():
                                     else:
                                         st.info("No documents to display. 🧐")
                                         
-                        elif selected_view == 'New Record2':
-                            st.markdown("#### Create a new document:")
-                            
-                            if st.button("🤖 Insert Auto-Generated Record"):
-                                auto_doc = {
-                                    "id": generate_unique_id(),
-                                    "name": f"Auto-generated Record {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
-                                    "content": "This is an auto-generated record.",
-                                    "timestamp": datetime.now().isoformat()
-                                }
-                                success, message = save_or_clone_to_cosmos_db(container, document=auto_doc)
-                                if success:
-                                    st.success(message)
-                                    st.rerun()
-                                else:
-                                    st.error(message)
-                            else:
-                                new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
-                                default_doc = {
-                                    "id": new_id,
-                                    "name": "New Document",
-                                    "content": "",
-                                    "timestamp": datetime.now().isoformat()
-                                }
-                                new_doc_str = st.text_area("Document Content (in JSON format)", 
-                                                          value=json.dumps(default_doc, indent=2), 
-                                                          height=300)
-                                
-                                if st.button("➕ Create New Document"):
-                                    try:
-                                        new_doc = json.loads(new_doc_str)
-                                        new_doc['id'] = new_id  # Ensure ID matches input field
-                                        success, message = insert_record(container, new_doc)
-                                        if success:
-                                            st.success(f"New document created with id: {new_doc['id']} 🎉")
-                                            st.session_state.selected_document_id = new_doc['id']
-                                            st.rerun()
-                                        else:
-                                            st.error(message)
-                                    except json.JSONDecodeError as e:
-                                        st.error(f"Invalid JSON: {str(e)} 🚫")
-                        
-                                    
-                                    st.subheader(f"📊 Container: {st.session_state.selected_container}")
-                                    if st.session_state.selected_container:
-                                        if documents_to_display:
-                                            Label = '# 📊 Data display - Data tells tales that words cannot'
-                                            st.markdown(Label)
-                                            df = pd.DataFrame(documents_to_display)
-                                            st.dataframe(df)
-                                        else:
-                                            st.info("No documents to display. 🧐")