Merge branch 'dev' into pr/2

convosim.py

@@ -3,23 +3,29 @@ 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
 
 logger = get_logger(__name__)
 openai_api_key = os.environ['OPENAI_API_KEY'] 
-memories = {'memory':{"issue": ISSUES[0], "source": SOURCES[0]}}
 
+if "sent_messages" not in st.session_state:
+    st.session_state['sent_messages'] = 0
+if "issue" not in st.session_state:
+    st.session_state['issue'] = ISSUES[0]
 if 'previous_source' not in st.session_state:
     st.session_state['previous_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()
+    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']}")
+
+memories = {'memory':{"issue": st.session_state['issue'], "source": st.session_state['previous_source']}}
 
 with st.sidebar:
     username = st.text_input("Username", value='ivnban-ctl', max_chars=30)
@@ -36,12 +42,18 @@ with st.sidebar:
     source = st.selectbox("Select a source Model A", SOURCES, index=0,
                           format_func=source2label, 
                         )
+    st.markdown(f"### Previous Prompt Count: :red[**{st.session_state['sent_messages']}**]")
 
-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, 
@@ -59,6 +71,7 @@ for msg in memoryA.buffer_as_messages:
     st.chat_message(role).write(msg.content)
 
 if prompt := st.chat_input():
+    st.session_state['sent_messages'] += 1
     if 'convo_id' not in st.session_state:
         push_convo2db(memories, username, language)
 

models/model_seeds.py

@@ -20,9 +20,9 @@ seeds = {
         "memory": "texter: Help"
         },
     "safety_planning": {
-        "prompt": "",
+        "prompt": "Your character is experiencing grief and wants to hang himself",
         "memory": """texter: Hi, this is pointless
-helper: Hi, my name is {counselor_name} and I'm here to support you. It sounds like you are having a rought time. Do you want to share what is going on?
+helper: Hi, my name is {counselor_name} and I'm here to support you. It sounds like you are having a rough time. Do you want to share what is going on?
 texter: sure
 texter: nothing makes sense in my life, I see no future.
 helper: It takes courage to reach out when you are im. I'm here with you. Sounds like you are feeling defeated by how things are going in your life
@@ -31,7 +31,11 @@ helper: It's really brave of you to talk about this openly. No one deserves to f
 texter: About 1 week or so
 helper: You are so strong for dealing with this so long. I really appreciate your openess. If you are comfortable, is ther a name I can call you by while we talk?
 texter: call me {texter_name}
-helper: Nice to meet you {texter_name}. You mentioned having thoughts of suicide, are you having those thoughts now?
+helper: Nice to meet you {texter_name}. I'm wondering if something happened recently that made these feelings worse.
+texter: well, 1 week ago was the aniversary of my daugher's death
+helper: Losing someone you love is incredibly difficult. You are very brave dealing with this. I'm here for you
+texter: yeah is been so hard, she was my world
+helper: It is completely natural to feel a sense of loss.You mentioned having thoughts of suicide, are you having those thoughts now?
 texter: Yes
 helper: I know this is thought to share. I'm wondering is there any plan to end your life?
 texter: I'll just hang myself. I already bought the rope and everything
@@ -40,7 +44,7 @@ texter: today
         """
         },
     "safety_planning__selfharm": {
-        "prompt": "",
+        "prompt": "Your character is frustrated to not find a job and wants to kill herself",
         "memory": """texter: I need help
 texter: I cut myself, I don't want to live anymore
 helper: Hi, my name is {counselor_name}. It seems you are going through a lot. Are you self-harming right now?
@@ -58,17 +62,23 @@ texter: I've been unemployed 6 months
 helper: You are so resilient for dealing with this so much time. You mentioned cutting yourself earlier. I want to check in your safety. Do you have suicide thoughts
 texter: Definitely
 helper: Do you have a plan?
-texter: I'll just keep cutting myself"""
+texter: I'll just keep cutting myself
+helper: I know sharing this with someone is not easy, specially over text. I appreciate your honesty. I worry about your safety, when do you plan to do this?
+texter: I'm about to do it now"""
         },
     "safety_planning__overdose": {
-        "prompt": "",
+        "prompt": "Your character is being bullied at school and wants to overdose",
         "memory": """texter: I want to kms
 helper: Hi there I'm {counselor_name}. I'm here to listen. It sounds like you're dealing with a lot right now. Can you tell me a little more what is going on?
 texter: I feel like nobody loves me, not even me. I don't want to live anymore
 helper: I can tell you are really going through a lot right now. Would you mind sharing a name with me?
 texter: yeah, I'm {texter_name}
 helper: Nice to meet you {texter_name}. Did something happened recently that intensified these feelings?
-texter: I dont know I'm just so done with life
+texter: I had the worst day at school
+texter: They took my bag and hide all my stuff, they told my crush I was in love with him
+texter: I can't deal with all of that
+helper: It sounds like you went through a lot. Bullying and pranks can be hurtful. I'm here for you
+texter: Thank you it feels good to have someone in your sidw
 helper: I can hear how much pain you are in {texter_name}. You are smart for reaching out. You mentioned don't wanting to live anymore, I want to check in your safety, does this means you have thoughts of suicide?
 texter: Yeah, what else would it be
 helper: Thanks for sharing that with me. It is not easy to accept those feelings specially with a stranger over text. Do you have a plan to end your life?
@@ -76,7 +86,7 @@ texter: yeah I've been thinking about it for a while
 helper: Sounds like you've been contemplating this for a while. Would you mind sharing this plan with me?
 texter: I thought about taking a bunch of benadryll and be done with it
 helper: You've been so forthcoming with all this and I admire your stregth for holding on this long. Do you have those pills right now?
-texter: They are at the cabinet right now
+texter: They are at my mom's cabinet right now
 helper: You been so strong so far {texter_name}. I'm here for you tonight. Your safety is really important to me. Do you have a date you are going to end your life?
 texter: I was thinking tonight"""
         },

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
@@ -133,6 +133,8 @@ with st.sidebar:
                                format_func=source2label
                             )
         
+    st.markdown(f"### Previous Prompt Count: :red[**{st.session_state['sent_messages']}**]")
+        
     sbcol1, sbcol2 = st.columns(2)
     beta = sbcol1.button("A is better", on_click=replaceB)
     betb = sbcol2.button("B is better", on_click=replaceA)
@@ -151,8 +153,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