changes to comparisor on new role models GCT and SP

convosim.py

@@ -42,12 +42,12 @@ changed_source = st.session_state['previous_source'] != source
 if changed_source:
     st.session_state["counselor_name"] = get_random_name()
     st.session_state["texter_name"] = get_random_name()
-texter_name = create_memory_add_initial_message(memories, 
-                                                issue, 
-                                                language, 
-                                                changed_source=changed_source, 
-                                                counselor_name=st.session_state["counselor_name"], 
-                                                texter_name=st.session_state["texter_name"])
+create_memory_add_initial_message(memories,
+                                  issue, 
+                                  language, 
+                                  changed_source=changed_source, 
+                                  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]]
 llm_chain, stopper = get_chain(issue, language, source, memoryA, temperature, texter_name=st.session_state["texter_name"])

pages/comparisor.py

@@ -6,7 +6,9 @@ import streamlit as st
 from streamlit.logger import get_logger
 from langchain.schema.messages import HumanMessage
 from utils.mongo_utils import get_db_client, new_comparison, new_battle_result
-from utils.app_utils import create_memory_add_initial_message, clear_memory, get_chain, push_convo2db
+from utils.app_utils import create_memory_add_initial_message, get_random_name
+from utils.memory_utils import clear_memory, push_convo2db
+from utils.chain_utils import get_chain
 from app_config import ISSUES, SOURCES, source2label
 
 logger = get_logger(__name__)
@@ -21,7 +23,11 @@ if 'db_client' not in st.session_state:
 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[1]
+    st.session_state['previous_sourceB'] = SOURCES[0]
+if 'counselor_name' not in st.session_state:
+    st.session_state["counselor_name"] = get_random_name()
+if 'texter_name' not in st.session_state:
+    st.session_state["texter_name"] = get_random_name()
 
 def delete_last_message(memory):
     last_prompt = memory.chat_memory.messages[-2].content
@@ -104,11 +110,11 @@ with st.sidebar:
     issue = st.selectbox("Select an Issue", ISSUES, index=0,
                             on_change=clear_memory, kwargs={"memories":memories, "username":username, "language":"English"}
                         )
-    supported_languages = ['English', "Spanish"] if issue == "Anxiety" else ['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"}
-                        )
-    
+                        )    
     with st.expander("Model A"):
         temperatureA = st.slider("Temperature Model A", 0., 1., value=0.8, step=0.1)
         sourceA = st.selectbox("Select a source Model A", SOURCES, index=0,
@@ -116,7 +122,7 @@ with st.sidebar:
                             )
     with st.expander("Model B"):
         temperatureB = st.slider("Temperature Model B", 0., 1., value=0.8, step=0.1)
-        sourceB = st.selectbox("Select a source Model B", SOURCES, index=1,
+        sourceB = st.selectbox("Select a source Model B", SOURCES, index=0,
                                format_func=source2label
                             )
         
@@ -140,11 +146,19 @@ changed_source = any([
     st.session_state['previous_sourceA'] != sourceA,
     st.session_state['previous_sourceB'] != sourceB
 ])
-create_memory_add_initial_message(memories, username, language, changed_source=changed_source)
+if changed_source:
+    st.session_state["counselor_name"] = get_random_name()
+    st.session_state["texter_name"] = get_random_name()
+create_memory_add_initial_message(memories,
+                                  issue, 
+                                  language, 
+                                  changed_source=changed_source, 
+                                  counselor_name=st.session_state["counselor_name"], 
+                                  texter_name=st.session_state["texter_name"])
 memoryA = st.session_state[list(memories.keys())[0]]
 memoryB = st.session_state[list(memories.keys())[1]]
-llm_chainA, stopperA = get_chain(issue, language, sourceA, memoryA, temperatureA)
-llm_chainB, stopperB = get_chain(issue, language, sourceB, memoryB, temperatureB)
+llm_chainA, stopperA = get_chain(issue, language, sourceA, memoryA, temperatureA, texter_name=st.session_state["texter_name"])
+llm_chainB, stopperB = get_chain(issue, language, sourceB, memoryB, temperatureB, texter_name=st.session_state["texter_name"])
 
 st.title(f"💬 History") 
 for msg in st.session_state['commonMemory'].buffer_as_messages: