TA-fixesDB (#2)

- ID exposure + DB fixes (4eaaa739f5fec0ec0ebac6a2b5ecf2714319c482)

models/ta_models/config.py

@@ -19,6 +19,8 @@ def cpc_label2str(phase):
 def phase2int(phase):
     return int(phase.split("_")[0])
 
+BP_LBL_OPTS = ["None", "Advice", "Personal Info", "Advice & Personal Info"]
+
 BP_THRESHOLD = 0.7
 BP_LAB2STR = {
     "is_advice": "Advice",

pages/convosim.py

@@ -7,7 +7,7 @@ from utils.app_utils import create_memory_add_initial_message, get_random_name,
 from utils.memory_utils import clear_memory, push_convo2db
 from utils.chain_utils import get_chain, custom_chain_predict
 from app_config import ISSUES, SOURCES, source2label, issue2label, MAX_MSG_COUNT, WARN_MSG_COUT
-from models.ta_models.config import CPC_LBL_OPTS, cpc_label2str, BP_LAB2STR
+from models.ta_models.config import CPC_LBL_OPTS, cpc_label2str, BP_LAB2STR, BP_LBL_OPTS
 from models.ta_models.cpc_utils import cpc_push2db
 from models.ta_models.bp_utils import bp_predict_message, bp_push2db
 
@@ -75,6 +75,7 @@ if changed_source:
     st.session_state['issue'] = issue
     st.session_state['sent_messages'] = 0
     st.session_state['total_messages'] = 0
+
 create_memory_add_initial_message(memories,
                                   issue, 
                                   language, 
@@ -121,17 +122,24 @@ def sent_request_llm(llm_chain, prompt):
 if prompt := st.chat_input(disabled=st.session_state['total_messages'] > MAX_MSG_COUNT - 4): #account for next interaction
     if 'convo_id' not in st.session_state:
         push_convo2db(memories, username, language)
+    
+    if st.session_state["sent_messages"] > 0:
+        if st.session_state.changed_cpc:
+            st.session_state["sel_phase"] = None
+            st.session_state.changed_cpc = False
+        else:
+            cpc_push2db(True) 
+        
+        if st.session_state.changed_bp:
+            st.session_state["sel_bp"] = None
+            st.session_state.changed_bp = False
+        else:
+            bp_push2db({x['label']:x['score'] for x in st.session_state['bp_prediction']})
+
+
     st.session_state['context'] = llm_chain.memory.load_memory_variables({})[llm_chain.memory.memory_key]
     st.session_state['last_message'] = prompt
-    if (not st.session_state.changed_cpc) and st.session_state["sent_messages"] > 0:
-        cpc_push2db(True)
-    else: st.session_state.changed_cpc = False
-    if (not st.session_state.changed_bp) and st.session_state["sent_messages"] > 0:
-        bp_push2db({x['label']:x['score'] for x in st.session_state['bp_prediction']})
-    else: st.session_state.changed_bp = False
-
-    context = llm_chain.memory.load_memory_variables({})[llm_chain.memory.memory_key]
-    st.session_state['bp_prediction'] = bp_predict_message(context, prompt)
+    st.session_state['bp_prediction'] = bp_predict_message(st.session_state['context'], prompt)
     if any([x['score'] for x in st.session_state['bp_prediction']]):
         for bp in st.session_state['bp_prediction']:
             if bp["score"]:
@@ -142,6 +150,8 @@ if prompt := st.chat_input(disabled=st.session_state['total_messages'] > MAX_MSG
     #     sent_request_llm(llm_chain, prompt)
 
 with st.sidebar:
+    if "convo_id" in st.session_state:
+        st.write(f"Conversation ID is `{st.session_state['convo_id']}`")
     st.divider()
     st.markdown(f"### Total Sent Messages: :red[**{st.session_state['sent_messages']}**]")
     st.markdown(f"### Total Messages: :red[**{st.session_state['total_messages']}**]")
@@ -158,7 +168,7 @@ with st.sidebar:
             cpc_label2str(st.session_state['last_phase'])
         }**]. If not please select from the following options""", 
 
-            CPC_LBL_OPTS, index=None,format_func=cpc_label2str, on_change=on_change_cpc,
+            CPC_LBL_OPTS, index=None, format_func=cpc_label2str, on_change=on_change_cpc,
             key="sel_phase",
         )
 
@@ -168,7 +178,7 @@ with st.sidebar:
         }**].""" if len(BPs) > 0 else "Last Human Message was NOT considered Bad Practice."
         _ = st.selectbox(selecttitle + " If not please select from the following options""", 
 
-            ["None", "Advice", "Personal Info", "Advice & Personal Info"], index=None, on_change=on_change_bp,
+            BP_LBL_OPTS, index=None, format_func=lambda x: x, on_change=on_change_bp,
             key="sel_bp"
         )
 

utils/mongo_utils.py

@@ -114,7 +114,7 @@ def new_cpc_comparison(client, convo_id, model, context, last_message, ytrue, yp
     db = client[DB_SCHEMA]
     cpc_comps = db[DB_CPC]
     comarison_id = cpc_comps.insert_one(comp).inserted_id
-    logger.debug(f"DBUTILS: new error id is {comarison_id}")
+    logger.debug(f"DBUTILS: new CPC id is {comarison_id}")
 
 def new_bp_comparison(client, convo_id, model, context, last_message, ytrue, ypred):
     # context = memory.load_memory_variables({})[memory.memory_key]