Update pages/convosim.py

pages/convosim.py

@@ -39,8 +39,8 @@ if "changed_cpc" not in st.session_state:
     st.session_state["changed_cpc"] = False
 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 "first_message_ts" not in st.session_state:
+    st.session_state["first_message_ts"] = time.time()
 if "scored" not in st.session_state:
     st.session_state['scored'] = False
 
@@ -79,7 +79,7 @@ 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["first_message_ts"] = time.time()
     st.session_state['scored'] = False
 
 create_memory_add_initial_message(memories,
@@ -145,7 +145,7 @@ def push_latest_confirmation_db():
 
 disable_chat = (st.session_state['total_messages'] > MAX_MSG_COUNT - 2) 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()
+    # st.session_state['last_message_ts'] = time.time()
     if 'convo_id' not in st.session_state:
         push_convo2db(memories, username, language)
     
@@ -211,6 +211,8 @@ if st.session_state['total_messages'] >= MAX_MSG_COUNT:
 elif st.session_state['total_messages'] >= WARN_MSG_COUT:
     st.toast(f"The conversation will end at {MAX_MSG_COUNT} Total Messages ", icon=":material/warning:")
 
-if time.time() - st.session_state['last_message_ts'] > 2400: # > 40 min
+if time.time() - st.session_state['first_message_ts'] > 1200: # Check every 20 min for model aliveness
     if not are_models_alive():
         st.switch_page("pages/model_loader.py")
+    else:
+        st.session_state["first_message_ts"] = time.time()