Update app.py

app.py

@@ -3,8 +3,6 @@ import torch
 from peft import PeftModel
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import warnings
-import time
-import spaces
 
 warnings.filterwarnings('ignore')
 
@@ -14,19 +12,17 @@ tokenizer = None
 device = None
 
 def load_model():
-    """Load the psychology-tuned model with error handling"""
+    """Load the psychology-tuned model"""
     global model, tokenizer, device
     
     try:
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        print(f"Loading model on {device}...")
+        print(f"Loading psychology model on {device}...")
         
-        # Load tokenizer
         tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-0.5B")
         if tokenizer.pad_token is None:
             tokenizer.pad_token = tokenizer.eos_token
         
-        # Load base model
         base_model = AutoModelForCausalLM.from_pretrained(
             "Qwen/Qwen2.5-0.5B",
             torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
@@ -34,66 +30,43 @@ def load_model():
             trust_remote_code=True
         )
         
-        # Load PEFT adapter
         model = PeftModel.from_pretrained(base_model, "phxdev/psychology-qwen-0.5b")
         model = model.merge_and_unload()
         
         if not torch.cuda.is_available():
             model = model.to(device)
         
-        print("✅ Model loaded successfully!")
+        print("✅ Psychology model loaded successfully!")
         return True
         
     except Exception as e:
         print(f"❌ Error loading model: {e}")
         return False
 
-@spaces.GPU
-def generate_response(
-    message: str,
-    history: list,
-    temperature: float = 0.8,
-    max_tokens: int = 300,
-    prompt_style: str = "Therapeutic"
-) -> str:
-    """Generate psychology-focused response"""
+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
+    """
     
     if model is None:
-        return "⚠️ Model is still loading. Please wait a moment and try again."
+        return "⚠️ Psychology model is loading. Please try again in a moment."
     
-    # Define prompt templates
-    prompt_templates = {
-        "Therapeutic": """You're a supportive therapist in session. The client just said: "{message}"
-
-Respond with empathy and practical guidance. Start with validation, then give 2-3 specific strategies they can try:""",
-        
-        "Supportive Friend": """You're a caring friend who studied psychology. Someone you care about just told you: "{message}"
-
-Give them warm, understanding support with practical advice:""",
-        
-        "Crisis Support": """You are a crisis counselor. Someone is reaching out for support. They said: "{message}"
-
-Provide immediate, caring support and grounding techniques:""",
-        
-        "Anxiety Coach": """You're an anxiety specialist. Help them manage their anxiety with evidence-based techniques.
-
-They're struggling with: "{message}"
+    # Therapeutic prompt template
+    formatted_prompt = f"""You're a supportive therapist in session. The client just said: "{message}"
 
-Here are specific techniques they can try right now:""",
-        
-        "Mindfulness Guide": """You're a mindfulness teacher. Guide them toward present-moment awareness and self-compassion.
-
-They shared: "{message}"
-
-Offer mindful perspective and a gentle practice:"""
-    }
-    
-    # Select and format prompt
-    template = prompt_templates.get(prompt_style, prompt_templates["Therapeutic"])
-    formatted_prompt = template.format(message=message)
+Respond with empathy and practical guidance. Start with validation, then give 2-3 specific strategies they can try:"""
     
     try:
-        # Tokenize
         inputs = tokenizer(
             formatted_prompt,
             return_tensors="pt",
@@ -101,12 +74,11 @@ Offer mindful perspective and a gentle practice:"""
             max_length=512
         ).to(device)
         
-        # Generate
         with torch.no_grad():
             outputs = model.generate(
                 **inputs,
-                max_length=len(inputs.input_ids[0]) + max_tokens,
-                temperature=temperature,
+                max_length=len(inputs.input_ids[0]) + 300,
+                temperature=0.8,
                 top_p=0.9,
                 do_sample=True,
                 repetition_penalty=1.1,
@@ -114,11 +86,9 @@ Offer mindful perspective and a gentle practice:"""
                 eos_token_id=tokenizer.eos_token_id,
             )
         
-        # Decode
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
         generated_text = response[len(formatted_prompt):].strip()
         
-        # Clean up response
         if generated_text.startswith('"') and generated_text.endswith('"'):
             generated_text = generated_text[1:-1]
         
@@ -127,211 +97,109 @@ Offer mindful perspective and a gentle practice:"""
     except Exception as e:
         return f"⚠️ Error generating response: {str(e)}"
 
-def create_interface():
-    """Create the main Gradio interface"""
+def crisis_resources() -> str:
+    """
+    Get immediate crisis intervention resources and emergency contacts.
+    
+    Returns critical mental health crisis resources including suicide prevention
+    hotlines, crisis text lines, and emergency contacts for immediate help.
     
-    # Custom CSS for modern, calming design
-    custom_css = """
-    .main-header {
-        text-align: center;
-        padding: 2rem;
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        color: white;
-        border-radius: 10px;
-        margin-bottom: 2rem;
-    }
-    .disclaimer-box {
-        background-color: #f8f9fa;
-        border-left: 4px solid #17a2b8;
-        padding: 1rem;
-        margin: 1rem 0;
-        border-radius: 5px;
-    }
+    Returns:
+        str: Formatted list of crisis intervention resources and emergency contacts
+    """
+    return """🆘 IMMEDIATE CRISIS RESOURCES:
+
+**If you're in immediate danger, call 911 or your local emergency number.**
+
+**24/7 Crisis Support:**
+• 988 Suicide & Crisis Lifeline: Call or text 988 (US)
+• Crisis Text Line: Text HOME to 741741 (US)
+• International Association for Suicide Prevention: https://iasp.info/resources/Crisis_Centres/
+
+**Specialized Support:**
+• National Domestic Violence Hotline: 1-800-799-7233
+• SAMHSA National Helpline: 1-800-662-4357 (mental health & substance abuse)
+• Trans Lifeline: 877-565-8860
+• LGBT National Hotline: 1-888-843-4564
+• Veterans Crisis Line: 1-800-273-8255
+
+You are not alone. Help is available 24/7."""
+
+def mindfulness_exercise() -> str:
     """
+    Get a guided mindfulness or grounding exercise for immediate stress relief.
     
-    with gr.Blocks(css=custom_css, title="Psychology AI Assistant", theme=gr.themes.Soft()) as demo:
-        
-        # Header
-        gr.HTML("""
-        <div class="main-header">
-            <h1>🧠 Psychology AI Assistant</h1>
-            <p>A supportive AI trained in psychology and mental health</p>
-            <p><em>Powered by fine-tuned Qwen 2.5-0.5B</em></p>
-        </div>
-        """)
-        
-        # Disclaimer
-        gr.HTML("""
-        <div class="disclaimer-box">
-            <strong>⚠️ Important Disclaimer:</strong> This AI assistant provides supportive guidance based on psychological principles, 
-            but it is not a replacement for professional therapy or medical advice. If you're experiencing a mental health crisis, 
-            please contact a mental health professional or crisis hotline immediately.
-            <br><br>
-            <strong>Crisis Resources:</strong>
-            <ul>
-                <li><strong>988 Suicide & Crisis Lifeline:</strong> Call or text 988</li>
-                <li><strong>Crisis Text Line:</strong> Text HOME to 741741</li>
-                <li><strong>Emergency:</strong> Call 911 or your local emergency number</li>
-            </ul>
-        </div>
-        """)
-        
-        with gr.Row():
-            with gr.Column(scale=3):
-                # Main chat interface
-                chatbot = gr.Chatbot(
-                    height=500,
-                    placeholder="👋 Hi! I'm here to provide supportive guidance and practical strategies for mental wellness. What's on your mind today?",
-                    avatar_images=("🧑‍💼", "🧠")
-                )
-                
-                with gr.Row():
-                    message_input = gr.Textbox(
-                        placeholder="Share what's on your mind...",
-                        container=False,
-                        scale=4,
-                        lines=2
-                    )
-                    send_btn = gr.Button("Send", variant="primary", scale=1)
-                
-                # Simplified example prompts
-                gr.HTML("<h3>💡 Try these conversation starters:</h3>")
-                
-                with gr.Row():
-                    example1 = gr.Button("I feel overwhelmed with work and don't know how to manage everything", size="sm")
-                    example2 = gr.Button("I can't stop worrying about things that might go wrong", size="sm")
-                
-                with gr.Row():
-                    example3 = gr.Button("I have trouble setting boundaries with people", size="sm")
-                    example4 = gr.Button("I constantly criticize myself and focus on my mistakes", size="sm")
-                
-                with gr.Row():
-                    example5 = gr.Button("I'm going through a major life transition and feel lost", size="sm")
-                    example6 = gr.Button("I have panic attacks before important meetings", size="sm")
-            
-            with gr.Column(scale=1):
-                # Settings panel
-                gr.HTML("<h3>⚙️ Settings</h3>")
-                
-                prompt_style = gr.Dropdown(
-                    choices=["Therapeutic", "Supportive Friend", "Crisis Support", "Anxiety Coach", "Mindfulness Guide"],
-                    value="Therapeutic",
-                    label="Response Style",
-                    info="Choose the type of support you prefer"
-                )
-                
-                temperature = gr.Slider(
-                    minimum=0.1,
-                    maximum=1.0,
-                    value=0.8,
-                    step=0.1,
-                    label="Creativity",
-                    info="Higher = more creative responses"
-                )
-                
-                max_tokens = gr.Slider(
-                    minimum=100,
-                    maximum=500,
-                    value=300,
-                    step=50,
-                    label="Response Length",
-                    info="Maximum response length"
-                )
-                
-                clear_btn = gr.Button("🗑️ Clear Chat", variant="secondary")
-                
-                # Model info
-                gr.HTML("""
-                <div style="margin-top: 2rem; padding: 1rem; background-color: #f8f9fa; border-radius: 5px;">
-                    <h4>📊 Model Info</h4>
-                    <p><strong>Base Model:</strong> Qwen/Qwen2.5-0.5B</p>
-                    <p><strong>Fine-tuned:</strong> phxdev/psychology-qwen-0.5b</p>
-                    <p><strong>Specialization:</strong> Psychology & Mental Health</p>
-                    <p><strong>Training:</strong> PEFT/LoRA</p>
-                </div>
-                """)
-        
-        # Chat functionality
-        def respond(message, history, temp, max_tok, style):
-            if not message.strip():
-                return history, ""
-            
-            # Add user message
-            history = history + [[message, None]]
-            
-            # Generate response
-            bot_response = generate_response(message, history, temp, max_tok, style)
-            
-            # Add bot response
-            history[-1][1] = bot_response
-            
-            return history, ""
-        
-        def clear_chat():
-            return [], ""
-        
-        def set_example(text):
-            return text
-        
-        # Event handlers
-        send_btn.click(
-            respond,
-            inputs=[message_input, chatbot, temperature, max_tokens, prompt_style],
-            outputs=[chatbot, message_input]
-        )
-        
-        message_input.submit(
-            respond,
-            inputs=[message_input, chatbot, temperature, max_tokens, prompt_style],
-            outputs=[chatbot, message_input]
-        )
-        
-        clear_btn.click(clear_chat, outputs=[chatbot, message_input])
-        
-        # Example button handlers (simplified)
-        example1.click(set_example, inputs=[gr.State("I feel overwhelmed with work and don't know how to manage everything")], outputs=[message_input])
-        example2.click(set_example, inputs=[gr.State("I can't stop worrying about things that might go wrong")], outputs=[message_input])
-        example3.click(set_example, inputs=[gr.State("I have trouble setting boundaries with people")], outputs=[message_input])
-        example4.click(set_example, inputs=[gr.State("I constantly criticize myself and focus on my mistakes")], outputs=[message_input])
-        example5.click(set_example, inputs=[gr.State("I'm going through a major life transition and feel lost")], outputs=[message_input])
-        example6.click(set_example, inputs=[gr.State("I have panic attacks before important meetings")], outputs=[message_input])
-        
-        # Footer
-        gr.HTML("""
-        <div style="text-align: center; margin-top: 2rem; padding: 1rem; color: #666;">
-            <p>Built with ❤️ using Gradio • Fine-tuned by @phxdev</p>
-            <p><strong>For Mark. For Josh. For everyone who needs someone.</strong></p>
-        </div>
-        """)
+    Provides step-by-step mindfulness and grounding exercises that can be done 
+    anywhere to help manage anxiety, panic, or overwhelming emotions.
+    
+    Returns:
+        str: Step-by-step guided mindfulness or grounding exercise
+    """
     
-    return demo
+    return """🧘 5-MINUTE GROUNDING EXERCISE:
+
+**Right now, notice:**
+1. **5 things you can SEE** around you (colors, shapes, objects)
+2. **4 things you can TOUCH** (chair, table, your clothes, air temperature)  
+3. **3 things you can HEAR** (traffic, breathing, birds, AC)
+4. **2 things you can SMELL** (coffee, soap, air freshener)
+5. **1 thing you can TASTE** (gum, coffee, or just your mouth)
+
+**Breathe slowly:** In for 4 counts, hold for 4, out for 6.
 
-# Initialize the model
-print("���� Starting Psychology AI Assistant...")
+**Box Breathing:**
+• Inhale for 4 counts
+• Hold for 4 counts  
+• Exhale for 4 counts
+• Hold empty for 4 counts
+• Repeat 4-6 times
+
+You are safe. You are here. This moment will pass."""
+
+# Initialize model
+print("🚀 Starting Psychology Personal Agent...")
 model_loaded = load_model()
 
 if model_loaded:
-    print("✅ Model loaded successfully!")
-    demo = create_interface()
+    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"
+        ]
+    )
     
     if __name__ == "__main__":
         demo.launch(
-            share=True,
-            server_name="0.0.0.0",
+            mcp_server=True,  # MCP enabled
+            share=False,
+            server_name="0.0.0.0", 
             server_port=7860,
             show_error=True
         )
+        
 else:
-    print("❌ Failed to load model. Creating error interface...")
-    
-    with gr.Blocks() as demo:
-        gr.HTML("""
-        <div style="text-align: center; padding: 2rem;">
-            <h1>⚠️ Model Loading Error</h1>
-            <p>Sorry, there was an issue loading the psychology model.</p>
-            <p>Please try refreshing the page or contact support.</p>
-        </div>
-        """)
+    print("❌ Failed to load psychology model")
+    demo = gr.Interface(
+        fn=lambda x: "⚠️ Model loading error. Please try again later.",
+        inputs="text",
+        outputs="text",
+        title="⚠️ Personal Agent - Loading Error"
+    )
     
     if __name__ == "__main__":
         demo.launch()