NavyDevilDoc commited on
Commit
bff8b24
·
verified ·
1 Parent(s): ec56e85

Update src/app.py

Browse files

fixed persistence issues with chat log

Files changed (1) hide show
  1. src/app.py +17 -2
src/app.py CHANGED
@@ -26,8 +26,18 @@ if "authentication_status" not in st.session_state or st.session_state["authenti
26
 
27
  with login_tab:
28
  is_logged_in = tracker.check_login()
29
- # FIX: Trigger User DB Download ONLY on fresh login
30
  if is_logged_in:
 
 
 
 
 
 
 
 
 
 
 
31
  tracker.download_user_db(st.session_state.username)
32
  st.rerun() # Refresh to show the app
33
 
@@ -510,7 +520,7 @@ with tab4:
510
  # 1. Save temp file
511
  temp_path = rag_engine.save_uploaded_file(uploaded_file)
512
 
513
- # 2. Process into USER'S specific DB (st.session_state.username)
514
  success, msg = rag_engine.process_and_add_document(
515
  temp_path,
516
  st.session_state.username,
@@ -518,6 +528,10 @@ with tab4:
518
  )
519
 
520
  if success:
 
 
 
 
521
  st.success(msg)
522
  st.rerun()
523
  else:
@@ -560,6 +574,7 @@ with tab4:
560
  with st.spinner("Deleting..."):
561
  success, msg = rag_engine.delete_document(st.session_state.username, doc['source'])
562
  if success:
 
563
  st.success(msg)
564
  st.rerun()
565
  else:
 
26
 
27
  with login_tab:
28
  is_logged_in = tracker.check_login()
 
29
  if is_logged_in:
30
+ # Check if a different user was previously logged in
31
+ if "last_user" in st.session_state and st.session_state.last_user != st.session_state.username:
32
+ # WIPE EVERYTHING
33
+ st.session_state.messages = []
34
+ st.session_state.email_draft = ""
35
+ st.session_state.user_openai_key = None
36
+
37
+ # Update the tracker
38
+ st.session_state.last_user = st.session_state.username
39
+
40
+ # Download DB and Refresh
41
  tracker.download_user_db(st.session_state.username)
42
  st.rerun() # Refresh to show the app
43
 
 
520
  # 1. Save temp file
521
  temp_path = rag_engine.save_uploaded_file(uploaded_file)
522
 
523
+ # 2. Process locally
524
  success, msg = rag_engine.process_and_add_document(
525
  temp_path,
526
  st.session_state.username,
 
528
  )
529
 
530
  if success:
531
+ # 3. FIX: SYNC TO CLOUD IMMEDIATELY
532
+ with st.spinner("Backing up to Cloud..."):
533
+ tracker.upload_user_db(st.session_state.username)
534
+
535
  st.success(msg)
536
  st.rerun()
537
  else:
 
574
  with st.spinner("Deleting..."):
575
  success, msg = rag_engine.delete_document(st.session_state.username, doc['source'])
576
  if success:
577
+ tracker.upload_user_db(st.session_state.username)
578
  st.success(msg)
579
  st.rerun()
580
  else: