progress bar instead of spinner

app_config.py

@@ -48,7 +48,7 @@ def source2label(source):
 def issue2label(issue):
     return seed2str.get(issue, "GCT")
 
-ENVIRON = "prod"
+ENVIRON = "dev"
 
 DB_SCHEMA = 'prod_db' if ENVIRON == 'prod' else 'test_db' 
 DB_CONVOS = 'conversations'

pages/training_adherence.py

@@ -8,17 +8,20 @@ from models.ta_models.config import QUESTION2PHASE, NAME2QUESTION, TA_OPTIONS
 
 st.set_page_config(page_title="Conversation Simulator - Scoring")
 
-if "memory" not in st.session_state:
-    st.switch_page("pages/convosim.py")
-
 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")
+
 memory = st.session_state['memory']
-@st.cache_data(show_spinner="Retrieving responses from the server ...")
+progress_text = "Scoring Conversation using AI models ..."
+
+@st.cache_data(show_spinner=False)
 def get_ta_responses():
+    my_bar = st.progress(0, text=progress_text)
     data = defaultdict(defaultdict)
-    for question in QUESTION2PHASE.keys():
+    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.",
@@ -29,6 +32,10 @@ def get_ta_responses():
         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)
+    my_bar.empty()
     return data
 
 with st.container():