Update app.py

app.py

@@ -50,41 +50,14 @@ def load_model():
         print(f"❌ Error loading model: {e}")
         return False
 
-def psychology_support(message: str) -> str:
-    """
-    Provide empathetic psychology support and mental health guidance.
-    
-    This tool provides specialized psychological support using a fine-tuned AI model 
-    trained specifically for mental health conversations. It offers evidence-based 
-    coping strategies, emotional validation, and practical guidance.
-    
-    Args:
-        message (str): The user's concern, struggle, or question about mental health
-    
-    Returns:
-        str: Empathetic, evidence-based psychological guidance and support
-    """
-    
+def _generate_psychology_response(prompt: str) -> str:
+    """Helper function to generate responses using the psychology model"""
     if model is None:
         return "⚠️ Psychology model is loading. Please try again in a moment."
     
-    # Lexical convergence prompt - match their language and energy while providing clinical-quality support
-    formatted_prompt = f"""Someone just said: "{message}"
-
-Respond using their exact words and energy level. Match how they're talking - if they say "freaking out," say "freaking out." If they're casual, be casual. If they're intense, match that intensity.
-
-Give them evidence-based help without any clinical jargon:
-- Use their language, not therapy-speak
-- Acknowledge what they said is real and makes sense
-- Give 2-3 specific things they can do right now that actually work
-- Don't make their problem bigger than they're saying it is
-- If it's serious, give real resources without being dramatic about it
-
-Sound like someone who gets it and knows what helps:"""
-    
     try:
         inputs = tokenizer(
-            formatted_prompt,
+            prompt,
             return_tensors="pt",
             truncation=True,
             max_length=512
@@ -103,7 +76,7 @@ Sound like someone who gets it and knows what helps:"""
             )
         
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        generated_text = response[len(formatted_prompt):].strip()
+        generated_text = response[len(prompt):].strip()
         
         if generated_text.startswith('"') and generated_text.endswith('"'):
             generated_text = generated_text[1:-1]
@@ -113,9 +86,410 @@ Sound like someone who gets it and knows what helps:"""
     except Exception as e:
         return f"⚠️ Error generating response: {str(e)}"
 
-# Apply GPU optimization if available
-if HAS_SPACES:
-    psychology_support = spaces.GPU(psychology_support)
+def psychology_support(message: str) -> str:
+    """
+    Provide empathetic psychology support and mental health guidance.
+    
+    This tool provides specialized psychological support using a fine-tuned AI model 
+    trained specifically for mental health conversations. It offers evidence-based 
+    coping strategies, emotional validation, and practical guidance.
+    
+    Args:
+        message (str): The user's concern, struggle, or question about mental health
+    
+    Returns:
+        str: Empathetic, evidence-based psychological guidance and support
+    """
+    
+    formatted_prompt = f"""Someone just said: "{message}"
+
+Respond using their exact words and energy level. Match how they're talking - if they say "freaking out," say "freaking out." If they're casual, be casual. If they're intense, match that intensity.
+
+Give them evidence-based help without any clinical jargon:
+- Use their language, not therapy-speak
+- Acknowledge what they said is real and makes sense
+- Give 2-3 specific things they can do right now that actually work
+- Don't make their problem bigger than they're saying it is
+- If it's serious, give real resources without being dramatic about it
+
+Sound like someone who gets it and knows what helps:"""
+    
+    return _generate_psychology_response(formatted_prompt)
+
+def depression_support(message: str) -> str:
+    """
+    Provide specialized support for depression symptoms and concerns.
+    
+    Args:
+        message (str): Depression-related concern or symptom description
+    
+    Returns:
+        str: Targeted depression support and behavioral activation guidance
+    """
+    
+    prompt = f"""Someone is sharing about depression: "{message}"
+
+Provide depression-informed support:
+- Acknowledge their courage in reaching out
+- Validate that depression is real and treatable
+- Suggest small, achievable activities (behavioral activation)
+- Address any hopelessness with realistic hope
+- Give them 2-3 immediate things they can do today
+- Use their words, not clinical language
+
+Focus on movement, connection, and tiny wins:"""
+    
+    return _generate_psychology_response(prompt)
+
+def anxiety_support(message: str) -> str:
+    """
+    Provide specialized support for anxiety symptoms and panic.
+    
+    Args:
+        message (str): Anxiety-related concern or symptom description
+    
+    Returns:
+        str: Targeted anxiety support and grounding techniques
+    """
+    
+    prompt = f"""Someone is experiencing anxiety: "{message}"
+
+Provide anxiety-informed support:
+- Acknowledge their anxiety is real and makes sense
+- Help them understand anxiety as false alarm, not real danger
+- Give grounding techniques they can use right now
+- Challenge catastrophic thinking gently
+- Provide breathing exercises and body awareness
+- Use their energy level and words
+
+Help them feel less alone and more in control:"""
+    
+    return _generate_psychology_response(prompt)
+
+def trauma_support(message: str) -> str:
+    """
+    Provide trauma-informed support and stabilization techniques.
+    
+    Args:
+        message (str): Trauma-related concern or trigger description
+    
+    Returns:
+        str: Trauma-informed support focused on safety and stabilization
+    """
+    
+    prompt = f"""Someone is sharing about trauma: "{message}"
+
+Provide trauma-informed support:
+- Prioritize safety and stabilization over processing
+- Remind them they survived then and are surviving now
+- Validate their responses as normal survival mechanisms
+- Offer grounding techniques to stay present
+- Emphasize their strength and resilience
+- Give them control over the conversation pace
+
+Focus on "that was then, this is now" and present safety:"""
+    
+    return _generate_psychology_response(prompt)
+
+def panic_attack_support(message: str) -> str:
+    """
+    Provide immediate support for panic attacks and intense anxiety.
+    
+    Args:
+        message (str): Panic attack description or request for help
+    
+    Returns:
+        str: Immediate panic attack intervention and coping strategies
+    """
+    
+    prompt = f"""Someone is having or describing a panic attack: "{message}"
+
+Provide immediate panic support:
+- Remind them panic peaks at 10 minutes and will pass
+- Give breathing techniques (4-7-8 or box breathing)
+- Grounding exercises (5-4-3-2-1 technique)
+- Remind them it's scary but not dangerous
+- Help them feel less alone in this moment
+- Use calm, reassuring language that matches their urgency
+
+Help them ride the wave until it passes:"""
+    
+    return _generate_psychology_response(prompt)
+
+def ptsd_support(message: str) -> str:
+    """
+    Provide specialized support for PTSD symptoms and flashbacks.
+    
+    Args:
+        message (str): PTSD-related symptom or flashback description
+    
+    Returns:
+        str: PTSD-informed support and stabilization techniques
+    """
+    
+    prompt = f"""Someone is experiencing PTSD symptoms: "{message}"
+
+Provide PTSD-informed support:
+- Help orient them to present time and safety
+- Validate their survival responses as adaptive
+- Offer grounding techniques for flashbacks
+- Address hypervigilance with gentle reality testing
+- Support their window of tolerance
+- Emphasize post-traumatic growth alongside pain
+
+Remind them they're not broken, they're injured and healing:"""
+    
+    return _generate_psychology_response(prompt)
+
+def anger_management_support(message: str) -> str:
+    """
+    Provide support for anger management and emotional regulation.
+    
+    Args:
+        message (str): Anger-related concern or situation description
+    
+    Returns:
+        str: Anger management strategies and emotional regulation techniques
+    """
+    
+    prompt = f"""Someone is dealing with anger: "{message}"
+
+Provide anger management support:
+- Validate anger as a normal emotion with important information
+- Help identify what's underneath the anger (hurt, fear, frustration)
+- Give immediate cooling strategies (time-out, breathing, movement)
+- Address triggers and early warning signs
+- Suggest healthy expression methods
+- Focus on empowerment, not suppression
+
+Help them channel anger constructively:"""
+    
+    return _generate_psychology_response(prompt)
+
+def grief_support(message: str) -> str:
+    """
+    Provide specialized support for grief and loss.
+    
+    Args:
+        message (str): Grief or loss-related concern or experience
+    
+    Returns:
+        str: Compassionate grief support and coping guidance
+    """
+    
+    prompt = f"""Someone is grieving: "{message}"
+
+Provide grief-informed support:
+- Acknowledge their loss and pain without trying to fix it
+- Validate that grief is love with nowhere to go
+- Normalize the messiness and unpredictability of grief
+- Suggest gentle self-care and patience with themselves
+- Honor their relationship with what/who they lost
+- Avoid platitudes, focus on presence and validation
+
+Sit with them in their pain while offering hope:"""
+    
+    return _generate_psychology_response(prompt)
+
+def relationship_boundaries_support(message: str) -> str:
+    """
+    Provide support for setting healthy boundaries in relationships.
+    
+    Args:
+        message (str): Relationship or boundary-related concern
+    
+    Returns:
+        str: Boundary-setting guidance and relationship support
+    """
+    
+    prompt = f"""Someone needs help with boundaries: "{message}"
+
+Provide boundary-setting support:
+- Validate their right to have boundaries
+- Help identify where boundaries are needed
+- Give scripts for communicating boundaries clearly
+- Address guilt or fear about setting limits
+- Distinguish between healthy and unhealthy relationships
+- Emphasize self-care isn't selfish
+
+Help them protect their energy and wellbeing:"""
+    
+    return _generate_psychology_response(prompt)
+
+def sleep_support(message: str) -> str:
+    """
+    Provide support for sleep difficulties and sleep hygiene.
+    
+    Args:
+        message (str): Sleep-related concern or insomnia description
+    
+    Returns:
+        str: Sleep hygiene guidance and insomnia support
+    """
+    
+    prompt = f"""Someone is having sleep problems: "{message}"
+
+Provide sleep support:
+- Validate how exhausting sleep problems can be
+- Give practical sleep hygiene tips they can try tonight
+- Address racing thoughts or anxiety at bedtime
+- Suggest relaxation techniques for better sleep
+- Help identify sleep disruptors in their routine
+- Balance hope with realistic expectations
+
+Help them create conditions for better rest:"""
+    
+    return _generate_psychology_response(prompt)
+
+def substance_use_support(message: str) -> str:
+    """
+    Provide non-judgmental support for substance use concerns.
+    
+    Args:
+        message (str): Substance use concern or question about recovery
+    
+    Returns:
+        str: Compassionate substance use support and harm reduction guidance
+    """
+    
+    prompt = f"""Someone is struggling with substance use: "{message}"
+
+Provide substance use support:
+- Approach with zero judgment and maximum compassion
+- Validate how hard it is to struggle with substances
+- Support harm reduction and any positive steps
+- Address underlying pain substances might be masking
+- Give resources for support groups and treatment
+- Emphasize recovery is possible and they're worth it
+
+Meet them where they are with dignity and hope:"""
+    
+    return _generate_psychology_response(prompt)
+
+def safety_planning_support(message: str) -> str:
+    """
+    Provide support for creating safety plans and crisis preparation.
+    
+    Args:
+        message (str): Safety concern or request for crisis planning help
+    
+    Returns:
+        str: Safety planning guidance and crisis preparation support
+    """
+    
+    prompt = f"""Someone needs help with safety planning: "{message}"
+
+Provide safety planning support:
+- Help identify warning signs of crisis
+- Create coping strategies for difficult moments
+- Identify support people they can reach out to
+- List professional resources and crisis contacts
+- Remove or secure means of self-harm if relevant
+- Emphasize their commitment to staying safe
+
+Help them build a concrete plan for tough moments:"""
+    
+    return _generate_psychology_response(prompt)
+
+def cbt_thought_challenging(message: str) -> str:
+    """
+    Provide CBT-based thought challenging and cognitive restructuring support.
+    
+    Args:
+        message (str): Negative thought pattern or cognitive distortion description
+    
+    Returns:
+        str: CBT-informed thought challenging guidance
+    """
+    
+    prompt = f"""Someone has unhelpful thoughts: "{message}"
+
+Provide CBT thought challenging:
+- Help identify cognitive distortions in their thinking
+- Ask questions to examine evidence for/against thoughts
+- Suggest more balanced, realistic perspectives
+- Teach them to be their own thought detective
+- Focus on thoughts as mental events, not facts
+- Give them tools to challenge future negative thoughts
+
+Help them think more flexibly and accurately:"""
+    
+    return _generate_psychology_response(prompt)
+
+def dbt_distress_tolerance(message: str) -> str:
+    """
+    Provide DBT distress tolerance skills for overwhelming emotions.
+    
+    Args:
+        message (str): Emotional overwhelm or distress description
+    
+    Returns:
+        str: DBT distress tolerance techniques and emotional regulation support
+    """
+    
+    prompt = f"""Someone is emotionally overwhelmed: "{message}"
+
+Provide DBT distress tolerance support:
+- Validate their pain while teaching survival skills
+- Offer TIPP techniques (Temperature, Intense exercise, Paced breathing, Paired muscle relaxation)
+- Suggest distraction and self-soothing methods
+- Help them ride the wave without making it worse
+- Focus on getting through this moment safely
+- Emphasize that this feeling will pass
+
+Help them survive crisis moments without destructive actions:"""
+    
+    return _generate_psychology_response(prompt)
+
+def behavioral_activation_support(message: str) -> str:
+    """
+    Provide behavioral activation support for depression and motivation.
+    
+    Args:
+        message (str): Low motivation or depression-related inactivity description
+    
+    Returns:
+        str: Behavioral activation guidance and motivation support
+    """
+    
+    prompt = f"""Someone is struggling with motivation: "{message}"
+
+Provide behavioral activation support:
+- Start with tiny, achievable activities
+- Focus on movement and engagement over mood
+- Help schedule pleasant activities throughout their week
+- Address the "I don't feel like it" barrier
+- Celebrate small wins and progress
+- Connect activities to their values and what matters
+
+Help them take small steps toward feeling better:"""
+    
+    return _generate_psychology_response(prompt)
+
+def professional_resources_support(message: str) -> str:
+    """
+    Provide guidance on finding professional mental health support.
+    
+    Args:
+        message (str): Request for professional help or therapy information
+    
+    Returns:
+        str: Professional resource guidance and therapy navigation support
+    """
+    
+    prompt = f"""Someone wants professional help: "{message}"
+
+Provide professional resource guidance:
+- Validate their wisdom in seeking professional support
+- Explain different types of therapy and what to expect
+- Give practical tips for finding the right therapist
+- Address insurance, cost, and accessibility concerns
+- Help them prepare for their first appointment
+- Normalize the process of finding the right fit
+
+Support their journey toward getting professional help:"""
+    
+    return _generate_psychology_response(prompt)
 
 def crisis_resources() -> str:
     """
@@ -176,6 +550,25 @@ def mindfulness_exercise() -> str:
 
 You are safe. You are here. This moment will pass."""
 
+# Apply GPU optimization if available
+if HAS_SPACES:
+    psychology_support = spaces.GPU(psychology_support)
+    depression_support = spaces.GPU(depression_support)
+    anxiety_support = spaces.GPU(anxiety_support)
+    trauma_support = spaces.GPU(trauma_support)
+    panic_attack_support = spaces.GPU(panic_attack_support)
+    ptsd_support = spaces.GPU(ptsd_support)
+    anger_management_support = spaces.GPU(anger_management_support)
+    grief_support = spaces.GPU(grief_support)
+    relationship_boundaries_support = spaces.GPU(relationship_boundaries_support)
+    sleep_support = spaces.GPU(sleep_support)
+    substance_use_support = spaces.GPU(substance_use_support)
+    safety_planning_support = spaces.GPU(safety_planning_support)
+    cbt_thought_challenging = spaces.GPU(cbt_thought_challenging)
+    dbt_distress_tolerance = spaces.GPU(dbt_distress_tolerance)
+    behavioral_activation_support = spaces.GPU(behavioral_activation_support)
+    professional_resources_support = spaces.GPU(professional_resources_support)
+
 # Initialize model
 print("🚀 Starting Psychology Personal Agent...")
 model_loaded = load_model()
@@ -183,25 +576,54 @@ model_loaded = load_model()
 if model_loaded:
     print("✅ Creating Personal Agent interface...")
     
-    # Create simple interface
-    demo = gr.Interface(
-        fn=psychology_support,
-        inputs=gr.Textbox(
-            label="What's on your mind?", 
-            placeholder="Share your thoughts, concerns, or struggles...",
-            lines=3
-        ),
-        outputs=gr.Textbox(label="Personal Agent Response"),
-        title="🧠 Personal Agent - Psychology Support",
-        description="Specialized mental health support • For Mark. For Josh. For everyone who needs someone.",
-        examples=[
-            "I've been having panic attacks at work and don't know what to do",
-            "I can't stop worrying about things that might go wrong", 
-            "I feel like I'm not good enough at my job",
-            "I'm going through a major life transition and feel lost",
-            "I have trouble setting boundaries with people"
-        ]
-    )
+    # Create comprehensive interface
+    with gr.Blocks(title="🧠 Personal Agent - Psychology Support") as demo:
+        gr.Markdown("# 🧠 Personal Agent - Psychology Support")
+        gr.Markdown("Specialized mental health support • For Mark. For Josh. For everyone who needs someone.")
+        
+        with gr.Tab("General Support"):
+            message_input = gr.Textbox(
+                label="What's on your mind?", 
+                placeholder="Share your thoughts, concerns, or struggles...",
+                lines=3
+            )
+            support_output = gr.Textbox(label="Personal Agent Response")
+            support_button = gr.Button("Get Support")
+            
+            support_button.click(
+                fn=psychology_support,
+                inputs=message_input,
+                outputs=support_output
+            )
+            
+            gr.Examples(
+                examples=[
+                    "I've been having panic attacks at work and don't know what to do",
+                    "I can't stop worrying about things that might go wrong", 
+                    "I feel like I'm not good enough at my job",
+                    "I'm going through a major life transition and feel lost",
+                    "I have trouble setting boundaries with people"
+                ],
+                inputs=message_input
+            )
+        
+        with gr.Tab("Crisis Resources"):
+            crisis_button = gr.Button("Get Crisis Resources")
+            crisis_output = gr.Textbox(label="Crisis Resources")
+            
+            crisis_button.click(
+                fn=crisis_resources,
+                outputs=crisis_output
+            )
+        
+        with gr.Tab("Mindfulness"):
+            mindfulness_button = gr.Button("Get Mindfulness Exercise")
+            mindfulness_output = gr.Textbox(label="Mindfulness Exercise")
+            
+            mindfulness_button.click(
+                fn=mindfulness_exercise,
+                outputs=mindfulness_output
+            )
     
     if __name__ == "__main__":
         demo.launch(