push2dbOnScore

pages/convosim.py

@@ -41,13 +41,15 @@ if "changed_bp" not in st.session_state:
     st.session_state["changed_bp"] = False
 if "last_message_ts" not in st.session_state:
     st.session_state["last_message_ts"] = time.time()
+if "scored" not in st.session_state:
+    st.session_state['scored'] = False
 
 # st.session_state["sel_phase"] = st.session_state["last_phase"]
 
 memories = {'memory':{"issue": st.session_state['issue'], "source": st.session_state['previous_source']}}
 
 with st.sidebar:
-    username = st.text_input("Username", value='Dani', max_chars=30)
+    username = st.text_input("Username", value='', max_chars=30)
     if 'counselor_name' not in st.session_state:
         st.session_state["counselor_name"] = username #get_random_name(names_df=DEFAULT_NAMES_DF)
     # temperature = st.slider("Temperature", 0., 1., value=0.8, step=0.1)
@@ -77,6 +79,8 @@ if changed_source:
     st.session_state['issue'] = issue
     st.session_state['sent_messages'] = 0
     st.session_state['total_messages'] = 0
+    st.session_state["last_message_ts"] = time.time()
+    st.session_state['scored'] = False
 
 create_memory_add_initial_message(memories,
                                   issue, 
@@ -123,23 +127,30 @@ def sent_request_llm(llm_chain, prompt):
 #         sent_request_llm(llm_chain, prompt)
 #         st.rerun()
 
-if prompt := st.chat_input(disabled=st.session_state['total_messages'] > MAX_MSG_COUNT - 4): #account for next interaction
+
+def push_latest_confirmation_db():
+    if st.session_state.changed_cpc:
+        if not st.session_state['scored']:
+            st.session_state["sel_phase"] = None
+        st.session_state.changed_cpc = False
+    else:
+        cpc_push2db(True) 
+    
+    if st.session_state.changed_bp:
+        if not st.session_state['scored']:
+            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']})
+
+disable_chat = (st.session_state['total_messages'] > MAX_MSG_COUNT - 4) and not st.session_state['scored']
+if prompt := st.chat_input(disabled=disable_chat): #account for next interaction
     st.session_state['last_message_ts'] = time.time()
     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']})
+        push_latest_confirmation_db()
 
 
     st.session_state['context'] = llm_chain.memory.load_memory_variables({})[llm_chain.memory.memory_key]
@@ -189,6 +200,8 @@ with st.sidebar:
         )
 
         if st.button("Score Conversation"):
+            st.session_state['scored'] = True
+            push_latest_confirmation_db()
             st.switch_page("pages/training_adherence.py")
 
 st.session_state['total_messages'] = len(memoryA.chat_memory.messages)