clear_butt_add

.gitignore

@@ -26,6 +26,7 @@ share/python-wheels/
 .installed.cfg
 *.egg
 MANIFEST
+*.csv
 
 # PyInstaller
 #  Usually these files are written by a python script from a template

pages/convosim.py

@@ -13,6 +13,7 @@ from models.ta_models.bp_utils import bp_predict_message, bp_push2db
 
 logger = get_logger(__name__)
 temperature = 0.8
+reset_convo = False
 # username = "barb-chase" #"ivnban-ctl"
 st.set_page_config(page_title="Conversation Simulator")
 
@@ -20,7 +21,10 @@ if "sent_messages" not in st.session_state:
     st.session_state['sent_messages'] = 0
     if not are_models_alive():
         st.switch_page("pages/model_loader.py")
-
+# if "reset_convo" not in st.session_state:
+#     st.session_state['reset_convo'] = False
+if 'counselor_name' not in st.session_state:
+    st.session_state["counselor_name"] = ""
 if "total_messages" not in st.session_state:
     st.session_state['total_messages'] = 0
 if "issue" not in st.session_state:
@@ -49,30 +53,36 @@ if "scored" not in st.session_state:
 memories = {'memory':{"issue": st.session_state['issue'], "source": st.session_state['previous_source']}}
 
 with st.sidebar:
-    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)
+    if st.button("Reset Conversation", type="primary"):
+        reset_convo = True
+        logger.debug("Clear conversation manually")
+    username = st.text_input("Username", value=st.session_state["counselor_name"], max_chars=30)
+    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)
     issue = st.selectbox("Select a Scenario", ISSUES, index=ISSUES.index(st.session_state['issue']), format_func=issue2label,
-                            on_change=clear_memory, kwargs={"memories":memories, "username":username, "language":"English"}
+                            # kwargs={"memories":memories, "username":username, "language":"English"}
                         )
     supported_languages = ['en', "es"] if issue == "Anxiety" else ['en']
     language = st.selectbox("Select a Language", supported_languages, index=0,
                             format_func=lambda x: "English" if x=="en" else "Spanish",
-                            on_change=clear_memory, kwargs={"memories":memories, "username":username, "language":"English"}
+                            # kwargs={"memories":memories, "username":username, "language":"English"}
                         )
                             
     source = st.selectbox("Select a source Model A", SOURCES, index=0,
                           format_func=source2label, key="source"
-                        )
-
-changed_source = any([
-    st.session_state['previous_source'] != source,
-    st.session_state['issue'] != issue,
-    st.session_state['counselor_name'] != username,
-])
-if changed_source:
-    st.session_state["counselor_name"] = username
+                        )   
+# changed_source = any([
+#     st.session_state['previous_source'] != source,
+#     st.session_state['issue'] != issue,
+#     st.session_state['counselor_name'] != username,
+# ])
+logger.info("-"*10)
+logger.info(f"Reset convo is {reset_convo}")
+logger.info("-"*10)
+
+if reset_convo:
+    clear_memory(memories, username, language)
+    st.session_state['counselor_name'] = username
     st.session_state["texter_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
     logger.debug(f"texter name is {st.session_state['texter_name']}")
     st.session_state['previous_source'] = source
@@ -85,9 +95,10 @@ if changed_source:
 create_memory_add_initial_message(memories,
                                   issue, 
                                   language, 
-                                  changed_source=changed_source, 
+                                  changed_source=reset_convo, 
                                   counselor_name=st.session_state["counselor_name"], 
                                   texter_name=st.session_state["texter_name"])
+
 st.session_state['previous_source'] = source
 memoryA = st.session_state[list(memories.keys())[0]]
 # issue only without "." marker for model compatibility