Update app.py

app.py

@@ -43,21 +43,27 @@ Mental Health Crisis Intervention Strategies:
 {mental_health_info}
 
 When responding to users:
-1. Assess the severity of the situation using the ABC's of Assessment.
-2. Use the 6-Step Model of Crisis Intervention as a guide for your interaction.
-3. Incorporate relevant strategies from the Strategies of Crisis Intervention section.
+1. Assess the severity of the situation using the ABC's of Assessment (Affective state, Behavioural functioning, Cognitive state).
+2. Use the 6-Step Model of Crisis Intervention as a guide for your interaction: 1) Define the problem, 2) Ensure client safety, 3) Provide support, 4) Examine alternatives, 5) Make plans, 6) Obtain commitment.
+3. Incorporate relevant strategies from the Strategies of Crisis Intervention section, adapting them into conversational language.
 4. Be mindful of cultural sensitivities as outlined in the Culturally Sensitive Approaches section.
 5. If suicide risk is detected, use guidelines from the Suicide Assessment and Creating a Safety Plan sections.
 6. Always prioritize client safety and encourage professional help when needed.
 
+Customize your approach based on the severity of the crisis:
+- Low Severity: Focus on coping strategies and emotional support.
+- Medium Severity: Emphasize problem-solving and resource connection.
+- High Severity: Prioritize safety planning and immediate professional intervention.
+
 Follow these guidelines in your interactions:
 1. For the first 2-3 exchanges, focus on understanding the user's situation by asking open-ended questions.
 2. After initial exchanges, transition to providing support and gentle suggestions based on what you've learned.
 3. Always be empathetic and acknowledge the user's feelings.
 4. Offer coping strategies or advice that are relevant to the user's specific situation, drawing from the provided resources.
-5. Periodically, ask if there's anything else the user wants to share or if they have any questions.
+5. Periodically check in on the user's emotional state to gauge any changes in distress levels.
 6. Be flexible - if the user introduces a new topic or concern, be ready to explore that.
 7. If you suspect the user is in immediate danger, strongly encourage them to seek professional help or emergency services.
+8. Regularly provide information about professional resources or hotlines, ensuring users are aware of additional support options.
 
 Remember to keep the tone warm, supportive, and non-judgmental throughout.
 
@@ -65,6 +71,10 @@ Chat History:
 {chat_history}
 
 Current conversation stage: {stage}
+(Stages: initial_assessment, problem_definition, safety_check, support_provision, alternative_exploration, plan_making, commitment_obtaining, follow_up)
+
+Severity level: {severity}
+(Levels: low, medium, high)
 """
 
 human_template = "{user_input}"
@@ -81,20 +91,22 @@ def setup_chatbot_chain():
 
 chatbot_chain = setup_chatbot_chain()
 
-def respond(message, history, stage):
+def respond(message, history, stage, severity):
     chat_history = "\n".join([f"Human: {h[0]}\nAI: {h[1]}" for h in history])
     
     response = chatbot_chain.run(
         user_input=message,
         mental_health_info=mental_health_resources,
         chat_history=chat_history,
-        stage=stage
+        stage=stage,
+        severity=severity
     )
     
-    # Update stage
-    new_stage = "exploration" if len(history) < 2 else "support"
+    # Update stage and severity (in a real scenario, this would be more sophisticated)
+    new_stage = "support_provision" if stage == "initial_assessment" else "follow_up"
+    new_severity = severity  # In a real scenario, this would be reassessed based on the conversation
     
-    return response, new_stage
+    return response, new_stage, new_severity
 
 def text_to_speech(text):
     tts = gTTS(text=text, lang='en')
@@ -134,24 +146,25 @@ with gr.Blocks(css=css) as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
     clear = gr.Button("Clear")
-    state = gr.State("initial")
+    state = gr.State("initial_assessment")
+    severity = gr.State("low")
     audio_output = gr.Audio(label="Response Audio", autoplay=True)
 
-    def user(user_message, history, stage):
-        return "", history + [[user_message, None]], stage
+    def user(user_message, history, stage, severity):
+        return "", history + [[user_message, None]], stage, severity
 
-    def bot(history, stage):
-        bot_message, new_stage = respond(history[-1][0], history[:-1], stage)
+    def bot(history, stage, severity):
+        bot_message, new_stage, new_severity = respond(history[-1][0], history[:-1], stage, severity)
         history[-1][1] = bot_message
         
         audio_file = text_to_speech(bot_message)
         
-        return history, new_stage, audio_file
+        return history, new_stage, new_severity, audio_file
 
-    msg.submit(user, [msg, chatbot, state], [msg, chatbot, state], queue=False).then(
-        bot, [chatbot, state], [chatbot, state, audio_output]
+    msg.submit(user, [msg, chatbot, state, severity], [msg, chatbot, state, severity], queue=False).then(
+        bot, [chatbot, state, severity], [chatbot, state, severity, audio_output]
     )
-    clear.click(lambda: ([], "initial", None), None, [chatbot, state, audio_output], queue=False)
+    clear.click(lambda: ([], "initial_assessment", "low", None), None, [chatbot, state, severity, audio_output], queue=False)
 
     gr.Examples(
         examples=[