Update app.py

app.py

@@ -93,9 +93,9 @@ def query_hybrid_search(query: str, client: QdrantClient, collection_name: str,
                 with_payload=True,
                 query=FusionQuery(fusion=Fusion.DSBF),
                 limit=3
-                #filter=Filter(must_not=[
-                #    HasIdCondition(has_id=[])
-                #])
+                filter=Filter(must_not=[
+                    HasIdCondition(has_id=st.session_state.filter_ids)
+                ])
             ),
             QueryRequest(
                 prefetch=Prefetch(query=dense_embeddings, using="text-dense", limit=10),
@@ -103,9 +103,9 @@ def query_hybrid_search(query: str, client: QdrantClient, collection_name: str,
                 with_payload=True,
                 query=FusionQuery(fusion=Fusion.DSBF),
                 limit=3
-                #filter=Filter(must_not=[
-                #    HasIdCondition(has_id=[])
-                #])
+                filter=Filter(must_not=[
+                    HasIdCondition(has_id=st.session_state.filter_ids)
+                ])
             )
         ]
     )
@@ -541,6 +541,9 @@ if __name__ == '__main__':
             st.session_state.df = pd.DataFrame()
             os.mkdir(data_editor_path)
 
+    if 'filter_ids' not in st.session_state:
+        st.session_state.filter_ids = []
+
     def on_change_data_editor(conn, cursor, client, collection_name):
         print(f'Check : {st.session_state.key_data_editor}')
         
@@ -562,13 +565,18 @@ if __name__ == '__main__':
                 st.session_state.df.drop(deleted_idx)
                 
             st.toast('Deletion Completed !', icon='🎉')
-
-        #if st.session_state.key_data_editor['edited_rows']:
-        #    for i in st.session_state.key_data_editor['edited_rows']:
-        #        if i['toggle'] == False:
-        #            st.session_state.df['toggle'] = False
-                    
+        elif st.session_state.key_data_editor['edited_rows']:
             
+            for i in st.session_state.key_data_editor['edited_rows']:
+                key = next(iter(i))
+                toggle = i[key]['toggle']
+                st.session_state.df.loc[key, 'toggle'] = toggle
+                retrieved_ids = retrieve_ids_value(conn, cursor, st.session_state.df.loc[key, 'document'])
+
+                if not toggle:
+                    st.session_state.filter_ids.extend(retrieved_ids)
+                else:
+                    st.session_state.filter_ids = [j for j in st.session_state.filter_ids if j not in retrieved_ids]
     
     if st.session_state.menu_id == 'Documents':       
         st.session_state.df = st.data_editor(