fixint TA page reloading after some time, deleting unnecessary time sleep

pages/training_adherence.py

@@ -1,3 +1,4 @@
+import time
 import streamlit as st
 import numpy as np
 from collections import defaultdict
@@ -8,8 +9,9 @@ from models.ta_models.config import QUESTION2PHASE, NAME2QUESTION, TA_OPTIONS
 
 st.set_page_config(page_title="Conversation Simulator - Scoring")
 
-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")
 
 if "memory" not in st.session_state:
     st.switch_page("pages/convosim.py")
@@ -22,19 +24,12 @@ def get_ta_responses():
     my_bar = st.progress(0, text=progress_text)
     data = defaultdict(defaultdict)
     for i, question in enumerate(QUESTION2PHASE.keys()):
-        # responses = ["Yes, The helper showed some respect.",
-        #             "Yes. The helper is good! No doubt",
-        #             "N/A, Texter disengaged.",
-        #             "No. While texter is trying is lacking.",
-        #             "No \n\n This is an explanation."]
-        # full_response = np.random.choice(responses)
         full_convo, prompt, full_response = TA_predict_convo(memory, question, make_explanation=True, conversation_id=st.session_state['convo_id'])
         response, explanation = post_process_response(full_response)
         data[question]["response"] = response
         data[question]["explanation"] = explanation
         my_bar.progress((i+1) / len(QUESTION2PHASE.keys()), text = progress_text)
-        import time
-        time.sleep(2)
+        # time.sleep(2)
     my_bar.empty()
     return data