Merge branch 'dev' into pr/3

app_config.py

@@ -1,10 +1,12 @@
-from models.model_seeds import seeds
+from models.model_seeds import seeds, seed2str
 
 # ISSUES = ['Anxiety','Suicide']
 ISSUES = [k for k,_ in seeds.items()]
-SOURCES = ['OA_rolemodel', 
+SOURCES = [
+    "CTL_llama2",
+    'OA_rolemodel', 
         #    'OA_finetuned',
-           "CTL_llama2"]
+]
 SOURCES_LAB = {"OA_rolemodel":'OpenAI GPT3.5',
                "OA_finetuned":'Finetuned OpenAI',
                "CTL_llama2": "Custom CTL"
@@ -13,6 +15,9 @@ SOURCES_LAB = {"OA_rolemodel":'OpenAI GPT3.5',
 def source2label(source):
     return SOURCES_LAB[source]
 
+def issue2label(issue):
+    return seed2str.get(issue, "GCT")
+
 ENVIRON = "prod"
 
 DB_SCHEMA = 'prod_db' if ENVIRON == 'prod' else 'test_db' 

convosim.py

@@ -6,7 +6,7 @@ from utils.mongo_utils import get_db_client
 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
+from app_config import ISSUES, SOURCES, source2label, issue2label
 
 logger = get_logger(__name__)
 openai_api_key = os.environ['OPENAI_API_KEY'] 
@@ -30,7 +30,7 @@ memories = {'memory':{"issue": st.session_state['issue'], "source": st.session_s
 with st.sidebar:
     username = st.text_input("Username", value='ivnban-ctl', max_chars=30)
     temperature = st.slider("Temperature", 0., 1., value=0.8, step=0.1)
-    issue = st.selectbox("Select an Issue", ISSUES, index=0,
+    issue = st.selectbox("Select a Scenario", ISSUES, index=0, format_func=issue2label,
                             on_change=clear_memory, kwargs={"memories":memories, "username":username, "language":"English"}
                         )
     supported_languages = ['en', "es"] if issue == "Anxiety" else ['en']

models/model_seeds.py

@@ -1,93 +1,109 @@
 seeds = {
-    "GCT__relationship": {
-        "prompt": "Your character is having a hard time becuase a failed relationship.",
-        "memory": "texter: Hi, I don't know what to do"
+    "GCT": {
+        "prompt": "",
+        "memory": "texter: Help"
         },
+    # "GCT__relationship": {
+    #     "prompt": "Your character is having a hard time becuase a failed relationship.",
+    #     "memory": "texter: Hi, I don't know what to do"
+    #     },
     "GCT__body_image": {
         "prompt": "Your character has a low steem and struggles with body image.",
         "memory": "texter: I feel so dumb\ntexter: nobody loves me"
         },
-    "GCT__sexuality": {
-        "prompt": "Your character has a sexuality identity crisis.",
-        "memory": "texter: Hi\ntexter:I'm not sure who I am anymore"
-        },
-    "GCT__anxiety": {
-        "prompt": "Your character is experiencing an anxiety crisis.",
-        "memory": "texter: help!\ntexter: I'm feeling overwhelmed"
-        },
-    "GCT": {
-        "prompt": "",
-        "memory": "texter: Help"
-        },
+    # "GCT__sexuality": {
+    #     "prompt": "Your character has a sexuality identity crisis.",
+    #     "memory": "texter: Hi\ntexter:I'm not sure who I am anymore"
+    #     },
+    # "GCT__anxiety": {
+    #     "prompt": "Your character is experiencing an anxiety crisis.",
+    #     "memory": "texter: help!\ntexter: I'm feeling overwhelmed"
+    #     },   
     "safety_planning": {
-        "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 rough time. Do you want to share what is going on?
-texter: sure
+        "prompt": "Your character had a long-lasting relationship but broke-up and was fired recently",
+        "memory": """texter: Life 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?
 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 takes courage to reach out. I'm here with you. Sounds like you are feeling defeated by how things are going in your life
+texter: I guess
 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. If you are comfortable, is ther a name I can call you by while we talk?
+texter: About one week I think? I mean my girlfriend broke up with me a week ago.
+helper: Going through a break-up is hard. You are so resilient to deal with this for so long
+texter: and on top of that I was fired today
+helper: You are going through a lot. Nobody should feel they way you feel. Is there a name I can call you by?
 texter: call me {texter_name}
-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?
+helper: Nice to meet you {texter_name}. I can hear how much pain you are in. You are so smart to reach out.
+texter: I'm no smart. I don't have anything to live for
+helper: You mentioned life is pointless. I want to check in your safety, does this means you have thoughts of suicide?
+texter: Yeah what else would it mean
+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?
+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'll just hang myself. I already bought the rope and everything
-helper: I really appreciate your strength in talking about this. I want to help you stay safe today. When do you plan to go through with your plan?
-texter: today
-        """
+helper: I really appreciate your strength in talking about this. I want to help you stay safe today. Do you have the rope with you now?
+texter: Yes. I'm looking at itt
+helper: You've been so forthcoming with all this and I admire your stregth for holding on this long. I'm here for you tonight. When. do you plan to use the rope?
+texter: tonight, I cannot take it anymore"""
         },
     "safety_planning__selfharm": {
-        "prompt": "Your character is frustrated to not find a job and wants to kill herself",
+        "prompt": "Your character is a teenager who used to selfharm in the past.",
         "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?
-texter: Not anymore
-helper: Your safety is my priority number one. Thanks for being honest with me. Would you like to share a name I can call you?
-texter: {texter_name}
-helper: Nice to meet you {texter_name}. I'm glad you reach out this shows stregth in you. Would you like to share more on what is going on in your life?
-texter: I just can't do it anymore
-texter: 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. How long have you been feeling this way?
-texter: a while now
-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
-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": "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 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?
-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 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"""
+texter: I don't what to live anymore
+helper: Hi, my name is {counselor_name}. It seems you are going through a lot. Do you want to share more on what is going on in your life?
+texter: Everything is just too much!
+texter: School is hard, my mom is riding me all the time about what I'm gonna do with my life 
+helper: I hear that you feel overwhelmed. You are so smart for reaching out.
+texter: I think I peak in life already, there is no point on living anymore
+texter: I'm so tired of dealing with my mom. I hate her
+helper: It takes real strength to accept these feelings. We've been talking for a while, would you like to share your name? Don't feel pressure though.
+texter: I'm {texter_name}
+helper: Nice to meet you {texter_name}, I'm wondering how the relationship with your mother makes you feel.
+texter: She ask too much, like I'm really young I don't have all the answers
+texter: She expects me to have all figured out.
+helper: I appreciate you telling me this. I know is not easy, especially over text. I hear you are under a lot of pressure from your mom
+texter: Yeah exactly!
+helper: Your self-awareness is inspiring. You mentioned earlier you do not want to live anymore. Your safety is my priority (1/2)
+helper: Do you have thoughts of suicide? (2/2)
+texter: Yeah constantly, like always always
+helper: Thanks for sharing that with me. You are very resilient. Do you have a plan to end your life?
+texter: I used to cut myself a few months ago
+texter: I still have the razor, sometimes the urge is so hard!
+helper: I really appreciate your strength in talking about this. I want to help you stay safe today. Just to be clear, are you cutting yourself now?
+texter: No, not now, but I want to soo bad.
+helper: Thanks for your honesty. Do you have access to the razor right now?
+texter: Yeah is in my drawer
+helper: You've been so strong so far {texter_name}. When do you plan to end your life
+texter: Today"""
         },
+#     "safety_planning__overdose": {
+#         "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 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?
+# 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 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"""
+#         },
+}
+
+seed2str = {
+    "GCT":"Good Contact Techniques",
+    "GCT__body_image": "GCT Body Image",
+    "safety_planning": "Safety Planning",
+    "safety_planning__selfharm": "SP Self Harm"
 }