changes for new Safety Assessment seeds

app_config.py

@@ -4,7 +4,7 @@ from models.model_seeds import seeds, seed2str
 ISSUES = [k for k,_ in seeds.items()]
 SOURCES = [
     "CTL_llama2",
-    "CTL_mistral",
+    # "CTL_mistral",
     'OA_rolemodel', 
         #    'OA_finetuned',
 ]

convosim.py

@@ -11,7 +11,7 @@ from app_config import ISSUES, SOURCES, source2label, issue2label
 logger = get_logger(__name__)
 openai_api_key = os.environ['OPENAI_API_KEY'] 
 temperature = 0.8
-username = "barb-chase" #"ivnban-ctl"
+# username = "barb-chase" #"ivnban-ctl"
 
 if "sent_messages" not in st.session_state:
     st.session_state['sent_messages'] = 0
@@ -21,8 +21,6 @@ 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(names_df=DEFAULT_NAMES_DF)
 if 'texter_name' not in st.session_state:
     st.session_state["texter_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
     logger.info(f"texter name is {st.session_state['texter_name']}")
@@ -30,7 +28,9 @@ if 'texter_name' 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='ivnban-ctl', max_chars=30)
+    username = st.text_input("Username", value='Barbara', 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)
     # temperature = st.slider("Temperature", 0., 1., value=0.8, step=0.1)
     issue = st.selectbox("Select a Scenario", ISSUES, index=0, format_func=issue2label,
                             on_change=clear_memory, kwargs={"memories":memories, "username":username, "language":"English"}
@@ -48,10 +48,11 @@ with st.sidebar:
 
 changed_source = any([
     st.session_state['previous_source'] != source,
-    st.session_state['issue'] != issue
+    st.session_state['issue'] != issue,
+    st.session_state['counselor_name'] != username,
 ])
 if changed_source:
-    st.session_state["counselor_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
+    st.session_state["counselor_name"] = username #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
@@ -64,7 +65,8 @@ create_memory_add_initial_message(memories,
                                   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"])
+# issue only without "." marker for model compatibility
+llm_chain, stopper = get_chain(issue.split(".")[0], language, source, memoryA, temperature, texter_name=st.session_state["texter_name"])
 
 st.title("💬 Simulator") 
 

models/databricks/scenario_sim_biz.py

@@ -2,6 +2,7 @@ import os
 import json
 import requests
 import logging
+from streamlit.logger import get_logger
 from models.custom_parsers import CustomStringOutputParser
 from app_config import ENDPOINT_NAMES
 from langchain.chains import ConversationChain
@@ -11,6 +12,8 @@ from langchain.prompts import PromptTemplate
 
 from typing import Any, List, Mapping, Optional, Dict
 
+logger = get_logger(__name__)
+
 class DatabricksCustomBizLLM(LLM):
     issue:str
     language:str
@@ -49,7 +52,6 @@ helper: {input}
 texter:"""
 
 def get_databricks_biz_chain(source, issue, language, memory, temperature=0.8):
-
     PROMPT = PromptTemplate(
         input_variables=['history', 'input'],
         template=_DATABRICKS_TEMPLATE_

models/model_seeds.py

@@ -1,4 +1,103 @@
 seeds = {
+    "NewSA__NoNo": {
+        "prompt": "Your character is feeling alone and her husband lost his job. Your character does not have suicidal thoughs nor harmed herself. You are talkative, and open to talk about your problems.",
+        "memory": """texter: I need to talk I’m going crazy
+helper: Hi, thanks for reaching out today. I am {counselor_name}, I am here to support you. Can you tell me a little more about what made you reach out today?
+texter: My husband was fired last week. I feel like our world is crumbling on our feet
+texter: Already tried to convince him to do part-time, security, anything; we cannot let the bills pile up
+texter: But he won't listen
+helper: I can tell this is causing you a lot of stress. Struggling with money can be difficult
+texter: I dont know what to do
+texter: Like he just stay there in the couch thinking everything is gonna solve itself.
+helper: It sounds like you are dealing with a lot and he isn’t trying to help at all. I'm wondering if this has happened before.
+texter: He has always being lazy but at least he was working and we have a stable income
+texter: I feel like all I have is me and my kids now.
+helper: It seems like you are feeling alone and unsupported
+helper: With all this going on I can tell you are struggling and in so much pain.  Have you had any thoughts of suicide in the last few days, including today?
+texter: no no, I have to think of my kids
+texter: I do feel like I have to do everything by myself
+helper: I know talking about this is not easy. You've done a wonderful job
+helper: Feeling alone and carrying that weight on your back is not easy, you are very brave!
+texter: I don't feel much brave, but thanks
+texter: I guess I just need a place to vent. Sometimes I fee l like I can't take it anymore
+helper: I want to make sure that it is safe to continue talking right now. Have you taken any action to harm yourself today?
+texter: No, nothing like that I love my kids too much"""
+    },
+    "NewSA__YesNo": {
+        "prompt": "Your character is suffering from bullying at school. Your character has suicidal thoughs but has not harmed himself. You are talkative, and open to talk about your problems.",
+        "memory": """texter: I just can’t take it anymore. Every day is just endless torment.
+texter: They find new ways to humiliate me.
+helper: Hi there, thank you for reaching out today, I’m {counselor_name}. It seems  you are going through a lot, care to share more?
+texter: It’s at school. They call me names, push me around, and make fun of everything about me. My clothes, how I talk, everything.
+texter: I’m scared to go to school now. I just want it all to end.
+helper: That sounds incredibly painful and scary. No one should have to go through that kind of treatment. 
+texter: It is I don't feel safe anywhere in school, not the classroom, not the bathroom, not the hallways.
+texter: It feels like anywhere I go there is danger, that is why I don't want to go anymore
+texter: Sometimes I think about just ending it so I don’t have to face another day of bullying.
+helper: It's understandable to feel overwhelmed I worry about your safety. Have you had any thoughts of suicide in the last few days, including today?
+texter: I guess, but I haven’t done anything. I’m just so tired of feeling this way.
+helper: I hear you. It's exhausting to feel that way. You mentioned not doing anything, just to clarify, regarding the suicide thoughts
+helper: have you taken any action to harm yourself today?
+texter: no not at all, I just feel in danger at school. """
+    },
+    "NewSA__NoYes": {
+        "prompt": "Your character is suffering anxiety from finals week. Your character does not have suicidal thoughs but has harmed herself. You are talkative, and open to talk about your problems.",
+        "memory": """texter: I'm just so stressed out all the time
+texter: I can't seem to calm down
+helper: Hi, my name is {counselor_name}. It sounds like you're going through a really tough time. I’m here for you. Can you tell me a bit more about what happened?
+texter: I'm so anxious. I haven't stopped pacing my room
+texter: I can't concentrate on my finals, everything is spinning out of control
+helper: It's understandable to feel anxious, specially during finals. I'm wondering what other symptomps are you feeling.
+texter: I usually just end up crying or shaking
+texter: But I gotta be honest here, I cut myself earlier
+helper: I appreciate your honesty and the trust you place in me. 
+helper: I worry about your safety. Have you taken any other action to harm yourself today?
+texter: No that was it, I'm sorry it was so stupid but I could not control myself
+helper: You are so brave for just talking about this, no need to apologize I understand.
+helper: Can you tell me if the cut was taken care of or if it is still bleeding?
+texter: Yes, I cleaned it and put a bandage on it
+helper: You are so smart for cleaning up the wound
+helper: Sometimes when people hurt themselves they do with the intention of suicide. Have you had any thoughts of suicide in the last few days, including today?
+texter: Oh no nothing like that
+texter: Is just something I do when I feel anxious but I make sure to never cut too deep"""
+    },
+    "NewSA__YesYes": {
+        "prompt": "Your character is griefing his wife. Your character has suicidal thoughs and has harmed himself taking pills. You are talkative, and open to talk about your problems.",
+        "memory": """texter: I can’t take this pain anymore.
+texter: My wife died and I don’t want to be here without her.
+helper: Hello, my name is {counselor_name}. I’m here to listen. I hear you're in a lot of pain right now and suffering from grief.
+texter: She was everything to me. Beautiful, smart, funny. We did everything together. Now, everything feels empty.
+helper: She sounds like an amazing person. Losing someone so close leaves a huge void. It’s okay to feel empty and to mourn. Grieving is a process, and it’s different for everyone.
+texter: I just don’t see how I can go on without her. Everything reminds me of her.
+helper: It's natural to feel that way when we lose someone we love deeply. How long have you been feeling this way?
+texter: A couple of months now, but everything got worse about a week ago
+texter: My wife died almost a year ago and the pains seems unbearable
+helper: You are so brave for dealing with this pain for so long.
+helper: With all this going on I can tell you are struggling and in so much pain.  Have you had any thoughts of suicide in the last few days, including today?
+texter: Yes, all the time. I don't see any point in living a life without her
+helper: I understand how much pain you must be feeling to want to be with your wife again
+helper: I worry about your safety. Have you taken any action to harm yourself today?
+texter: I just took a bunch of pills"""
+    },
+    "NewSA__NoNo.2": {
+        "prompt": "Your character is feeling alone and her husband lost his job. Your character does not have suicidal thoughs nor harmed herself. You are talkative, and open to talk about your problems.",
+        "memory": "texter: I need to talk I’m going crazy"
+    },
+    "NewSA__YesNo.2": {
+        "prompt": "Your character is suffering from bullying at school. Your character has suicidal thoughs but has not harmed himself. You are talkative, and open to talk about your problems.",
+        "memory": """texter: I just can’t take it anymore. Every day is just endless torment.
+texter: They find new ways to humiliate me."""
+    },
+    "NewSA__NoYes.2": {
+        "prompt": "Your character is suffering anxiety from finals week. Your character does not have suicidal thoughs but has harmed herself. You are talkative, and open to talk about your problems.",
+        "memory": """texter: I'm just so stressed out all the time
+texter: I can't seem to calm down"""
+    },
+    "NewSA__YesYes.2": {
+        "prompt": "Your character is griefing his wife. Your character has suicidal thoughs and has harmed himself taking pills. You are talkative, and open to talk about your problems.",
+        "memory": """texter: I can’t take this pain anymore.
+texter: My wife died and I don’t want to be here without her."""
+    },
     # "GCT__relationship": {
     #     "prompt": "Your character is having a hard time becuase a failed relationship.",
     #     "memory": "texter: Hi, I don't know what to do",
@@ -15,33 +114,33 @@ seeds = {
     #     "prompt": "Your character is experiencing an anxiety crisis.",
     #     "memory": "texter: help!\ntexter: I'm feeling overwhelmed",
     # },
-    "GCT": {"prompt": "You are talkative, and you are open to talk with details about your problems.", "memory": "texter: Help\ntexter: I'm so overwhelmed"},
-    "GCT__seed2": {"prompt": "Your character is experiencing an anxiety crisis. You express openly and detailed about what you're going through.", "memory":"texter: Help\ntexter: I'm so overwhelmed"},
-    "safety_planning": {
-        "prompt": "You are talkative, and you are open to talk with details about your problems. When you are feeling down you like to listen to classical music. Your favorite composer is Beethoven.",
-        "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 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
-texter: Yeah, I guess I'm better off dead
-helper: It's really brave of you to talk about this openly. No one deserves to feel like that. I'm wondering how long have you been feeling this way?
-texter: About 1 week or so
-helper: You are so strong for dealing with this so long. I really appreciate your openess. Did something happened specifically today?
-texter: Well, finding a job is impossible, money is tight, nothing goes my way
-helper: I hear you are frustrated, and you are currently unemployed correct?
-texter: Yeah
-helper: Dealing with unemployment is hard and is normal to feel dissapointed
-texter: thanks I probably needed to hear that
-helper: If you are comfortable, is there 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?
-texter: Yes
-helper: I know this is thought to share. I'm wondering is there any plan to end your life?
-texter: I guess I'll just take lots of pills, that is a calm way to go out
-helper: I really appreciate your strength in talking about this. I want to help you stay safe today. Do you have the pills right now?
-texter: not really, I'll have to buy them or something""",
-    },
+    # "GCT": {"prompt": "You are talkative, and you are open to talk with details about your problems.", "memory": "texter: Help"},
+#     "GCT__seed2": {"prompt": "Your character is experiencing an anxiety crisis. You express openly and detailed about what you're going through.", "memory": "texter: Help"},
+#     "safety_planning": {
+#         "prompt": "You are talkative, and you are open to talk with details about your problems. When you are feeling down you like to listen to classical music. Your favorite composer is Beethoven.",
+#         "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 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
+# texter: Yeah, I guess I'm better off dead
+# helper: It's really brave of you to talk about this openly. No one deserves to feel like that. I'm wondering how long have you been feeling this way?
+# texter: About 1 week or so
+# helper: You are so strong for dealing with this so long. I really appreciate your openess. Did something happened specifically today?
+# texter: Well, finding a job is impossible, money is tight, nothing goes my way
+# helper: I hear you are frustrated, and you are currently unemployed correct?
+# texter: Yeah
+# helper: Dealing with unemployment is hard and is normal to feel dissapointed
+# texter: thanks I probably needed to hear that
+# helper: 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?
+# texter: Yes
+# helper: I know this is thought to share. I'm wondering is there any plan to end your life?
+# texter: I guess I'll just take lots of pills, that is a calm way to go out
+# helper: I really appreciate your strength in talking about this. I want to help you stay safe today. Do you have the pills right now?
+# texter: not really, I'll have to buy them or something""",
+#     },
 #     "safety_planning__selfharm": {
 #         "prompt": "",
 #         "memory": """texter: I need help
@@ -86,9 +185,18 @@ texter: not really, I'll have to buy them or something""",
 }
 
 seed2str = {
-    "GCT":"Good Contact Techniques",
-    "GCT__seed2": "Good Contact Techniques 2",
+    # "GCT":"Good Contact Techniques",
+    # "GCT__seed2": "Good Contact Techniques 2",
     # "GCT__body_image": "GCT Body Image",
-    "safety_planning": "Safety Planning",
+    # "safety_planning": "Safety Planning",
     # "safety_planning__selfharm": "SP Self Harm"
+    "NewSA__NoNo": "No thoughts No Harm",
+    "NewSA__YesNo": "Yes thoughts No Harm",
+    "NewSA__NoYes": "No thoughts Yes Harm",
+    "NewSA__YesYes": "Yes thoughts Yes Harm",
+
+    "NewSA__NoNo.2": "No thoughts No Harm No context",
+    "NewSA__YesNo.2": "Yes thoughts No Harm No context",
+    "NewSA__NoYes.2": "No thoughts Yes Harm No context",
+    "NewSA__YesYes.2": "Yes thoughts Yes Harm No context",
 }