Update app.py

app.py

@@ -202,6 +202,72 @@ def retrieve_ids_value(conn, name):
     rows = cursor.fetchall()
     return [bytes_to_int(row[0]) for row in rows]
 
+    
+To delete the ids_value entries from table_ids alongside their associated key (i.e., the document name) from table_names, you'll need to perform the following steps:
+
+Delete the associated entries from table_ids first.
+Then delete the entry from table_names.
+Here’s how you can structure your code to achieve this:
+
+Updated Functions
+Function to Delete Entries
+python
+Copy code
+def delete_document(conn, cursor, name):
+    # Begin a transaction
+    conn.execute('BEGIN')
+
+    try:
+        # First, delete related entries from table_ids
+        cursor.execute('DELETE FROM table_ids WHERE name = ?', (name,))
+        
+        # Then, delete the document name from table_names
+        cursor.execute('DELETE FROM table_names WHERE doc_name = ?', (name,))
+        
+        # Commit the transaction
+        conn.commit()
+        print(f"Deleted document '{name}' and its associated IDs.")
+    except sqlite3.Error as e:
+        # Rollback the transaction on error
+        conn.rollback()
+        print(f"An error occurred: {e}")
+Example Usage
+Here’s a complete example demonstrating how to use the delete_document function:
+
+python
+Copy code
+import sqlite3
+
+def int_to_bytes(value):
+    return value.to_bytes(4, byteorder='big')
+
+def bytes_to_int(b):
+    return int.from_bytes(b, byteorder='big')
+
+def insert_data(conn, cursor, name, ids_array):
+    cursor.execute('INSERT INTO table_names (doc_name) VALUES (?)', (name,))
+    for ids in ids_array:
+        cursor.execute('INSERT INTO table_ids (name, ids_value) VALUES (?, ?)', (name, int_to_bytes(ids)))
+    conn.commit()
+
+def retrieve_ids_value(conn, cursor, name):
+    cursor.execute('SELECT ids_value FROM table_ids WHERE name = ?', (name,))
+    rows = cursor.fetchall()
+    return [bytes_to_int(row[0]) for row in rows]
+
+def delete_document(conn, cursor, name):
+    conn.execute('BEGIN')
+
+    try:
+        cursor.execute('DELETE FROM table_ids WHERE name = ?', (name,))
+        cursor.execute('DELETE FROM table_names WHERE doc_name = ?', (name,))
+        
+        conn.commit()
+        print(f"Deleted document '{name}' and its associated IDs.")
+    except sqlite3.Error as e:
+        conn.rollback()
+        print(f"An error occurred: {e}")
+
 @st.cache_resource(show_spinner=False)
 def load_models_and_documents():
     container = st.empty()
@@ -514,7 +580,7 @@ if __name__ == '__main__':
             st.session_state.df = pd.DataFrame()
             os.mkdir(data_editor_path)
 
-    def on_change_data_editor(conn, client, collection_name):
+    def on_change_data_editor(conn, cursor, client, collection_name):
         print(f'Check : {st.session_state.key_data_editor}')
         
         if st.session_state.key_data_editor['deleted'] and (st.session_state.key_data_editor['deleted'][-1] in st.session_state.df):
@@ -522,12 +588,13 @@ if __name__ == '__main__':
             name = st.session_state.df.loc[st.session_state.key_data_editor['deleted'][-1], 'document']
             print(f'{name}')
             os.remove(os.path.join(embeddings_path, name + '_ids.npy'))
-            ids_values = retrieve_ids_value(conn, name)
+            ids_values = retrieve_ids_value(conn, cursor, name)
             
             client.delete(
                 collection_name=collection_name, 
                 points_selector=PointIdsList(points=ids_values)
             )
+            delete_document(conn, cursor, name)
             
     
     if menu_id == 'Documents':       
@@ -537,7 +604,7 @@ if __name__ == '__main__':
             use_container_width=True,
             hide_index=True,
             on_change=on_change_data_editor,
-            args=(conn, client, collection_name),
+            args=(conn, cursor, client, collection_name),
             key='key_data_editor',
             column_config={
                 'icon': st.column_config.ImageColumn(