Update app.py

app.py

@@ -122,42 +122,41 @@ def _generate_response(formatted_prompt: str, max_new_tokens: int = 200) -> str:
         traceback.print_exc()
         return error_msg
 
-def psychology_support(message: str) -> str:
-    """Main psychology support function"""
+# EXPOSED MCP FUNCTIONS
+def psychology_support_exposed(message: str) -> str:
+    """Main psychology support function - EXPOSED TO MCP"""
     if not message.strip():
         return "Please share what's on your mind."
     
-    # Frame it as completing psychology content that happens to be conversational
     formatted_prompt = f"A person with boundary issues says: \"{message}\"\n\nA supportive response that helps with boundary setting:"
-    
     return _generate_response(formatted_prompt, max_new_tokens=150)
 
-def trauma_support(message: str) -> str:
-    """Trauma-informed support"""
+def trauma_support_exposed(message: str) -> str:
+    """Trauma-informed support - EXPOSED TO MCP"""
     if not message.strip():
         return "I'm here to listen. What would help you feel safer right now?"
     
     formatted_prompt = f"You are a trauma-informed friend who helps people feel safe. Draw on your vast experience of supporting trauma survivors. Someone dealing with trauma just told you: \"{message}\"\n\nRespond to help them feel safer:"
     return _generate_response(formatted_prompt, max_new_tokens=180)
 
-def panic_support(message: str) -> str:
-    """Immediate panic attack support"""
+def panic_support_exposed(message: str) -> str:
+    """Immediate panic attack support - EXPOSED TO MCP"""
     if not message.strip():
         return "I can help you through this. What are you experiencing right now?"
     
     formatted_prompt = f"You are a calm, grounding friend who helps with panic attacks. Draw on your vast experience of helping people through panic and anxiety. Someone having panic symptoms just told you: \"{message}\"\n\nRespond to help them through this moment:"
     return _generate_response(formatted_prompt, max_new_tokens=160)
 
-def process_emotion(message: str) -> str:
-    """Emotion processing and validation"""
+def process_emotion_exposed(message: str) -> str:
+    """Emotion processing and validation - EXPOSED TO MCP"""
     if not message.strip():
         return "What emotion are you working through right now?"
     
     formatted_prompt = f"You are an emotionally intelligent friend who helps people understand their feelings. Draw on your vast experience of emotional support and validation. Someone struggling with emotions just told you: \"{message}\"\n\nRespond to help them process what they're feeling:"
     return _generate_response(formatted_prompt, max_new_tokens=170)
 
-def crisis_resources() -> str:
-    """Get immediate crisis intervention resources"""
+def crisis_resources_exposed() -> str:
+    """Get immediate crisis intervention resources - EXPOSED TO MCP"""
     return """🆘 **IMMEDIATE CRISIS RESOURCES**
 
 **Emergency:** Call 911 or your local emergency number
@@ -176,8 +175,8 @@ def crisis_resources() -> str:
 
 You are not alone. Help is available 24/7."""
 
-def mindfulness_exercise() -> str:
-    """Guided mindfulness exercise"""
+def mindfulness_exercise_exposed() -> str:
+    """Guided mindfulness exercise - EXPOSED TO MCP"""
     return """🧘 **5-MINUTE GROUNDING EXERCISE**
 
 **Right now, notice:**
@@ -195,157 +194,157 @@ def mindfulness_exercise() -> str:
 
 You've got this. One breath at a time."""
 
+def sentiment_analysis_exposed(text: str) -> str:
+    """Analyze sentiment and emotional content of text - EXPOSED TO MCP"""
+    if not text.strip():
+        return "Please provide text to analyze."
+    
+    formatted_prompt = f"You are an expert in emotional analysis and psychology. Analyze the emotional content, sentiment, and psychological patterns in this text: \"{text}\"\n\nProvide a detailed psychological analysis including mood, emotional state, potential concerns, and supportive insights:"
+    return _generate_response(formatted_prompt, max_new_tokens=200)
+
+def relationship_analysis_exposed(message: str) -> str:
+    """Analyze relationship dynamics and patterns - EXPOSED TO MCP"""
+    if not message.strip():
+        return "Please describe the relationship situation you'd like analyzed."
+    
+    formatted_prompt = f"You are a relationship therapist analyzing interpersonal dynamics. Someone described this relationship situation: \"{message}\"\n\nProvide insights on relationship patterns, communication dynamics, and healthy approaches:"
+    return _generate_response(formatted_prompt, max_new_tokens=180)
+
+def coping_strategies_exposed(situation: str) -> str:
+    """Generate personalized coping strategies - EXPOSED TO MCP"""
+    if not situation.strip():
+        return "Please describe what you're dealing with so I can suggest coping strategies."
+    
+    formatted_prompt = f"You are a mental health counselor specializing in coping strategies. Someone is dealing with: \"{situation}\"\n\nProvide specific, practical coping strategies and techniques:"
+    return _generate_response(formatted_prompt, max_new_tokens=200)
+
 # Apply GPU optimization if available
 if HAS_SPACES:
-    psychology_support = spaces.GPU(psychology_support)
-    trauma_support = spaces.GPU(trauma_support)
-    panic_support = spaces.GPU(panic_support)
-    process_emotion = spaces.GPU(process_emotion)
+    psychology_support_exposed = spaces.GPU(psychology_support_exposed)
+    trauma_support_exposed = spaces.GPU(trauma_support_exposed)
+    panic_support_exposed = spaces.GPU(panic_support_exposed)
+    process_emotion_exposed = spaces.GPU(process_emotion_exposed)
+    sentiment_analysis_exposed = spaces.GPU(sentiment_analysis_exposed)
+    relationship_analysis_exposed = spaces.GPU(relationship_analysis_exposed)
+    coping_strategies_exposed = spaces.GPU(coping_strategies_exposed)
 
 # Initialize model
 print("🚀 Starting Psychology Personal Agent...")
 model_loaded = load_model()
 
 if model_loaded:
-    print("✅ Creating interface...")
+    print("✅ Creating MCP-exposed interfaces...")
+    
+    # Create individual interfaces for each function
+    psychology_interface = gr.Interface(
+        fn=psychology_support_exposed,
+        inputs=gr.Textbox(label="Message", placeholder="Share what's on your mind..."),
+        outputs=gr.Textbox(label="Psychology Support Response"),
+        title="Psychology Support",
+        description="General psychological support and guidance"
+    )
+    
+    trauma_interface = gr.Interface(
+        fn=trauma_support_exposed,
+        inputs=gr.Textbox(label="Message", placeholder="Share what feels safe..."),
+        outputs=gr.Textbox(label="Trauma Support Response"),
+        title="Trauma Support", 
+        description="Trauma-informed care and support"
+    )
+    
+    panic_interface = gr.Interface(
+        fn=panic_support_exposed,
+        inputs=gr.Textbox(label="Message", placeholder="What are you experiencing?"),
+        outputs=gr.Textbox(label="Panic Support Response"),
+        title="Panic Support",
+        description="Immediate support for panic and anxiety"
+    )
+    
+    emotion_interface = gr.Interface(
+        fn=process_emotion_exposed,
+        inputs=gr.Textbox(label="Message", placeholder="Describe your emotions..."),
+        outputs=gr.Textbox(label="Emotion Processing Response"),
+        title="Emotion Processing",
+        description="Help processing and understanding emotions"
+    )
+    
+    sentiment_interface = gr.Interface(
+        fn=sentiment_analysis_exposed,
+        inputs=gr.Textbox(label="Text to Analyze", placeholder="Enter text for sentiment analysis..."),
+        outputs=gr.Textbox(label="Sentiment Analysis"),
+        title="Sentiment Analysis",
+        description="Analyze emotional content and psychological patterns"
+    )
+    
+    relationship_interface = gr.Interface(
+        fn=relationship_analysis_exposed,
+        inputs=gr.Textbox(label="Relationship Situation", placeholder="Describe the relationship dynamics..."),
+        outputs=gr.Textbox(label="Relationship Analysis"),
+        title="Relationship Analysis",
+        description="Analyze relationship patterns and dynamics"
+    )
+    
+    coping_interface = gr.Interface(
+        fn=coping_strategies_exposed,
+        inputs=gr.Textbox(label="Situation", placeholder="What are you dealing with?"),
+        outputs=gr.Textbox(label="Coping Strategies"),
+        title="Coping Strategies",
+        description="Personalized coping strategies and techniques"
+    )
+    
+    crisis_interface = gr.Interface(
+        fn=crisis_resources_exposed,
+        inputs=[],
+        outputs=gr.Textbox(label="Crisis Resources"),
+        title="Crisis Resources",
+        description="24/7 crisis intervention resources"
+    )
     
-    with gr.Blocks(title="🧠 Personal Agent - Psychology Support", theme=gr.themes.Soft()) as demo:
+    mindfulness_interface = gr.Interface(
+        fn=mindfulness_exercise_exposed,
+        inputs=[],
+        outputs=gr.Textbox(label="Mindfulness Exercise"),
+        title="Mindfulness Exercise",
+        description="Guided grounding and mindfulness techniques"
+    )
+    
+    # Main tabbed interface
+    with gr.Blocks(title="🧠 Psychology Personal Agent - Full Suite", theme=gr.themes.Soft()) as demo:
         
-        # Safety modal
-        with gr.Column(visible=True, elem_id="safety_modal") as safety_modal:
-            gr.Markdown("# ⚠️ IMPORTANT SAFETY INFORMATION")
-            
-            gr.Markdown("""## 🆘 CRISIS RESOURCES - AVAILABLE 24/7
-
-**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)
-• **Emergency Services**: Call 911 for immediate emergency
-
-## ⚠️ MEDICAL DISCLAIMER
-
-**THIS IS NOT MEDICAL ADVICE.** This AI provides general support only.
-
-**If you are in crisis or considering harm, seek immediate medical attention.**
-
-This is not a substitute for professional care. Always consult qualified healthcare providers for medical/mental health concerns.""")
-            
-            safety_acknowledged = gr.Checkbox(
-                label="I understand this is not medical advice and will seek professional help for serious concerns.",
-                value=False
-            )
-            
-            continue_button = gr.Button("Continue to Psychology Support", variant="primary", size="lg")
+        # Safety warning
+        gr.Markdown("# ⚠️ SAFETY NOTICE")
+        gr.Markdown("""**THIS IS NOT MEDICAL ADVICE.** For crisis situations, call 911 or 988.
+        This AI provides supportive guidance only - always consult professionals for serious concerns.""")
         
-        # Main interface
-        with gr.Column(visible=False) as main_interface:
-            gr.Markdown("# 🧠 Personal Agent - Psychology Support")
-            gr.Markdown("*Specialized mental health support • Evidence-based • Always here*")
-            
-            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", lines=6)
-                support_button = gr.Button("Get Support", variant="primary")
-                
-                support_button.click(
-                    fn=psychology_support,
-                    inputs=message_input,
-                    outputs=support_output
-                )
-                
-                gr.Examples(
-                    examples=[
-                        "I've been having panic attacks at work",
-                        "I can't stop worrying about everything", 
-                        "I feel like I'm not good enough",
-                        "I'm going through a major life change",
-                        "I have trouble setting boundaries"
-                    ],
-                    inputs=message_input
-                )
-            
-            with gr.Tab("Trauma Support"):
-                trauma_input = gr.Textbox(
-                    label="Share what feels safe to share", 
-                    placeholder="I'm here to help you feel safer...",
-                    lines=3
-                )
-                trauma_output = gr.Textbox(label="Trauma-Informed Response", lines=6)
-                trauma_button = gr.Button("Get Trauma Support", variant="secondary")
-                
-                trauma_button.click(
-                    fn=trauma_support,
-                    inputs=trauma_input,
-                    outputs=trauma_output
-                )
-            
-            with gr.Tab("Panic Support"):
-                panic_input = gr.Textbox(
-                    label="What are you experiencing?", 
-                    placeholder="I can help you through this...",
-                    lines=2
-                )
-                panic_output = gr.Textbox(label="Immediate Support", lines=6)
-                panic_button = gr.Button("Get Panic Support", variant="stop")
-                
-                panic_button.click(
-                    fn=panic_support,
-                    inputs=panic_input,
-                    outputs=panic_output
-                )
-            
-            with gr.Tab("Process Emotions"):
-                emotion_input = gr.Textbox(
-                    label="What emotion are you experiencing?", 
-                    placeholder="Describe what you're feeling...",
-                    lines=2
-                )
-                emotion_output = gr.Textbox(label="Emotion Guidance", lines=6)
-                emotion_button = gr.Button("Process Emotion", variant="secondary")
-                
-                emotion_button.click(
-                    fn=process_emotion,
-                    inputs=emotion_input,
-                    outputs=emotion_output
-                )
-            
-            # Resource tabs
-            with gr.Tab("Crisis Resources"):
-                crisis_button = gr.Button("Show Crisis Resources", variant="stop")
-                crisis_output = gr.Textbox(label="24/7 Crisis Resources", lines=10)
-                
-                crisis_button.click(
-                    fn=crisis_resources,
-                    outputs=crisis_output
-                )
-            
-            with gr.Tab("Mindfulness"):
-                mindfulness_button = gr.Button("Get Grounding Exercise", variant="secondary")
-                mindfulness_output = gr.Textbox(label="Mindfulness Exercise", lines=10)
-                
-                mindfulness_button.click(
-                    fn=mindfulness_exercise,
-                    outputs=mindfulness_output
-                )
+        gr.Markdown("# 🧠 Psychology Personal Agent - Complete Suite")
+        gr.Markdown("*Advanced psychological support with multiple specialized functions*")
         
-        # Modal behavior
-        def show_main_interface(acknowledged):
-            if acknowledged:
-                return gr.update(visible=False), gr.update(visible=True)
-            else:
-                return gr.update(visible=True), gr.update(visible=False)
+        with gr.Tab("🗣️ General Support"):
+            psychology_interface.render()
         
-        continue_button.click(
-            fn=show_main_interface,
-            inputs=[safety_acknowledged],
-            outputs=[safety_modal, main_interface]
-        )
+        with gr.Tab("🛡️ Trauma Support"):
+            trauma_interface.render()
+        
+        with gr.Tab("⚡ Panic Support"):
+            panic_interface.render()
+        
+        with gr.Tab("💭 Emotion Processing"):
+            emotion_interface.render()
+        
+        with gr.Tab("📊 Sentiment Analysis"):
+            sentiment_interface.render()
+        
+        with gr.Tab("👥 Relationship Analysis"):
+            relationship_interface.render()
+        
+        with gr.Tab("🎯 Coping Strategies"):
+            coping_interface.render()
+        
+        with gr.Tab("🆘 Crisis Resources"):
+            crisis_interface.render()
+        
+        with gr.Tab("🧘 Mindfulness"):
+            mindfulness_interface.render()
     
     if __name__ == "__main__":
         demo.launch(
@@ -359,7 +358,7 @@ This is not a substitute for professional care. Always consult qualified healthc
 else:
     print("❌ Failed to load psychology model")
     
-    def error_response(x):
+    def error_response(x=""):
         return "⚠️ Model loading error. Please check console logs and try restarting."
     
     demo = gr.Interface(