Spaces:
Sleeping
Sleeping
Update src/app.py
Browse filesfixed persistence issues with chat log
- 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
|
| 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:
|