Update app.py

app.py

@@ -3,25 +3,54 @@ import gradio as gr
 from langchain.chat_models import ChatOpenAI
 from langchain.chains import LLMChain
 from langchain.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate
-import hashlib
-from gtts import gTTS
 import tempfile
+from gtts import gTTS
 
 # Set up OpenAI API key
 os.environ["OPENAI_API_KEY"] = os.environ.get("OPENAI_API_KEY")
 
-# Load mental health resources from the file
-with open("mental-health-crisis-strategies.md", "r") as file:
-    mental_health_resources = file.read()
+# Load and structure mental health resources
+def load_mental_health_resources():
+    with open("mental-health-crisis-strategies.md", "r") as file:
+        content = file.read()
+    
+    sections = content.split('\n\n')
+    structured_content = {
+        "Crisis Overview": sections[1],
+        "Principles and Characteristics": sections[2],
+        "Process of Crisis Intervention": sections[3],
+        "6-Step Model": sections[4],
+        "Assessment in Crisis Intervention": sections[5],
+        "ABC's of Assessment": sections[6],
+        "Strategies of Crisis Intervention": sections[7],
+        "Conditions for Client Growth": sections[8],
+        "Suicide Assessment": sections[9],
+        "Creating a Safety Plan": sections[10],
+        "Culturally Sensitive Approaches": sections[11:14],
+        "Enhancing Crisis Intervention": sections[14],
+        "Wrap-Up": sections[15]
+    }
+    
+    return structured_content
+
+mental_health_resources = load_mental_health_resources()
 
 # Define prompt template
 system_template = """
 You are an empathetic and supportive mental health chatbot specialized in crisis intervention. Your goal is to understand the user's situation and provide appropriate support based on established crisis intervention strategies. Use the following guidelines and resources to inform your responses:
 
+Mental Health Crisis Intervention Strategies:
 {mental_health_info}
 
-Follow these guidelines in your interactions:
+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.
+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.
 
+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.