using last message timining difference and removing unnecessary logging

pages/convosim.py

@@ -1,4 +1,4 @@
-import os
+import time
 import streamlit as st
 from streamlit.logger import get_logger
 from langchain.schema.messages import HumanMessage
@@ -39,6 +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()
 
 # st.session_state["sel_phase"] = st.session_state["last_phase"]
 
@@ -99,7 +101,6 @@ def sent_request_llm(llm_chain, prompt):
     responses = custom_chain_predict(llm_chain, prompt, stopper)
     for response in responses:
         st.chat_message("assistant").write(response)
-    logger.info(f"After Change: {[x.response_metadata for x in memoryA.chat_memory.messages]}")
     transcript = memoryA.load_memory_variables({})[memoryA.memory_key]
     update_convo(st.session_state["db_client"], st.session_state["convo_id"], transcript)
 
@@ -123,6 +124,7 @@ def 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
+    st.session_state['last_message_ts'] = time.time()
     if 'convo_id' not in st.session_state:
         push_convo2db(memories, username, language)
     
@@ -162,7 +164,6 @@ with st.sidebar:
     def on_change_cpc():
         cpc_push2db(False)
         modify_last_human_message(memoryA, st.session_state['sel_phase'])
-        logger.info(f"After Change: {[x.response_metadata for x in memoryA.chat_memory.messages]}")
         st.session_state.changed_cpc = True
     def on_change_bp():
         bp_push2db()
@@ -196,5 +197,6 @@ 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 not are_models_alive():
-    st.switch_page("pages/model_loader.py")
+if time.time() - st.session_state['last_message_ts'] > 2400: # > 40 min
+    if not are_models_alive():
+        st.switch_page("pages/model_loader.py")