fixed bug on changin source in comparisor

models/custom_parsers.py

@@ -11,6 +11,7 @@ class CustomStringOutputParser(BaseOutputParser[List[str]]):
     def parse(self, text: str) -> str:
         """Parse the output of an LLM call."""
         text = text.split("texter:")[0]
+        text = text.split("helper")[0]
         text = text.rstrip("\n")
         text = text.strip()
         return text

pages/comparisor.py

@@ -13,17 +13,24 @@ from app_config import ISSUES, SOURCES, source2label
 
 logger = get_logger(__name__)
 openai_api_key = os.environ['OPENAI_API_KEY'] 
-memories = {
-    'memoryA': {"issue": ISSUES[0], "source": SOURCES[0]},
-    'memoryB': {"issue": ISSUES[0], "source": SOURCES[1]},
-    'commonMemory': {"issue": ISSUES[0], "source": SOURCES[0]}
-}
-if 'db_client' not in st.session_state:
-    st.session_state["db_client"] = get_db_client()
+
+if "sent_messages" not in st.session_state:
+    st.session_state['sent_messages'] = 0
+logger.info(f'sent messages {st.session_state["sent_messages"]}')
+if "issue" not in st.session_state:
+    st.session_state['issue'] = ISSUES[0]
 if 'previous_sourceA' not in st.session_state:
     st.session_state['previous_sourceA'] = SOURCES[0]
 if 'previous_sourceB' not in st.session_state:
     st.session_state['previous_sourceB'] = SOURCES[0]
+
+memories = {
+    'memoryA': {"issue": st.session_state['issue'], "source": st.session_state['previous_sourceA']},
+    'memoryB': {"issue": st.session_state['issue'], "source": st.session_state['previous_sourceB']},
+    'commonMemory': {"issue": st.session_state['issue'], "source": SOURCES[0]}
+}
+if 'db_client' not in st.session_state:
+    st.session_state["db_client"] = get_db_client()
 if 'counselor_name' not in st.session_state:
     st.session_state["counselor_name"] = get_random_name()
 if 'texter_name' not in st.session_state:
@@ -91,7 +98,7 @@ def regenerateBoth():
                    st.session_state['commonMemory'].buffer_as_str, prompt, responseA, responseB)
 
 def bothGood():
-    if len(memoryA.buffer_as_messages) == 1:
+    if st.session_state['sent_messages'] == 0:
         pass
     else:
         i = random.choice([memoryA, memoryB])
@@ -137,18 +144,19 @@ with st.sidebar:
     # regenB = sbcol2.button("Regenerate B", on_click=regenerateB)
     clear = st.button("Clear History", on_click=clear_memory, kwargs={"memories":memories, "username":username, "language":language})
 
-memories = {
-    'memoryA': {"issue": issue, "source": sourceA},
-    'memoryB': {"issue": issue, "source": sourceB},
-    'commonMemory': {"issue": issue, "source": SOURCES[0]}
-}
 changed_source = any([
     st.session_state['previous_sourceA'] != sourceA,
-    st.session_state['previous_sourceB'] != sourceB
+    st.session_state['previous_sourceB'] != sourceB,
+    st.session_state['issue'] != issue
 ])
 if changed_source:
+    print("changed something")
     st.session_state["counselor_name"] = get_random_name()
     st.session_state["texter_name"] = get_random_name()
+    st.session_state['previous_sourceA'] = sourceA
+    st.session_state['previous_sourceB'] = sourceB
+    st.session_state['issue'] = issue
+    st.session_state['sent_messages'] = 0
 create_memory_add_initial_message(memories,
                                   issue, 
                                   language, 
@@ -185,6 +193,7 @@ def disable_chat():
         return True
 
 if prompt := st.chat_input(disabled=disable_chat()):
+    st.session_state['sent_messages'] += 1 
     if 'convo_id' not in st.session_state:
         push_convo2db(memories, username, language)