devve1 commited on
Commit
39636c1
1 Parent(s): 71d12b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -406,10 +406,9 @@ def load_models_and_documents():
406
  collection_name=collection_name,
407
  optimizer_config=OptimizersConfigDiff(indexing_threshold=20000)
408
  )
409
- hop = status.update(
410
  label="Processing Complete!", state="complete", expanded=False
411
  )
412
- print(f'HOP : {hop}')
413
 
414
  time.sleep(5)
415
  container.empty()
@@ -529,9 +528,9 @@ if __name__ == '__main__':
529
  def on_change_data_editor(conn, cursor, client, collection_name):
530
  print(f'Check : {st.session_state.key_data_editor}')
531
 
532
- if st.session_state.key_data_editor['deleted'] and (st.session_state.key_data_editor['deleted'][-1] in st.session_state.df):
533
  print('HELLO')
534
- name = st.session_state.df.loc[st.session_state.key_data_editor['deleted'][-1], 'document']
535
  print(f'{name}')
536
  os.remove(os.path.join(embeddings_path, name + '_ids.npy'))
537
  ids_values = retrieve_ids_value(conn, cursor, name)
 
406
  collection_name=collection_name,
407
  optimizer_config=OptimizersConfigDiff(indexing_threshold=20000)
408
  )
409
+ status.update(
410
  label="Processing Complete!", state="complete", expanded=False
411
  )
 
412
 
413
  time.sleep(5)
414
  container.empty()
 
528
  def on_change_data_editor(conn, cursor, client, collection_name):
529
  print(f'Check : {st.session_state.key_data_editor}')
530
 
531
+ if st.session_state.key_data_editor['deleted_rows'] and (st.session_state.key_data_editor['deleted_rows'][-1] in st.session_state.df):
532
  print('HELLO')
533
+ name = st.session_state.df.loc[st.session_state.key_data_editor['deleted_rows'][-1], 'document']
534
  print(f'{name}')
535
  os.remove(os.path.join(embeddings_path, name + '_ids.npy'))
536
  ids_values = retrieve_ids_value(conn, cursor, name)