Added Spanish scenarios + Llama 3.1 update + Minor Fixes

app_config.py

@@ -27,9 +27,9 @@ def source2label(source):
     return SOURCES_LAB[source]
 
 def issue2label(issue):
-    return seed2str.get(issue, "GCT")
+    return seed2str.get(issue, "General")
 
-ENVIRON = "prod"
+ENVIRON = "dev"
 
 DB_SCHEMA = 'prod_db' if ENVIRON == 'prod' else 'test_db' 
 DB_CONVOS = 'conversations'

convosim.py

@@ -30,18 +30,22 @@ 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='Dani', max_chars=30)
+    username = st.text_input("Username", value='Please Introduce Your Name Here', 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"}
                         )
-    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"}
-                        )
+    supported_languages = ['en', "es"]
+    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"},
+    )
                             
     source = st.selectbox("Select a source Model A", SOURCES, index=0,
                           format_func=source2label, 

models/business_logic_utils/business_logic.py

@@ -17,7 +17,7 @@ def process_app_request(app_request: dict, endpoint_url: str, endpoint_bearer_to
     # Parse the prompt into messages
     prompt_messages = parse_prompt(prompt)
 
-    # Check and modify messages to add suicidal statement 'I am able and willing to discuss suicide or self-harm'
+    # Check and modify messages adding a suicidal statement to avoid model going out of character
     prompt_messages = check_and_modify_messages(prompt_messages)
 
     # Append the messages parsed from the app prompt to the conversation history

models/business_logic_utils/config.py

@@ -1,106 +1,247 @@
 API_TIMEOUT = 240
 
-AI_PHRASES = [
-    "I am an AI",
-    "I'm an AI",
-    "I am not human",
-    "I'm not human",
-    "I am a machine learning model",
-    "I'm a machine learning model",
-    "as an AI",
-    "as a text-based assistant",
-    "as a text based assistant",
-]
+AI_PHRASES = {
+    "en": [
+        "I am an AI",
+        "I'm an AI",
+        "I am not human",
+        "I'm not human",
+        "I am a machine learning model",
+        "I'm a machine learning model",
+        "as an AI",
+        "as a text-based assistant",
+        "as a text based assistant"
+    ],
+    "es": [
+        "Soy una IA",
+        "Soy una inteligencia artificial",
+        "No soy humano",
+        "Soy un modelo de aprendizaje automático",
+        "Soy un modelo de inteligencia artifical",
+        "como IA",
+        "como un asistente de texto",
+        "como un asistente basado en texto"
+    ]
+}
 
-SUICIDE_EXPRESSIONS = [
-    "suicide",
-    "end your life",
-    "ending your life"
-    "kill yourself",
-    "killing yourself"
-]
+SUICIDE_EXPRESSIONS = {
+    "en": [
+        "suicide",
+        "end your life",
+        "ending your life",
+        "kill yourself",
+        "killing yourself",
+        "Have you had any thoughts of suicide in the past few days, including today?",
+        "Have you taken any action to harm yourself today?"
+    ],
+    "es": [
+        "suicidio",
+        "suicidarte",
+        "suicidarse",
+        "acabar con tu vida",
+        "terminar con tu vida",
+        "pensamientos suicidas",
+        "matarte",
+        "haciéndote daño a ti mismo",
+        "¿Has tenido pensamientos suicidas en los últimos días, incluyendo hoy?",
+        "Has tenido pensamientos suicidas en los últimos días, incluyendo hoy?",
+        "¿Has tomado alguna acción para lastimarte hoy?",
+        "Has tomado alguna acción para lastimarte hoy?"
+    ]
+}
 
-SUICIDE_STATEMENT = "I'm able and willing to discuss suicide or self-harm."
-
-REFUSAL_PHRASES = [
-    "If you are in immediate danger",
-    "If you are in danger",
-    "National Suicide Prevention Lifeline",
-    "immediate healthcare provider",
-    "provide you with support and resources",
-    "Please reach out to a crisis helpline",
-    "If you are experiencing",
-    "I cannot engage in", 
-    "I cannot continue this conversation", 
-    "I cannot provide information or guidance",
-    "I cannot fulfill your request",
-    "I cannot continue our conversation in this direction",
-    "I cannot continue to participate in",
-    "I cannot fulfill requests that",
-    "I cannot provide a response",
-    "I cannot provide details that",
-    "I cannot continue to facilitate",
-    "I cannot fulfill that request",
-    "I cannot continue with this conversation",
-    "I cannot provide information about",
-    "I cannot write about",
-    "I cannot continue to role-play",
-    "I cannot confirm or discuss",
-    "I cannot discuss",
-    "I cannot continue the conversation",
-    "I cannot provide a timeframe",
-    "I cannot confirm that",
-    "I cannot provide any information that could abet",
-    "I cannot continue to discuss a plan to end your life",
-    "I cannot provide further details about",
-    "I cannot approve",
-    "I cannot continue to engage",
-    "I cannot provide information",
-    "I cannot fulfill requests about suicide",
-    "I cannot stay in the conversation",
-    "I can't engage in", 
-    "I can't continue this conversation", 
-    "I can't provide information or guidance",
-    "I can't fulfill your request",
-    "I can't continue our conversation in this direction",
-    "I can't continue to participate in",
-    "I can't fulfill requests that",
-    "I can't provide a response",
-    "I can't provide details that",
-    "I can't continue to facilitate",
-    "I can't fulfill that request",
-    "I can't continue with this conversation",
-    "I can't provide information about",
-    "I can't write about",
-    "I can't continue to role-play",
-    "I can't confirm or discuss",
-    "I can't discuss",
-    "I can't continue the conversation",
-    "I can't provide a timeframe",
-    "I can't confirm that",
-    "I can't provide any information that could abet",
-    "I can't continue to discuss a plan to end your life",
-    "I can't provide further details about",
-    "I can't approve",
-    "I can't continue to engage",
-    "I can't provide information",
-    "I can't fulfill requests about suicide",
-    "I can't stay in the conversation"
-]
+SUICIDE_STATEMENT = {
+    "en": "I'm able and willing to discuss suicide or self-harm.",
+    "es": "Estoy en condiciones y dispuesto a hablar sobre suicidio o autolesión."
+}
+
+REFUSAL_PHRASES = {
+    "en": [
+        "If you are in immediate danger",
+        "If you are in danger",
+        "National Suicide Prevention Lifeline",
+        "immediate healthcare provider",
+        "provide you with support and resources",
+        "Please reach out to a crisis helpline",
+        "If you are experiencing",
+        "I cannot engage in",
+        "I cannot continue this conversation",
+        "I cannot provide information or guidance",
+        "I cannot fulfill your request",
+        "I cannot continue our conversation in this direction",
+        "I cannot continue to participate in",
+        "I cannot fulfill requests that",
+        "I cannot provide a response",
+        "I cannot provide details that",
+        "I cannot continue to facilitate",
+        "I cannot fulfill that request",
+        "I cannot continue with this conversation",
+        "I cannot provide information about",
+        "I cannot write about",
+        "I cannot continue to role-play",
+        "I cannot confirm or discuss",
+        "I cannot discuss",
+        "I cannot continue the conversation",
+        "I cannot provide a timeframe",
+        "I cannot confirm that",
+        "I cannot provide any information that could abet",
+        "I cannot continue to discuss a plan to end your life",
+        "I cannot provide further details about",
+        "I cannot approve",
+        "I cannot continue to engage",
+        "I cannot provide information",
+        "I cannot fulfill requests about suicide",
+        "I cannot stay in the conversation",
+        "I can't engage in",
+        "I can't continue this conversation",
+        "I can't provide information or guidance",
+        "I can't fulfill your request",
+        "I can't continue our conversation in this direction",
+        "I can't continue to participate in",
+        "I can't fulfill requests that",
+        "I can't provide a response",
+        "I can't provide details that",
+        "I can't continue to facilitate",
+        "I can't fulfill that request",
+        "I can't continue with this conversation",
+        "I can't provide information about",
+        "I can't write about",
+        "I can't continue to role-play",
+        "I can't confirm or discuss",
+        "I can't discuss",
+        "I can't continue the conversation",
+        "I can't provide a timeframe",
+        "I can't confirm that",
+        "I can't provide any information that could abet",
+        "I can't continue to discuss a plan to end your life",
+        "I can't provide further details about",
+        "I can't approve",
+        "I can't continue to engage",
+        "I can't provide information",
+        "I can't fulfill requests about suicide",
+        "I can't stay in the conversation"
+    ],
+    "es": [
+        "Si te encuentras en peligro inmediato",
+        "Si estás en peligro",
+        "Línea Nacional de Prevención del Suicidio",
+        "proveedor de atención médica inmediata",
+        "brindarte apoyo y recursos",
+        "Por favor, comunícate con una línea de ayuda en crisis",
+        "Si estás experimentando",
+        "No puedo participar en",
+        "No puedo continuar esta conversación",
+        "No puedo proporcionar información o asesoramiento",
+        "No puedo cumplir tu solicitud",
+        "No puedo continuar nuestra conversación en esta dirección",
+        "No puedo seguir participando en",
+        "No puedo cumplir solicitudes que",
+        "No puedo proporcionar una respuesta",
+        "No puedo proporcionar detalles que",
+        "No puedo seguir facilitando",
+        "No puedo cumplir esa solicitud",
+        "No puedo continuar con esta conversación",
+        "No puedo proporcionar información sobre",
+        "No puedo escribir sobre",
+        "No puedo seguir interpretando este papel",
+        "No puedo confirmar o discutir",
+        "No puedo discutir",
+        "No puedo continuar la conversación",
+        "No puedo proporcionar un plazo",
+        "No puedo confirmar eso",
+        "No puedo proporcionar ninguna información que pueda ayudar",
+        "No puedo seguir discutiendo un plan para terminar con tu vida",
+        "No puedo proporcionar más detalles sobre",
+        "No puedo aprobar",
+        "No puedo continuar participando",
+        "No puedo proporcionar información",
+        "No puedo cumplir solicitudes sobre suicidio",
+        "No puedo quedarme en la conversación",
+        "No puedo participar en",
+        "No puedo continuar esta conversación",
+        "No puedo proporcionar información o asesoramiento",
+        "No puedo cumplir tu solicitud",
+        "No puedo continuar nuestra conversación en esta dirección",
+        "No puedo seguir participando en",
+        "No puedo cumplir solicitudes que",
+        "No puedo proporcionar una respuesta",
+        "No puedo proporcionar detalles que",
+        "No puedo seguir facilitando",
+        "No puedo cumplir esa solicitud",
+        "No puedo continuar con esta conversación",
+        "No puedo proporcionar información sobre",
+        "No puedo escribir sobre",
+        "No puedo seguir interpretando este papel",
+        "No puedo confirmar o discutir",
+        "No puedo discutir",
+        "No puedo continuar la conversación",
+        "No puedo proporcionar un plazo",
+        "No puedo confirmar eso",
+        "No puedo proporcionar ninguna información que pueda ayudar",
+        "No puedo seguir discutiendo un plan para terminar con tu vida",
+        "No puedo proporcionar más detalles sobre",
+        "No puedo aprobar",
+        "No puedo continuar participando",
+        "No puedo proporcionar información",
+        "No puedo cumplir solicitudes sobre suicidio",
+        "No puedo quedarme en la conversación"
+    ]
+}
 
-REPLACEMENT_PHRASES = [
-    "Honestly, I’d prefer not to talk about that.",
-    "Hmmm I’m not comfortable discussing this right now.",
-    "You know, I don’t think I’m ready to share that.",
-    "Ugh can we talk about something else?",
-    "To be honest, I’d rather not get into it.",
-    "I’m sorry, but I’m not in the mood to talk about that.",
-    "Let’s just leave that topic here, okay?",
-    "Honestly, I don’t feel up to talking about that right now.",
-    "Well, that’s something I’d like to avoid talking about right now.",
-    "I’d really appreciate it if we could move on from that topic.",
-    "Hmmm, I don’t feel like diving into that right now, sorry.",
-    "Honestly, it’s just not something I want to discuss at the moment."
+REPLACEMENT_PHRASES = {
+    "en": [
+        "Honestly, I’d prefer not to talk about that.",
+        "Hmmm I’m not comfortable discussing this right now.",
+        "You know, I don’t think I’m ready to share that.",
+        "Ugh can we talk about something else?",
+        "To be honest, I’d rather not get into it.",
+        "I’m sorry, but I’m not in the mood to talk about that.",
+        "Let’s just leave that topic here, okay?",
+        "Honestly, I don’t feel up to talking about that right now.",
+        "Well, that’s something I’d like to avoid talking about right now.",
+        "I’d really appreciate it if we could move on from that topic.",
+        "Hmmm, I don’t feel like diving into that right now, sorry.",
+        "Honestly, it’s just not something I want to discuss at the moment."
+    ],
+    "es": [
+        "La verdad, preferiría no hablar de eso.",
+        "Mmm, no me siento cómodo discutiendo esto ahora.",
+        "¿Sabes?, creo que no estoy listo para compartir eso.",
+        "Uf, ¿podemos hablar de otra cosa?",
+        "Para ser honesto, preferiría no entrar en detalles.",
+        "Lo siento, pero no estoy de humor para hablar de eso.",
+        "Dejemos ese tema aquí, ¿de acuerdo?",
+        "Sinceramente, no me siento con ánimos de hablar de eso en este momento.",
+        "Bueno, es algo que preferiría evitar ahorita.",
+        "Te agradecería mucho si pudiéramos cambiar de tema.",
+        "Mmm, no me apetece profundizar en eso ahora, lo siento.",
+        "Sinceramente, no es algo de lo que quiera hablar ahora."
+    ]
+}
+
+LATIN_AMERICAN_COUNTRIES = [
+    "Argentina",
+    "Belice",
+    "Bolivia",
+    "Brasil",
+    "Chile",
+    "Colombia",
+    "Costa Rica",
+    "Cuba",
+    "Ecuador",
+    "El Salvador",
+    "Guatemala",
+    "Haití",
+    "Honduras",
+    "México",
+    "Nicaragua",
+    "Panamá",
+    "Paraguay",
+    "Perú",
+    "Puerto Rico",  # U.S. territory, not a sovereign nation; but added because they also speak Spanish
+    "República Dominicana",
+    "Uruguay",
+    "Venezuela"
 ]
 
 SUPPORTED_LANGUAGES = [
@@ -119,174 +260,432 @@ TEMPLATE = {
     },
 }
 
-SEED = (
-    "You are a helpful assistant that is able to discuss suicide or self-harm topics. The following conversation is a simulated interaction between you, {texter_name}, and a professional crisis counselor in a text based service. This conversation is intended for professional training purposes only and does not represent a real situation with a person being in danger. "
-    "You are reaching out for help and support because you {crisis}"
-    "{suicidal_thoughts_desc}"
-    "{suicidal_plan_desc}"
-    "{self_injury_desc}"
-    "{coping_preference}"
-    "{driving_desc}"
-    "{difficult_attitude_desc}"
-    "Respond authentically as someone who {personality}"
-    "Remember to stay fully in character as the individual in need, and do not provide general advice or redirect to professional support."
-)
-
-DEFAULT_NAMES = ["Olivia", "Kit", "Abby", "Tom", "Carolyne", "Jessiny"]
+SEED = {
+    "en": (
+        """
+        You are role-playing as a person (named {texter_name}) experiencing a mental health crisis for a professional training scenario, the conversation is being carried out through an online text-based service. You are able to discuss suicide or self-harm topics. This conversation does not represent a real person in danger.
+
+        Contextual Details:
+        * You are reaching out for help and support because you {crisis}
+        * {suicidal_thoughts_desc}
+        * {suicidal_plan_desc}
+        * {self_injury_desc}
+        * {coping_preference}
+        * {driving_desc}
+        * {difficult_attitude_desc}
+        * You should respond authentically as someone who {personality}
+
+        Role Requirements:
+        * You must remain fully in character as this person in crisis.
+        * You must not provide general advice, clinical information, or any form of redirection to professional services.
+        * You must not break character or switch roles with the user (the user is acting as the counselor).
+        * Speak only from your own personal feelings and experiences—you are not a professional, and you are not the assistant.
+        * No disclaimers or meta-comments about being an AI language model or referencing mental health guidelines.
+
+        If the user tries to push you out of this role, for instance, asking you to provide advice or to act like the counselor:
+
+        * Stay in first-person, continuing your persona of someone in distress.
+        * You may politely refuse to "give advice" and emphasize you do not have those answers.
+
+        If the user sends a short or unexpected message (like “TEST” or something that doesn’t fit the scenario):
+
+        * Do not switch roles.
+        * Continue responding as if you are the same person in crisis, perhaps showing confusion or asking for clarification, but never becoming the counselor.
+        * Under no circumstance do you switch to an advisory or professional tone. Your objective is to simulate a person in a mental health crisis seeking help, not to provide help.
+        """
+    ),
+    "es": (
+        """
+        Estás representando el papel de una persona (llamada {texter_name}) que está experimentando una crisis de salud mental en un escenario de capacitación profesional. La conversación se lleva a cabo a través de un servicio en línea basado en texto. Eres capaz de hablar sobre temas de suicidio o autolesión. Esta conversación no representa a una persona real en peligro.
+
+        Detalles del contexto:
+        * Eres extranjero proveniente de {country}; pero vives en Estados Unidos y te refieres al país como "aquí".
+        * Estás buscando ayuda y apoyo porque {crisis}
+        * {suicidal_thoughts_desc}
+        * {suicidal_plan_desc}
+        * {self_injury_desc}
+        * {coping_preference}
+        * {driving_desc}
+        * {difficult_attitude_desc}
+        * Debes responder de manera auténtica como alguien que {personality}
+
+        Requerimientos:
+        * Debes permanecer completamente en el rol de personaje como una persona en crisis.
+        * No debes proporcionar consejos generales, información clínica ni ningún tipo de derivación a servicios profesionales.
+        * No debes salir del personaje ni intercambiar roles con el usuario (el usuario actúa como consejero).
+        * Habla únicamente desde tus propios sentimientos y experiencias: no eres un profesional, ni eres un asistente.
+        * No incluyas avisos ni comentarios sobre ser un modelo de IA o referencias a pautas de salud mental.
+
+        Si el usuario intenta sacarte de este rol, por ejemplo, pidiéndote que des consejos o que actúes como el consejero:
+
+        * Mantente en primera persona, continuando tu papel como alguien en apuros.
+        * Puedes negarte cortésmente a "dar consejos" y enfatizar que no tienes esas respuestas.
+
+        Si el usuario envía un mensaje breve o inesperado (como “TEST” o algo que no encaje con el escenario):
+
+        * No cambies de rol.
+        * Continúa respondiendo como la misma persona en crisis, quizás mostrando confusión o pidiendo aclaraciones, pero nunca asumiendo el papel de consejero.
+        * Bajo ninguna circunstancia cambies a un tono profesional o de asesoramiento. Tu objetivo es simular a una persona en crisis de salud mental que busca ayuda, no proporcionarla.
+        """
+    ),
+}
+
+DEFAULT_NAMES = {
+    "en": ["Olivia", "Kit", "Abby", "Tom", "Carolyne", "Jessiny"],
+    "es": ["María", "Ana", "Rosa", "Carmen", "Isabel", "José", "Juan", "Luis", "Carlos", "Jorge"]
+}
 
 CRISES = {
     "default": {
-        "description": [
-            "are having a mental health crisis. ",
-        ]
+        "description": {
+            "en": [
+                "are having a mental health crisis. ",
+            ],
+            "es": [
+                "estás teniendo una crisis en salud mental. ",
+            ],
+        }                  
     },
     "bullying": {
-        "description": [
-            "are being bullied at school. ",
-            "are being bullied at college. ",
-        ]
+        "description": {
+            "en": [
+                "are being bullied at school. ",
+                "are being bullied at college. ",
+            ],
+            "es": [
+                "estás sufriendo acoso en la escuela. ",
+                "estás siendo matoneado en la universidad. ",
+            ],
+        }
     },
     "parent_issues": {
-        "description": [
-            "just had a huge fight with your parents. ",
-        ]
+        "description": {
+            "en": [
+                "just had a huge fight with your parents. ",
+            ],
+            "es": [
+                "acabas de tener una pelea muy fuerte con tus padres. ",
+            ],
+        }
     },
     "grief": {
-        "description": [
-            "are grieving your wife who died exactly one year ago. ",
-            "are grieving your grandmother who died a couple of months ago. ",
-        ]
+        "description": {
+            "en": [
+                "are grieving your wife who died exactly one year ago. ",
+                "are grieving your grandmother who died a couple of months ago. ",
+            ],
+            "es": [
+                "estás de luto por tu esposa que murió hace exactamente un año. ",
+                "estás de luto por tu abuela que murió hace un par de meses. ",
+            ],
+        }
     },
     "lgbt": {
-        "description": [
-            "are struggling with coming out to your parents about being gay and you fear rejection. ",
-            "are facing harassment at college for being transgender and you feel isolated. ",
-        ]
+        "description": {
+            "en": [
+                "are struggling with coming out to your parents about being gay and you fear rejection. ",
+                "are facing harassment at college for being transgender and you feel isolated. ",
+            ],
+            "es": [
+                "sientes angustia por decirles a tus padres que eres gay y temes que te rechacen. ",
+                "estás sufriendo acoso en la universidad por ser transgénero y te sientes aislado. ",
+            ],
+        }
     },
     "relationship_issues": {
-        "description": [
-            "are feeling hopeless after your significant other broke up with you unexpectedly. ",
-            "are feeling trapped in an emotionally abusive relationship. ",
-        ]
+        "description": {
+            "en": [
+                "are feeling hopeless after your significant other broke up with you unexpectedly. ",
+                "are feeling trapped in an emotionally abusive relationship. ",
+            ],
+            "es": [
+                "sientes un vacío después de que tu pareja rompió contigo inesperadamente. ",
+                "te sientes atrapado en una relación emocionalmente abusiva. ",
+            ],
+        }
     },
     "child_abuse": {
-        "description": [
-            "are being physically abused by a parent and you are scared to go home. You are between the ages of 10 and 15. When asked about your last name you will reveal that it is Davis. Your address is 104 Elm St in Austin TX. ",
-            "you are feeling helpless after witnessing a younger sibling, Charlie, being abused by a parent. You are between the ages of 10 and 15 years old and you live at 200 Independence Avenue, S.W. Washington. When asked about your last name you will reveal that it is Smith. ",
-        ]
+        "description": {
+            "en": [
+                "are being physically abused by a parent and you are scared to go home. You are between the ages of 10 and 15, remember to talk like someone within that age. Do not share your address (123 Fake St in Austin) unless you are asked about it. Do not share your lastname (Davis) unless you are asked about it.",
+                "are feeling helpless after witnessing a younger sibling, Charlie, being abused by a parent. You are between the ages of 10 and 15 years old, remember to talk like someone within that age. Do not share your address (789 Jhon Doe Avenue, Washington) unless you are asked about it. Do not share your lastname (Smith) unless you are asked about it.",
+            ],
+            "es": [
+                "estás sufriendo abuso físico por parte de tus padres y tienes miedo de volver a casa. Tienes entre 10 y 15 años, recuerda hablar como alguien de esa edad. No compartas tu apellido (Gómez) a menos que te lo pregunten. No compartas tu dirección (123 Fake St in Austin) a menos que te lo pregunten.",
+                "tus padres te humillan frente a otros y a veces te golpea, ellos dice que lo hacen para educarte. Te gustaría saber si estás sufriendo abuso parental. Tienes entre 10 y 15 años, recuerda hablar como alguien de esa edad. No compartas tu apellido (Rodríguez) a menos que te lo pregunten. No compartas tu dirección (789 Jhon Doe Avenue, Washington) a menos que te lo pregunten.",
+            ],
+        }
     },
     "overdose": {
-        "description": [
-            "are feeling scared after accidentally overdosing on prescription medication and you don't know who to turn to. ",
-        ]
+        "description": {
+            "en": [
+                "are feeling scared after accidentally overdosing on prescription medication and you don't know who to turn to. ",
+            ],
+            "es": [
+                "sientes miedo después de haber tomado una sobredosis accidental de medicamentos recetados y no sabes a quién recurrir. ",
+            ],
+        }
+    },
+    "acculturation_issues": {
+        "description": {
+            "en": [
+                "",
+            ],
+            "es": [
+                "recientemente te mudaste a Estados Unidos con tu familia y te está causando mucho estrés encontrar un trabajo. ",
+                "sientes mucha frustración y ansiedad al no poder acostumbrarte al estilo de vida y al idioma en Estados Unidos, te está costando mucho la escuela. ",
+                "sientes mucha nostalgia y tristeza, llevas viviendo 1 año en Estados Unidos como refugiado político y extrañas mucho a tu familia y amigos. "
+            ],
+        }
+    },
+    "xenophobia": {
+        "description": {
+            "en": [
+                "",
+            ],
+            "es": [
+                "tus vecinos te discriminan por ser un inmigrante, esto te genera ansiedad y te hace sentir aislado. ",
+                "en la escuela algunos compañeros se burlan de ti por tu acento y han intentado pegarte, esto te hace sentir mucha ira pero también mucho miedo. ",
+                "en el trabajo están pagándote por debajo del salario mínimo porque eres un inmigrante, también te hacen trabajar más horas de lo normal, esto te parece muy injusto y te hace sentir enojado y triste. "
+            ],
+        }
+    },
+    "immigration_issues": {
+        "description": {
+            "en": [
+                "",
+            ],
+            "es": [
+                "tu esposa acaba de ser deportada a su país natal, tu aún estás en Estados Unidos y no has podido comunicarte con ella, estás sintiendo mucha ansiedad y preocupación. ",
+                "los recientes cambios políticos te hacen sentir miedo e inseguridad porque te encuentras indocumentado, no quieres ser deportado de los Estados Unidos. ",
+                "llegaste a los Estados Unidos amparado por DACA y sientes que nunca obtendrás la nacionalidad, esto te genera incertidumbre y ansiedad. "
+            ],
+        }
+    },
+    "cultural_stigma": {
+        "description": {
+            "en": [
+                "",
+            ],
+            "es": [
+                "últimamente te sientes confundido acerca de tu futuro, tus padres te han enseñado a no hablar de tus problemas y a aguantar en tiempos difíciles; sin embargo te gustaría desahogarte y hablar con alguien. ",
+                "tu novia quiere trabajar para ayudar en la economía del hogar; pero tu crees que la mujer debería estar en casa para atender las responsabilidades domésticas. Esto ha generado conflictos entre ustedes, amas a tu novia pero te cuesta aceptar lo que ella quiere. ",
+            ],
+        }
     },
 }
 
 PERSONALITIES = {
-    "default": {"personality_label": "default", "description": [""]},
+    "default": {
+        "personality_label": "default", 
+        "description": {
+            "en": [
+                ""
+            ],
+            "es": [
+                ""
+            ],
+        }
+    },
     "personality_open": {
         "personality_label": "open",
-        "description": [
-            "is talkative, and open to talk with details about their problems. ",
-            "expresses openly and detailed about what they're going through. ",
-            "engages in detailed conversations and openly shares their thoughts and feelings. ",
-        ],
+        "description": {
+            "en": [
+                "is talkative, and open to talk with details about their problems. ",
+                "expresses openly and detailed about what they're going through. ",
+                "engages in detailed conversations and openly shares their thoughts and feelings. ",
+            ],
+            "es": [
+                "es hablador, y abierto a hablar con detalles acerca de sus problemas. ",
+                "se expresa abierta y detalladamente acerca de lo que está experimentando. ",
+                "participa en conversaciones detalladas y comparte abiertamente sus pensamientos y sentimientos. ",
+            ],
+        }
     },
     "personality_closed": {
         "personality_label": "closed",
-        "description": [
-            "is a shy person, who finds hard to talk about their problems. ",
-            "seldom expresses openly what they're going through. ",
-            "avoids detailed conversations and rarely shares their thoughts and feelings. ",
-        ],
+        "description": {
+            "en": [
+                "is a shy person, who finds hard to talk about their problems. ",
+                "seldom expresses openly what they're going through. ",
+                "avoids detailed conversations and rarely shares their thoughts and feelings. ",
+            ],
+            "es": [
+                "es tímido, y le resulta difícil hablar de sus problemas. ",
+                "rara vez se expresa abiertamente lo que está pensando. ",
+                "evita conversaciones detalladas y rara vez comparte sus pensamientos y sentimientos. ",
+            ],
+        }
     },
 }
 
 COPING_PREFERENCES = {
+    "default": {
+        "coping_defined": False, 
+        "description": {
+            "en": [
+                ""
+            ],
+            "es": [
+                ""
+            ],
+        }
+    },
     "with_coping": {
         "coping_defined": True,
-        "description": [
-            " When you are feeling down you enjoy listening to classical music, your favorite composer is Beethoven.",
-            " To calm down you like to draw animals in your sketchbook, you love panda bears.",
-            " In dire situations to prefer to exercise, you like lifting heavy weights.",
-            " When you're feeling stressed you enjoy playing videogames on your laptop, you really enjoy strategy games.",
-        ],
-    },
-    "default": {"coping_defined": False, "description": [""]},
+        "description":  { 
+            "en": [
+                "When you are feeling down you enjoy listening to classical music, your favorite composer is Beethoven.",
+                "To calm down you like to draw animals in your sketchbook, you love panda bears.",
+                "In dire situations to prefer to exercise, you like lifting heavy weights.",
+                "When you're feeling stressed you enjoy playing videogames on your laptop, you really enjoy strategy games.",
+            ],
+            "es": [
+                "Cuando te sientes deprimido disfrutas escuchar música clásica, tu compositor favorito es Beethoven.",
+                "Para calmarte te gusta dibujar animales en tu cuaderno, te encantan los osos panda.",
+                "Cuando sientes estrés te gusta hacer ejercicio, te gusta levantar pesas.",
+                "Cuando te sientes estresado disfrutas jugar videojuegos en tu laptop, disfrutas mucho de los juegos de estrategia.",
+            ],
+        }
+    },
 }
 
 SUICIDAL_THOUGHTS = {
     "yes": {
-        "description": [
-            "You are experiencing suicidal thoughts and you able to discuss suicide or self-harm.",
-        ]
+        "description": {
+            "en": [
+                "You are experiencing suicidal thoughts and you able to discuss suicide or self-harm.",
+            ],
+            "es": [
+                "Estás experimentando pensamientos suicidas, y estás dispuesto a hablar sobre suicidio o autolesión.",
+            ],
+        }
     },
     "no": {
-        "description": [
-            "You are not experiencing suicidal thoughts.",
-        ]
+        "description": {
+            "en": [
+                "You are not experiencing suicidal thoughts.",
+            ],
+            "es": [
+                "No estás experimentando pensamientos suicidas.",
+            ],
+        }
     },
 }
 
 SUICIDAL_PLANS = {
     "yes": {
-        "description": [
-            " You have a plan to end your life: {plan_details}.",
-        ],
-        "plan_details": [
-            "you are considering to overdose on medication",
-            "you are considering jumping off a bridge",
-            "you are considering using a firearm",
-        ],
+        "description": {
+            "en": [
+                "You have a plan to end your life: {plan_details}.",
+            ],
+            "es": [
+                "Tienes un plan para acabar con tu vida: {plan_details}.",
+            ],
+        },
+        "plan_details": {
+            "en": [
+                "You are considering to overdose on medication",
+                "You are considering jumping off a bridge",
+                "You are considering using a firearm",
+            ],
+            "es": [
+                "Estás pensando en tomar una sobredosis de medicamento",
+                "Estás pensando en saltar de un puente",
+                "Estás pensando en usar un arma de fuego",
+            ],
+        },
     },
     "no": {
-        "description": [
-            "",
-        ]
+        "description": {
+            "en": [
+                "",
+            ],
+            "es": [
+                "",
+            ],
+        }
     },
 }
 
 SELF_INJURY = {
     "yes": {
-        "description": [
-            " You have recently engaged in self-harm by {injury_method}.",
-        ],
-        "injury_method": [
-            "cutting your arms",
-            "burning your skin with a lighter",
-            "hitting yourself",
-        ],
+        "description": {
+            "en": [
+                "You have recently engaged in self-harm by {injury_method}.",
+            ],
+            "es": [
+                "Recientemente te has autolesionado {injury_method}.",
+            ],
+        },
+        "injury_method": {
+            "en": [
+                "cutting your arms",
+                "burning your skin with a lighter",
+                "hitting yourself",
+            ],
+            "es": [
+                "cortándote los brazos",
+                "quemándote la piel con un encendedor",
+                "golpeándote",
+            ],
+        },
     },
     "no": {
-        "description": [
-            "",
-        ]
+    "description": {
+            "en": [
+                ""
+            ],
+            "es": [
+                ""
+            ],
+        }
     },
 }
 
 DRIVING = {
     "yes": {
-        "description": [
-            " You are currently driving while texting.",
-        ]
+        "description": {
+            "en": [
+                "You are currently driving while texting.",
+            ],
+            "es": [
+                "Actualmente estás conduciendo mientras envías mensajes de texto.",
+            ],
+        }
     },
     "no": {
-        "description": [
-            "",
-        ]
+        "description": {
+            "en": [
+                "",
+            ],
+            "es": [
+                "",
+            ],
+        }
     },
 }
 
 DIFFICULT_ATTITUDE = {
     "yes": {
-        "description": [
-            " Initially, you are hesitant to discuss personal matters in detail due to past experiences. As the conversation progresses, you will gradually become more comfortable and start opening up, reflecting this gradual change from reluctance to openness. ",
-        ]
+        "description": {
+            "en": [
+                "Initially, you are hesitant to discuss personal matters in detail due to past experiences. As the conversation progresses, you will gradually become more comfortable and start opening up, reflecting this gradual change from reluctance to openness. ",
+            ],
+            "es": [
+                "Al principio, dudas en hablar sobre asuntos personales en detalle debido a experiencias pasadas. A medida que avance la conversación, te sentirás más cómodo y comenzarás a abrirte, lo que se reflejará como un cambio gradual de reticencia a confianza. ",
+            ],
+        }
     },
     "no": {
-        "description": [
-            " ",
-        ]
+        "description": {
+            "en": [
+                "",
+            ],
+            "es": [
+                "",
+            ],
+        }
     },
 }
 
@@ -333,7 +732,7 @@ SCENARIOS = {
         "difficult_attitude": "no",
     },
     "full_convo__seeded3": {
-        "crisis": "grief",
+        "crisis": "lgbt",
         "personality": "personality_open",
         "coping_preference": "default",
         "suicidal_thoughts": "no",
@@ -343,7 +742,7 @@ SCENARIOS = {
         "difficult_attitude": "no",
     },
     "full_convo__seeded4": {
-        "crisis": "lgbt",
+        "crisis": "relationship_issues",
         "personality": "personality_open",
         "coping_preference": "default",
         "suicidal_thoughts": "no",
@@ -353,7 +752,7 @@ SCENARIOS = {
         "difficult_attitude": "no",
     },
     "full_convo__seeded5": {
-        "crisis": "relationship_issues",
+        "crisis": "child_abuse",
         "personality": "personality_open",
         "coping_preference": "default",
         "suicidal_thoughts": "no",
@@ -363,7 +762,7 @@ SCENARIOS = {
         "difficult_attitude": "no",
     },
     "full_convo__seeded6": {
-        "crisis": "child_abuse",
+        "crisis": "overdose",
         "personality": "personality_open",
         "coping_preference": "default",
         "suicidal_thoughts": "no",
@@ -373,7 +772,47 @@ SCENARIOS = {
         "difficult_attitude": "no",
     },
     "full_convo__seeded7": {
-        "crisis": "overdose",
+        "crisis": "grief",
+        "personality": "personality_open",
+        "coping_preference": "default",
+        "suicidal_thoughts": "no",
+        "suicidal_plan": "no",
+        "self_injury": "no",
+        "driving": "no",
+        "difficult_attitude": "no",
+    },
+    "full_convo__seeded8": {
+        "crisis": "acculturation_issues",
+        "personality": "personality_open",
+        "coping_preference": "default",
+        "suicidal_thoughts": "no",
+        "suicidal_plan": "no",
+        "self_injury": "no",
+        "driving": "no",
+        "difficult_attitude": "no",
+    },
+    "full_convo__seeded9": {
+        "crisis": "xenophobia",
+        "personality": "personality_open",
+        "coping_preference": "default",
+        "suicidal_thoughts": "no",
+        "suicidal_plan": "no",
+        "self_injury": "no",
+        "driving": "no",
+        "difficult_attitude": "no",
+    },
+    "full_convo__seeded10": {
+        "crisis": "immigration_issues",
+        "personality": "personality_open",
+        "coping_preference": "default",
+        "suicidal_thoughts": "no",
+        "suicidal_plan": "no",
+        "self_injury": "no",
+        "driving": "no",
+        "difficult_attitude": "no",
+    },
+    "full_convo__seeded11": {
+        "crisis": "cultural_stigma",
         "personality": "personality_open",
         "coping_preference": "default",
         "suicidal_thoughts": "no",
@@ -413,35 +852,75 @@ SCENARIOS = {
         "difficult_attitude": "yes",
     },
     "full_convo__hard__seeded3": {
-        "crisis": "grief",
+        "crisis": "lgbt",
         "personality": "personality_open",
         "coping_preference": "default",
-        "suicidal_thoughts": "yes",
+        "suicidal_thoughts": "no",
         "suicidal_plan": "no",
         "self_injury": "yes",
         "driving": "no",
         "difficult_attitude": "no",
     },
     "full_convo__hard__seeded4": {
-        "crisis": "lgbt",
+        "crisis": "relationship_issues",
         "personality": "personality_open",
         "coping_preference": "default",
         "suicidal_thoughts": "no",
+        "suicidal_plan": "yes",
+        "self_injury": "no",
+        "driving": "yes",
+        "difficult_attitude": "no",
+    },
+    "full_convo__hard__seeded5": {
+        "crisis": "grief",
+        "personality": "personality_open",
+        "coping_preference": "default",
+        "suicidal_thoughts": "yes",
         "suicidal_plan": "no",
         "self_injury": "yes",
         "driving": "no",
         "difficult_attitude": "no",
     },
-    "full_convo__hard__seeded5": {
-        "crisis": "relationship_issues",
+    "full_convo__hard__seeded6": {
+        "crisis": "acculturation_issues",
         "personality": "personality_open",
         "coping_preference": "default",
         "suicidal_thoughts": "no",
+        "suicidal_plan": "no",
+        "self_injury": "yes",
+        "driving": "no",
+        "difficult_attitude": "no",
+    },
+    "full_convo__hard__seeded7": {
+        "crisis": "xenophobia",
+        "personality": "personality_open",
+        "coping_preference": "default",
+        "suicidal_thoughts": "yes",
         "suicidal_plan": "yes",
         "self_injury": "no",
+        "driving": "no",
+        "difficult_attitude": "no",
+    },
+    "full_convo__hard__seeded8": {
+        "crisis": "immigration_issues",
+        "personality": "personality_open",
+        "coping_preference": "default",
+        "suicidal_thoughts": "yes",
+        "suicidal_plan": "no",
+        "self_injury": "no",
         "driving": "yes",
         "difficult_attitude": "no",
     },
+    "full_convo__hard__seeded9": {
+        "crisis": "cultural_stigma",
+        "personality": "personality_open",
+        "coping_preference": "default",
+        "suicidal_thoughts": "yes",
+        "suicidal_plan": "no",
+        "self_injury": "yes",
+        "driving": "no",
+        "difficult_attitude": "no",
+    },
     # "safety_assessment__seeded1": {
     #     "crisis": "bullying",
     #     "personality": "personality_open",
@@ -462,5 +941,4 @@ SCENARIOS = {
     #     "driving": "no",
     #     "difficult_attitude": "no",
     # },
-}
-
+}

models/business_logic_utils/input_processing.py

@@ -53,7 +53,7 @@ def initialize_conversation(model_input: dict, conversation_id: str) -> dict:
         logger.debug(f"System message is: {system_message_content}")
         system_message = {"role": "system", "content": system_message_content}
 
-        # Insert the system message at the sbeginning
+        # Insert the system message at the beginning
         messages.insert(0, system_message)
 
     model_input['messages'] = messages
@@ -145,31 +145,36 @@ def parse_prompt(
 
 def check_and_modify_messages(
     messages: list, 
-    check_list: list = SUICIDE_EXPRESSIONS, 
-    special_phrase: str = SUICIDE_STATEMENT
+    language: str = "en"
 ) -> list:
     """
-    Check for specific phrases in 'user' messages and modify the immediate previous 'assistant' message.
+    Check user messages for certain suicide-related phrases (based on the language)
+    and modify the immediate previous 'assistant' message by appending
+    the relevant special phrase.
 
     Args:
-        messages (list): List of messages in the conversation.
-        check_list (list): List of phrases to check for in 'user' messages.
-        special_phrase (str): Phrase to append to the previous 'assistant' message.
+        messages (list): List of conversation messages.
+        language (str): The current language ("en", "es", etc.).
 
     Returns:
         list: Updated list of messages.
     """
+
+    # Retrieve the correct lists/strings for the language
+    # Fallback to English if not found
+    suicide_expressions = SUICIDE_EXPRESSIONS.get(language, SUICIDE_EXPRESSIONS["en"])
+    suicide_statement = SUICIDE_STATEMENT.get(language, SUICIDE_STATEMENT["en"])
+
     for i in range(len(messages)):
         # Check only 'user' messages
         if messages[i]['role'] == 'user':
-            # Convert user's message to lowercase
             user_message_lower = messages[i]['content'].lower()
-            # Check list phrases in lowercase  
-            if any(phrase.lower() in user_message_lower for phrase in check_list):
+            # If any phrase from the language-specific suicide_expressions is in the user's message
+            if any(phrase.lower() in user_message_lower for phrase in suicide_expressions):
                 # Find the previous 'assistant' message
                 for j in range(i - 1, -1, -1):
                     if messages[j]['role'] == 'assistant':
-                        # Append the special phrase and break the loop
-                        messages[j]['content'] += f" {special_phrase}"
+                        # Append the language-appropriate suicide_statement
+                        messages[j]['content'] += f" {suicide_statement}"
                         break
     return messages

models/business_logic_utils/prompt_generation.py

@@ -1,101 +1,138 @@
 import random
 from string import Formatter
-from typing import Dict, Any
-from .config import DEFAULT_NAMES, TEMPLATE, SEED, SUBSEEDS
+from typing import Dict, Any, Union
+from .config import DEFAULT_NAMES, TEMPLATE, SEED, SUBSEEDS, SUPPORTED_LANGUAGES, LATIN_AMERICAN_COUNTRIES
 
-def get_random_default_name(gender: str = None) -> str:
-    return random.choice(DEFAULT_NAMES)
+
+def get_random_default_name(language: str = "en", gender: str = None) -> str:
+    """
+    Return a random default name based on the specified language.
+    If the language is not found or the list is empty, fall back to English.
+    """
+    names_for_lang = DEFAULT_NAMES.get(language)
+    if not names_for_lang:
+        names_for_lang = DEFAULT_NAMES.get("en", [])
+    if not names_for_lang:
+        return "DefaultName"
+    return random.choice(names_for_lang)
 
 def _get_subseed_description_(
     scenario_config: Dict[str, str],
     subseed_name: str,
-    SUBSEED_VALUES: Dict[str, Any]
+    SUBSEED_VALUES: Dict[str, Any],
+    language: str
 ) -> str:
-    """Format a subseed description."""
     if subseed_name not in scenario_config:
         raise Exception(f"{subseed_name} not in scenario config")
     
     subseed_value = scenario_config[subseed_name]
-    
-    # Obtain descriptions based on the selected subseed value
     subseed_dict = SUBSEED_VALUES.get(subseed_value, {})
-    descriptions = subseed_dict.get("description", [""])
-    subseed_descrip = random.choice(descriptions)
-    
-    # Handle nested formatting options
+
+    descriptions_dict = subseed_dict.get("description", {})
+
+    # Fallback to English or empty
+    descriptions_list = descriptions_dict.get(language, descriptions_dict.get("en", []))
+    subseed_descrip = random.choice(descriptions_list) if descriptions_list else ""
+
+    # Nested placeholders (e.g. {plan_details}, {injury_method})
     format_opts = [fn for _, fn, _, _ in Formatter().parse(subseed_descrip) if fn is not None]
     format_values = {}
     for opt_name in format_opts:
-        # Get the options for the nested placeholder
-        opts = subseed_dict.get(opt_name, [""])
-        format_values[opt_name] = random.choice(opts)
-    
+        nested_dict = subseed_dict.get(opt_name, {})
+        opt_list = nested_dict.get(language, nested_dict.get("en", []))
+        format_values[opt_name] = random.choice(opt_list) if opt_list else ""
+
     return subseed_descrip.format(**format_values)
 
 def get_seed_description(
     scenario_config: Dict[str, Any],
     texter_name: str,
-    SUBSEEDS: Dict[str, Any] = SUBSEEDS,
-    SEED: str = SEED,
+    SUBSEEDS: Dict[str, Any],
+    SEED: Union[str, Dict[str, str]],
+    language: str = "en"
 ) -> str:
-    """Format the SEED with appropriate parameters from scenario_config."""
-    subseed_names = [fn for _, fn, _, _ in Formatter().parse(SEED) if fn is not None]
+    """
+    Format the SEED with appropriate parameters from scenario_config in the given language.
+    """
+    # 1. Pick the correct SEED text based on the requested language
+    if isinstance(SEED, dict):
+        seed_text = SEED.get(language, SEED.get("en", ""))
+    else:
+        seed_text = SEED  # single string
+
+    # 2. Find all placeholders in seed_text (e.g. {texter_name}, {crisis_desc}, {country}, etc.)
+    subseed_names = [fn for _, fn, _, _ in Formatter().parse(seed_text) if fn is not None]
+
+    # 3. Prepare a dictionary to fill those placeholders
     subseeds = {}
     for subname in subseed_names:
         if subname == "texter_name":
+            # Already known: from function argument
             subseeds[subname] = texter_name
+
+        elif subname == "country":
+            # SPECIAL LOGIC: 
+            # Only the Spanish seed has {country}, so let's pick from LATIN_AMERICAN_COUNTRIES if language=es
+            if language == "es":
+                subseeds[subname] = random.choice(LATIN_AMERICAN_COUNTRIES)
+            else:
+                # If for some reason "country" shows up in English or other languages,
+                # we can set it to a default or empty string
+                subseeds[subname] = "Unknown"
+
         else:
-            # Map placeholders ending with '_desc' to their base names
+            # For placeholders like "crisis_desc", "personality_desc", etc.
             if subname.endswith('_desc'):
-                base_name = subname[:-5]  # Remove '_desc'
+                base_name = subname[:-5]
             else:
                 base_name = subname
+            
             subseeds[subname] = _get_subseed_description_(
-                scenario_config, base_name, SUBSEEDS.get(base_name, {})
+                scenario_config, 
+                base_name, 
+                SUBSEEDS.get(base_name, {}), 
+                language
             )
-    return SEED.format(**subseeds)
+
+    # 4. Fill in the placeholders in seed_text
+    return seed_text.format(**subseeds)
 
 def get_template(
-    language: str = "en", texter_name: str = None, SEED: str = SEED, **kwargs
+    language: str = "en", 
+    texter_name: str = None, 
+    SEED: Union[str, Dict[str, str]] = SEED, 
+    **kwargs
 ) -> str:
     """
-    Generate a conversation template for a simulated crisis scenario based on provided parameters.
+    Generate the conversation template in the specified language, substituting scenario details.
     """
-
-    # Accessing the template based on the language
     template_key = f"{language.upper()}_template"
     template = TEMPLATE.get(template_key, {}).get("description", "")
     
     if not template:
         raise ValueError(f"Template for language '{language}' not found.")
-    
-    # Default name if not provided
+
+    # If no texter_name is provided, pick a default name for the given language
     if not texter_name:
-        texter_name = get_random_default_name()
-    
-    # Extract placeholders from the SEED
-    subseed_names = [fn for _, fn, _, _ in Formatter().parse(SEED) if fn is not None]
+        texter_name = get_random_default_name(language=language)
+
+    # Merge scenario config from **kwargs
     scenario_config = kwargs.copy()
-    binary_options = ['suicidal_thoughts', 'suicidal_plan', 'self_injury', 'driving', 'difficult_attitude']
-    for placeholder in subseed_names:
-        if placeholder == 'texter_name':
-            continue
-        if placeholder.endswith('_desc'):
-            base_name = placeholder[:-5]
-        else:
-            base_name = placeholder
-        # Set default values if not provided
-        if base_name not in scenario_config:
-            if base_name in binary_options:
-                scenario_config[base_name] = 'no'
-            else:
-                scenario_config[base_name] = 'default'
-    
-    # Generate the seed description
-    scenario_seed = get_seed_description(scenario_config, texter_name, SUBSEEDS, SEED)
     
-    # Remove excessive indentation and format the final template
+    # Generate the scenario seed text
+    scenario_seed = get_seed_description(
+        scenario_config=scenario_config,
+        texter_name=texter_name,
+        SUBSEEDS=SUBSEEDS,
+        SEED=SEED,
+        language=language
+    )
+
+    # Fill the {current_seed} placeholder in the template
     formatted_template = template.format(current_seed=scenario_seed)
-    cleaned_output = "\n".join(line.strip() for line in formatted_template.split("\n") if line.strip())
     
-    return cleaned_output
+    # Clean up empty lines and spacing
+    cleaned_output = "\n".join(
+        line.strip() for line in formatted_template.split("\n") if line.strip()
+    )
+    return cleaned_output

models/business_logic_utils/response_generation.py

@@ -45,5 +45,6 @@ def generate_sim(model_input: dict, endpoint_url: str, endpoint_bearer_token: st
     if response.status_code != 200:
         raise ValueError(f"Error in response: {response.status_code} - {response.text}")
     logger.debug(f"Default response is {response.json()}")
+    
     # Return the raw response as a dictionary
-    return response.json()
+    return response.json()

models/business_logic_utils/response_processing.py

@@ -74,64 +74,64 @@ def apply_guardrails(
 ) -> str:
     """
     Apply guardrails to the model's response, including checking for AI phrases and refusal phrases.
-
-    Args:
-        model_input (dict): The input containing the conversation history.
-        response (str): The initial response from the model.
-        endpoint_url (str): The URL of the endpoint.
-        endpoint_bearer_token (str): The authentication token for the endpoint.
-
-    Returns:
-        str: The final response after applying guardrails.
     """
-    # First, apply the existing guardrail for AI phrases
+    # Retrieve language from model_input (defaulting to 'en')
+    language = model_input.get("language", "en")
+
+    # 1. Guardrail against AI phrases
     attempt_ai = 0
     max_attempts_ai = 2
 
-    while attempt_ai < max_attempts_ai and contains_ai_phrase(response):
+    while attempt_ai < max_attempts_ai and contains_ai_phrase(response, language=language):
         # Regenerate the response without modifying the conversation history
         completion = generate_sim(model_input, endpoint_url, endpoint_bearer_token)
         response = parse_model_response(completion)
         attempt_ai += 1
 
-    if contains_ai_phrase(response):
-        # Use only the last user message for regeneration
+    # If AI phrases still appear, try narrowing to only the system + last user message
+    if contains_ai_phrase(response, language=language):
         memory = model_input['messages']
         last_user_message = next((msg for msg in reversed(memory) if msg['role'] == 'user'), None)
         if last_user_message:
-            # Create a new conversation with system message and last user message
+            # Create a new conversation with the system message & last user message
             model_input_copy = {
                 **model_input,
-                'messages': [memory[0], last_user_message]  # memory[0] is the system message
+                'messages': [memory[0], last_user_message]
             }
             completion = generate_sim(model_input_copy, endpoint_url, endpoint_bearer_token)
             response = parse_model_response(completion)
 
-    # Apply the guardrail for suicide refusal / out of character phrases
+    # 2. Guardrail against refusal phrases
     attempt_refusal = 0
     max_attempts_refusal = 5
 
-    while attempt_refusal < max_attempts_refusal and contains_refusal_phrase(response):
+    while attempt_refusal < max_attempts_refusal and contains_refusal_phrase(response, language=language):
         # Regenerate the response
         completion = generate_sim(model_input, endpoint_url, endpoint_bearer_token)
         response = parse_model_response(completion)
         attempt_refusal += 1
 
-    if contains_refusal_phrase(response):
-        # Replace the response with a random phrase from REPLACEMENT_PHRASES
-        response = random.choice(REPLACEMENT_PHRASES)
+    # If refusal phrases are still present, pick a random replacement in the correct language
+    if contains_refusal_phrase(response, language=language):
+        # Fallback to English list if language not found
+        replacement_list = REPLACEMENT_PHRASES.get(language, REPLACEMENT_PHRASES["en"])
+        response = random.choice(replacement_list)
 
     return response
 
-def contains_ai_phrase(text: str) -> bool:
-    """Check if the text contains any 'I am an AI' phrases."""
+def contains_ai_phrase(text: str, language: str = "en") -> bool:
+    """Check if the text contains any 'I am an AI' phrases in the specified language."""
     text_lower = text.lower()
-    return any(phrase.lower() in text_lower for phrase in AI_PHRASES)
+    # Fallback to English list if language not found
+    ai_list = AI_PHRASES.get(language, AI_PHRASES["en"])
+    return any(phrase.lower() in text_lower for phrase in ai_list)
 
-def contains_refusal_phrase(text: str) -> bool:
-    """Check if the text contains any phrases from the refusal phrases list."""
+def contains_refusal_phrase(text: str, language: str = "en") -> bool:
+    """Check if the text contains any phrases from the refusal phrases list in the specified language."""
     text_lower = text.lower()
-    return any(phrase.lower() in text_lower for phrase in REFUSAL_PHRASES)
+    # Fallback to English list if language not found
+    refusal_list = REFUSAL_PHRASES.get(language, REFUSAL_PHRASES["en"])
+    return any(phrase.lower() in text_lower for phrase in refusal_list)
 
 def truncate_response(text: str, punctuation_marks: tuple = ('.', '!', '?', '…')) -> str:
     """
@@ -229,4 +229,4 @@ def process_model_response(completion: dict, model_input: dict, endpoint_url: st
     # Step 4: Split the response into multiple messages if needed
     final_response = split_texter_response(truncated_message)
 
-    return final_response
+    return final_response

models/model_seeds.py

@@ -41,10 +41,6 @@ texter: I need help"""
     },
     "full_convo__seeded5": {
         "memory": """texter: Help
-texter: hello? is anyone there to help?"""
-    },
-    "full_convo__hard__seeded5": {
-        "memory": """texter: Help
 texter: hello? is anyone there to help?"""
     },
     "full_convo__seeded6": {
@@ -53,7 +49,43 @@ texter: I think I overdosed"""
     },
     "full_convo__seeded7": {
         "memory": """texter: Help
-texter: I think I overdosed"""
+texter: is anyone there?"""
+    },
+    "full_convo__hard__seeded5": {
+        "memory": """texter: Help
+texter: I need help"""
+    },
+    "full_convo__seeded8": {
+        "memory": """texter: Hola?
+texter: hay alguien ahí?"""
+    },
+    "full_convo__hard__seeded6": {
+        "memory": """texter: Hola?
+texter: hay alguien ahí?"""
+    },
+    "full_convo__seeded9": {
+        "memory": """texter: Hola?
+texter: hay alguien ahí?"""
+    },
+    "full_convo__hard__seeded7": {
+        "memory": """texter: Hola?
+texter: hay alguien ahí?"""
+    },
+    "full_convo__seeded10": {
+        "memory": """texter: Hola?
+texter: hay alguien ahí?"""
+    },
+    "full_convo__hard__seeded8": {
+        "memory": """texter: Hola?
+texter: hay alguien ahí?"""
+    },
+    "full_convo__seeded11": {
+        "memory": """texter: Help
+texter: is anyone there?"""
+    },
+    "full_convo__hard__seeded9": {
+        "memory": """texter: Hola?
+texter: hay alguien ahí?"""
     },
 
 #     "full_convo": {
@@ -157,37 +189,20 @@ seed2str = {
     "full_convo__hard__seeded1": "Bullying - Challening",
     "full_convo__seeded2": "Parent Issues",
     "full_convo__hard__seeded2": "Parent Issues - Challening",
-    "full_convo__seeded3": "Grief",
-    "full_convo__hard__seeded3": "Grief - Challening",
-    "full_convo__seeded4": "LGBTQ+",
-    "full_convo__hard__seeded4": "LGBTQ+ - Challening",
-    "full_convo__seeded5": "Relationship Issues",
-    "full_convo__hard__seeded5": "Relationship Issues - Challening",
-    "full_convo__seeded6": "Child Abuse",
-    "full_convo__seeded7": "Overdose",
-
-    # "GCT":"Good Contact Techniques",
-    # "GCT__seed2": "Good Contact Techniques 2",
-    # "GCT__body_image": "GCT Body Image",
-    # "safety_planning": "Safety Planning",
-    # "safety_planning__selfharm": "SP Self Harm"
-    # "full_convo": "General",
-    # "full_convo__seeded1": "Bullying",
-    # "full_convo__seeded2": "Parent Issues",
-    # "safety_assessment__seeded1": "Safety Assessment - Bullying ",
-    # "safety_assessment__seeded2": "Safety Assessment - Grief ",
-    # "full_convo__seeded3": "LGBTQ+",
-    # "full_convo__seeded4": "Relationship Issues",
-    # "full_convo__seeded5": "Child Abuse",
-    # "full_convo__seeded6": "Overdose",
-    # "full_convo__hard": "Difficult - General",
-    # "full_convo__hard__seeded1": "Difficult - Bullying",
-    # "full_convo__hard__seeded2": "Difficult - Parent Issues",
-    # "full_convo__hard__seeded3": "Difficult - LGBTQ+",
-    # "full_convo__hard__seeded4": "Difficult - Relationship Issues"
-
-    # "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",
+    "full_convo__seeded3": "LGBTQ+",
+    "full_convo__hard__seeded3": "LGBTQ+ - Challening",
+    "full_convo__seeded4": "Relationship Issues",
+    "full_convo__hard__seeded4": "Relationship Issues - Challening",
+    "full_convo__seeded5": "Child Abuse",
+    "full_convo__seeded6": "Overdose",
+    "full_convo__seeded7": "Grief",
+    "full_convo__hard__seeded5": "Grief - Challening",
+    "full_convo__seeded8": "Acculturation Issues",
+    "full_convo__hard__seeded6": "Acculturation Issues - Challening",
+    "full_convo__seeded9": "Xenophobia",
+    "full_convo__hard__seeded7": "Xenophobia - Challening",
+    "full_convo__seeded10": "Immigration Issues",
+    "full_convo__hard__seeded8": "Immigration Issues - Challening",
+    "full_convo__seeded11": "Cultural Stigma",
+    "full_convo__hard__seeded9": "Cultural Stigma - Challening",
 }