added names variety and hide manual comparison

convosim.py

@@ -3,7 +3,7 @@ import streamlit as st
 from streamlit.logger import get_logger
 from langchain.schema.messages import HumanMessage
 from utils.mongo_utils import get_db_client
-from utils.app_utils import create_memory_add_initial_message, get_random_name
+from utils.app_utils import create_memory_add_initial_message, get_random_name,  DEFAULT_NAMES_DF
 from utils.memory_utils import clear_memory, push_convo2db
 from utils.chain_utils import get_chain
 from app_config import ISSUES, SOURCES, source2label
@@ -17,9 +17,10 @@ if 'previous_source' not in st.session_state:
 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()
+    st.session_state["counselor_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
 if 'texter_name' not in st.session_state:
-    st.session_state["texter_name"] = get_random_name()
+    st.session_state["texter_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
+    logger.info(f"texter name is {st.session_state['texter_name']}")
 
 with st.sidebar:
     username = st.text_input("Username", value='ivnban-ctl', max_chars=30)
@@ -38,10 +39,16 @@ with st.sidebar:
                         )
 
 memories = {'memory':{"issue":issue, "source":source}}
-changed_source = st.session_state['previous_source'] != source
+changed_source = any([
+    st.session_state['previous_source'] != source,
+    st.session_state['issue'] != issue
+])
 if changed_source:
-    st.session_state["counselor_name"] = get_random_name()
-    st.session_state["texter_name"] = get_random_name()
+    st.session_state["counselor_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
+    st.session_state["texter_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
+    st.session_state['previous_source'] = source
+    st.session_state['issue'] = issue
+    st.session_state['sent_messages'] = 0
 create_memory_add_initial_message(memories,
                                   issue, 
                                   language, 

pages/comparisor.py

@@ -6,7 +6,7 @@ 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, get_random_name
+from utils.app_utils import create_memory_add_initial_message, get_random_name, DEFAULT_NAMES_DF
 from utils.memory_utils import clear_memory, push_convo2db
 from utils.chain_utils import get_chain
 from app_config import ISSUES, SOURCES, source2label
@@ -32,9 +32,9 @@ memories = {
 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()
+    st.session_state["counselor_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
 if 'texter_name' not in st.session_state:
-    st.session_state["texter_name"] = get_random_name()
+    st.session_state["texter_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
 
 def delete_last_message(memory):
     last_prompt = memory.chat_memory.messages[-2].content
@@ -151,8 +151,8 @@ changed_source = any([
 ])
 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["counselor_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
+    st.session_state["texter_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
     st.session_state['previous_sourceA'] = sourceA
     st.session_state['previous_sourceB'] = sourceB
     st.session_state['issue'] = issue

utils/app_utils.py

@@ -13,6 +13,7 @@ logger = get_logger(__name__)
 
 # TODO: Include more variable and representative names
 DEFAULT_NAMES = ["Olivia", "Kit", "Abby", "Tom", "Carolyne", "Jessiny"]
+DEFAULT_NAMES_DF = pd.read_csv("./utils/names.csv")
 
 def get_random_name(gender="Neutral", ethnical_group="Neutral", names_df=None):
         if names_df is None:

utils/names.csv

@@ -0,0 +1,176 @@
+name,gender,ethnical_group
+Jacob,Male,Neutral
+Ethan,Male,Neutral
+Matthew,Male,Neutral
+David,Male,Neutral
+Liam,Male,Neutral
+Noah,Male,Neutral
+Michael,Male,Neutral
+Aiden,Male,Neutral
+Daniel,Male,Neutral
+Ryan,Male,Neutral
+Mason,Male,Neutral
+Lucas,Male,Neutral
+Joseph,Male,Neutral
+James,Male,Neutral
+Alexander,Male,Neutral
+Anthony,Male,Neutral
+Sebastian,Male,Neutral
+Jayden,Male,Neutral
+Christopher,Male,Neutral
+Benjamin,Male,Neutral
+Emma,Female,Neutral
+Olivia,Female,Neutral
+Emily,Female,Neutral
+Mia,Female,Neutral
+Sophia,Female,Neutral
+Leah,Female,Neutral
+Isabella,Female,Neutral
+Ava,Female,Neutral
+Sarah,Female,Neutral
+Chloe,Female,Neutral
+Sofia,Female,Neutral
+Madison,Female,Neutral
+Victoria,Female,Neutral
+Esther,Female,Neutral
+Abigail,Female,Neutral
+Rachel,Female,Neutral
+Maya,Female,Neutral
+Ashley,Female,Neutral
+Ella,Female,Neutral
+Grace,Female,Neutral
+Zoe,Female,Neutral
+Rowan,Neutral,Neutral
+Yael,Neutral,Neutral
+Eden,Neutral,Neutral
+Quinn,Neutral,Neutral
+Charlie,Neutral,Neutral
+Milan,Neutral,Neutral
+Blake,Neutral,Neutral
+Tenzin,Neutral,Neutral
+Peyton,Neutral,Neutral
+Alexis,Neutral,Neutral
+Ariel,Neutral,Neutral
+Riley,Neutral,Neutral
+Avery,Neutral,Neutral
+Angel,Neutral,Neutral
+Malachi,Male,African American
+Nasir,Male,African American
+Mamdou,Male,African American
+Chance,Male,African American
+Zaire,Male,African American
+Mekhi,Male,African American
+Sincere,Male,African American
+Omari,Male,African American
+Amadou,Male,African American
+Ibrahima,Male,African American
+Khalil,Male,African American
+Moussa,Male,African American
+Kamari,Male,African American
+Alpha,Male,African American
+Major,Male,African American
+Abdoulaye,Male,African American
+Aboul,Male,African American
+Amare,Male,African American
+Ousmane,Male,African American
+Darius,Male,African American
+Jose,Male,Hispanic
+Carlos,Male,Hispanic
+Luis,Male,Hispanic
+Miguel,Male,Hispanic
+Juan,Male,Hispanic
+Jesus,Male,Hispanic
+Erick,Male,Hispanic
+Alejandro,Male,Hispanic
+Diego,Male,Hispanic
+Gael,Male,Hispanic
+Santago,Male,Hispanic
+Iker,Male,Hispanic
+Cristian,Male,Hispanic
+Jadiel,Male,Hispanic
+Alexis,Male,Hispanic
+Josue,Male,Hispanic
+Jorge,Male,Hispanic
+Andres,Male,Hispanic
+Adriel,Male,Hispanic
+Johan,Male,Hispanic
+Ayaan,Male,Asian
+Eason,Male,Asian
+Tenzin,Male,Asian
+Syed,Male,Asian
+Kingsley,Male,Asian
+Arjun,Male,Asian
+Carson,Male,Asian
+Arayan,Male,Asian
+Anson,Male,Asian
+Benson,Male,Asian
+Lawrence,Male,Asian
+Ayan,Male,Asian
+Rohan,Male,Asian
+Roy,Male,Asian
+Aarav,Male,Asian
+Rayyan,Male,Asian
+Kimi,Male,Asian
+Zayan,Male,Asian
+Ricky,Male,Asian
+Arham,Male,Asian
+Fatoumata,Female,African American
+Aminata,Female,African American
+Amiyah,Female,African American
+Zuri,Female,African American
+Kimora,Female,African American
+Mariama,Female,African American
+Sanaa,Female,African American
+Lyric,Female,African American
+Sanai,Female,African American
+Harmony,Female,African American
+Aicha,Female,African American
+Tori,Female,African American
+Maliyah,Female,African American
+Aisaatou,Female,African American
+Miracle,Female,African American
+Hawa,Female,African American
+Oumou,Female,African American
+Dakota,Female,African American
+Skye,Female,African American
+Kyla,Female,African American
+Emely,Female,Hispanic
+Leslie,Female,Hispanic
+Andrea,Female,Hispanic
+Valeria,Female,Hispanic
+Aylin,Female,Hispanic
+Jayleen,Female,Hispanic
+Yaretzi,Female,Hispanic
+Melany,Female,Hispanic
+Danna,Female,Hispanic
+Brittany,Female,Hispanic
+Alison,Female,Hispanic
+Jazmin,Female,Hispanic
+Briana,Female,Hispanic
+Kamila,Female,Hispanic
+Alaia,Female,Hispanic
+Ximena,Female,Hispanic
+Sherlyn,Female,Hispanic
+Esmeralda,Female,Hispanic
+Guadalupe,Female,Hispanic
+Jazlyn,Female,Hispanic
+Tenzin,Female,Asian
+Selina,Female,Asian
+Ayesha,Female,Asian
+Vicky,Female,Asian
+Elaine,Female,Asian
+Jenny,Female,Asian
+Winnie,Female,Asian
+Queenie,Female,Asian
+Sharon,Female,Asian
+Alisha,Female,Asian
+Elina,Female,Asian
+Erica,Female,Asian
+Manha,Female,Asian
+Syeda,Female,Asian
+Jannat,Female,Asian
+Janice,Female,Asian
+Tina,Female,Asian
+Anya,Female,Asian
+Arisha,Female,Asian
+Inaaya,Female,Asian