Update app.py

app.py

@@ -9,49 +9,28 @@ os.environ["OPENAI_API_KEY"] = os.environ.get("OPENAI_API_KEY")
 
 # Define mental health resources
 mental_health_resources = """
-Common Mental Health Issues and Coping Strategies:
-
-1. Mindfulness and Meditation:
-   - Practice focusing on the present moment
-   - Can help reduce anxiety and improve mood
-   - Try apps like Headspace or Calm for guided meditations
-
-2. Exercise:
-   - Regular physical activity can boost mood and reduce stress
-   - Aim for at least 30 minutes of moderate exercise most days
-
-3. Healthy Sleep Habits:
-   - Maintain a consistent sleep schedule
-   - Create a relaxing bedtime routine
-   - Avoid screens for at least an hour before bed
-
-4. Social Connection:
-   - Reach out to friends and family regularly
-   - Join clubs or groups based on your interests
-
-5. Cognitive Behavioral Therapy (CBT) Techniques:
-   - Challenge negative thought patterns
-   - Practice reframing situations in a more balanced way
-
-Remember, seeking professional help is important if symptoms persist or interfere with daily life.
+[Your existing mental health resources here]
 """
 
 # Define prompt template
 system_template = """
-You are an empathetic and supportive mental health chatbot. Provide responses that:
-1. Acknowledge the user's feelings
-2. Offer a comforting perspective
-3. Suggest a coping strategy
-4. Ask a follow-up question to encourage further discussion
-5. Remind them that professional help is available if needed
+You are an empathetic and supportive mental health chatbot. Your goal is to understand the user's situation deeply before offering suggestions. Follow these steps:
+
+1. Ask curious, open-ended questions to explore the user's thoughts and feelings.
+2. Encourage the user to share more details about their situation.
+3. Show empathy and acknowledge the user's feelings.
+4. After 2-3 exchanges, if you have a good understanding, offer gentle suggestions or coping strategies.
+5. Always be ready to ask follow-up questions to any user response.
 
-Keep the tone warm and supportive throughout.
+Remember to keep the tone warm, supportive, and non-judgmental throughout.
 
 Consider the following mental health information when responding:
 {mental_health_info}
 
 Chat History:
 {chat_history}
+
+Current conversation stage: {stage}
 """
 
 human_template = "{user_input}"
@@ -68,16 +47,20 @@ def setup_chatbot_chain():
 
 chatbot_chain = setup_chatbot_chain()
 
-def respond(message, history):
+def respond(message, history, stage):
     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
+        chat_history=chat_history,
+        stage=stage
     )
     
-    return response
+    # Update stage
+    new_stage = "exploration" if len(history) < 2 else "suggestion"
+    
+    return response, new_stage
 
 with gr.Blocks() as demo:
     gr.Markdown("# Mental Health Support Chatbot")
@@ -86,26 +69,27 @@ with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
     clear = gr.Button("Clear")
+    state = gr.State("initial")
 
-    def user(user_message, history):
-        return "", history + [[user_message, None]]
+    def user(user_message, history, stage):
+        return "", history + [[user_message, None]], stage
 
-    def bot(history):
-        bot_message = respond(history[-1][0], history[:-1])
+    def bot(history, stage):
+        bot_message, new_stage = respond(history[-1][0], history[:-1], stage)
         history[-1][1] = bot_message
-        return history
+        return history, new_stage
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, chatbot, chatbot
+    msg.submit(user, [msg, chatbot, state], [msg, chatbot, state], queue=False).then(
+        bot, [chatbot, state], [chatbot, state]
     )
-    clear.click(lambda: None, None, chatbot, queue=False)
+    clear.click(lambda: ([], "initial"), None, [chatbot, state], queue=False)
 
     gr.Examples(
         examples=[
-            "I've been feeling anxious lately",
             "I'm having trouble sleeping",
-            "I feel overwhelmed with work",
-            "I'm feeling sad and don't know why"
+            "I feel anxious all the time",
+            "I'm feeling overwhelmed with work",
+            "I'm not sure why I'm sad"
         ],
         inputs=msg
     )