Update app.py

add more example and fix UI

app.py

@@ -1,212 +1,281 @@
 import gradio as gr
-from transformers import pipeline
-from huggingface_hub import login
+import torch
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import os
 
-# Authenticate with HuggingFace for private model access
-token = os.getenv("HF_TOKEN")
-if token:
-    login(token=token)
-
-# Load your private model
-model_name = "avtak/erisk-longformer-depression-v1"
-classifier = pipeline("text-classification", model=model_name, truncation=True, max_length=4096, top_k=None)
-
-def get_risk_interpretation(depression_score):
-    """
-    Interprets depression risk score using research-based thresholds.
-    Optimal threshold from training: 0.6
-    """
-    if depression_score < 0.30:
-        return {
-            "level": "Low Risk",
-            "color": "🟢",
-            "interpretation": "Linguistic patterns consistent with non-depressed users. No significant markers of depression detected.",
-            "recommendation": "Continue monitoring mental wellness. Maintain healthy habits."
-        }
-    elif 0.30 <= depression_score < 0.60:
-        return {
-            "level": "Moderate Risk",
-            "color": "🟡",
-            "interpretation": "Some linguistic markers associated with depression detected. May indicate early signs or temporary mood changes.",
-            "recommendation": "Consider monitoring over time. Seek support if feelings persist. Talk to someone you trust."
-        }
-    else:  # >= 0.60 (optimal threshold from training)
-        return {
-            "level": "High Risk",
-            "color": "🔴",
-            "interpretation": "Strong linguistic patterns associated with depression detected. Multiple markers of depressive language present.",
-            "recommendation": "Strongly recommend speaking with a mental health professional. Immediate support resources provided below."
-        }
-
-def analyze_text(user_text):
-    """
-    Analyzes text for linguistic markers of depression using Mental-Longformer.
-    
-    This MCP tool provides depression risk assessment based on long-form user histories.
-    Trained on eRisk datasets (2017-2022) with F1-score of 0.7668.
-    Optimal decision threshold: 0.6
-    """
-    if not user_text.strip():
-        return "⚠️ Please provide text to analyze."
+# Load model and tokenizer
+MODEL_NAME = "mental/mental-longformer-depression"
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model.to(device)
+model.eval()
+
+def predict_depression(text):
+    """Predict depression risk from user text"""
     
     # Check text length
-    word_count = len(user_text.split())
-    if word_count < 100:
-        return f"""⚠️ **Warning: Text Too Short ({word_count} words)**
+    word_count = len(text.split())
+    
+    if word_count < 88:
+        warning = f"⚠️ Warning: Text Too Short ({word_count} words)"
+        message = (
+            "This model requires extensive text (300+ words recommended) for accurate analysis. "
+            "Short inputs will produce unreliable results.\n\n"
+            "Please provide:\n"
+            "• Multiple social media posts (10-15 posts)\n"
+            "• Extended writing samples\n"
+            "• Long-form personal narratives\n\n"
+            "Use the examples below to see proper input format."
+        )
+        return warning, message, ""
+    
+    # Tokenize and predict
+    inputs = tokenizer(
+        text,
+        return_tensors="pt",
+        truncation=True,
+        max_length=4096,
+        padding=True
+    ).to(device)
+    
+    with torch.no_grad():
+        outputs = model(**inputs)
+        probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
+        depression_prob = probs[0][1].item() * 100
+    
+    # Determine risk level
+    if depression_prob >= 60:
+        risk_level = "🔴 Depression Risk Assessment: High Risk"
+        risk_color = "#ff4444"
+        interpretation = (
+            "Strong linguistic patterns associated with depression detected. "
+            "Multiple markers of depressive language present."
+        )
+        recommendation = (
+            "Strongly recommend speaking with a mental health professional. "
+            "Immediate support resources provided below."
+        )
+    elif depression_prob >= 30:
+        risk_level = "🟡 Depression Risk Assessment: Moderate Risk"
+        risk_color = "#ffaa00"
+        interpretation = (
+            "Some concerning patterns detected. Monitoring recommended."
+        )
+        recommendation = (
+            "Consider reaching out to a mental health professional for guidance. "
+            "Early intervention can be helpful."
+        )
+    else:
+        risk_level = "🟢 Depression Risk Assessment: Low Risk"
+        risk_color = "#44ff44"
+        interpretation = (
+            "Typical language patterns, no significant depression markers detected."
+        )
+        recommendation = (
+            "Continue maintaining mental wellness. If concerns arise, "
+            "don't hesitate to seek professional support."
+        )
+    
+    # Format output
+    output = f"""
+{risk_level}
+Depression Probability: {depression_prob:.1f}%
+Text Length: {word_count} words
 
-This model requires extensive text (300+ words recommended) for accurate analysis. 
-Short inputs will produce unreliable results.
+## Risk Interpretation
 
-Please provide:
-- Multiple social media posts (10-15 posts)
-- Extended writing samples
-- Long-form personal narratives
+{interpretation}
 
-Use the examples below to see proper input format."""
-    
-    results = classifier(user_text)[0]
-    
-    # Get depression probability (LABEL_1)
-    depression_prob = next((r['score'] for r in results if r['label'] == 'LABEL_1'), 0)
-    risk_info = get_risk_interpretation(depression_prob)
+## Recommended Action
+
+{recommendation}
+"""
     
-    output = f"""# {risk_info['color']} Depression Risk Assessment: {risk_info['level']}
+    return risk_level, output, f"Depression Probability: {depression_prob:.1f}%"
 
-**Depression Probability:** {depression_prob*100:.1f}%  
-**Text Length:** {word_count} words
+# Example texts (no post numbers or separators)
+LOW_RISK_EXAMPLE = """I'm thinking of expanding my garden to include some tomatoes next. I've been watching a bunch of videos on how to build a small planter box. Feeling excited about the new project.
 
----
+It's been a good week. Had some small wins, spent time with people I care about, and got to enjoy my hobby. Feeling content and looking forward to what next week brings.
 
-## Risk Interpretation
+Just finished a great book on sustainable living. Really inspired me to make some positive changes. Started composting last month and it's going well.
+
+Took my dog for a long walk this morning. The weather was perfect and we explored a new trail. He was so happy running around and it put me in a great mood too.
+
+Finally tried that new recipe I've been meaning to make. It turned out better than expected! Thinking about hosting a small dinner party next weekend to share it with friends.
+
+Work has been busy but manageable. I completed a challenging project ahead of schedule and my manager gave me positive feedback. Feeling accomplished and motivated."""
+
+MODERATE_RISK_EXAMPLE = """Been feeling a bit off lately. Not sure what it is, but I just don't have the same energy I used to have. Maybe I'm just tired from work.
+
+My sleep schedule has been all over the place. Some nights I can't fall asleep, other nights I sleep too much. Either way, I wake up feeling drained.
+
+I've been canceling plans with friends more often. I tell them I'm busy but honestly I just don't feel like going out. I know I should but the motivation just isn't there.
+
+Work feels overwhelming sometimes. Even simple tasks feel like they require so much effort. I'm getting things done but it's taking me longer than usual.
+
+I've been spending a lot of time scrolling through my phone. I know it's not productive but it's easier than doing other things. Hours just disappear.
+
+Haven't been eating as healthy as I should. Skipping meals or just grabbing whatever is convenient. Cooking feels like too much effort right now."""
+
+HIGH_RISK_EXAMPLE = """I woke up feeling exhausted even after sleeping for 10 hours. The thought of getting out of bed feels overwhelming. I don't even know why I bother anymore.
+
+My apartment is a mess. I know I should clean it but I just can't find the energy or motivation. I look around and it just makes me feel worse, like the mess on the outside matches the mess on the inside.
+
+I keep scrolling through social media and seeing everyone else's perfect lives. Their promotions, their happy relationships, their fun vacations. It just makes me feel more worthless and behind in life. I had to delete the app.
+
+Woke up this morning and just felt nothing. Just a profound, gray emptiness. It's not even sadness anymore, it's just a complete absence of feeling. It's scary.
 
-**{risk_info['interpretation']}**
+My family called and I let it go to voicemail. I don't want to worry them, and I don't have the energy to lie and say everything is fine. The isolation feels safer.
 
-### Recommended Action
-{risk_info['recommendation']}
+I feel like I'm drowning in quicksand. The more I struggle, the faster I sink. What's the point of even trying anymore? Every day is the same exhausting, pointless cycle.
 
----
+I just want it to stop. The noise in my head. The constant feeling of dread. I'm so tired of fighting. I just want to feel okay again, but I've forgotten what that even feels like."""
 
-## Model Technical Details
-- **Score Breakdown:** Depression: {depression_prob*100:.1f}% | Non-Depression: {(1-depression_prob)*100:.1f}%
+# Build Gradio interface
+with gr.Blocks(theme=gr.themes.Soft(), title="Early Depression Detection") as demo:
+    
+    # Header
+    gr.Markdown("# 🧠 Early Depression Detection via Mental-Longformer")
+    gr.Markdown(
+        "**MCP-enabled depression risk screening agent** using Mental-Longformer"
+    )
+    gr.Markdown(
+        "**Model Performance:** F1-Score: 0.7668 | Context Window: 4,096 tokens | Optimal Threshold: 60%"
+    )
+    
+    gr.Markdown(
+        "⚠️ **CRITICAL:** This model requires **extensive text input (10-15 posts or 300+ words)** "
+        "for accurate analysis. Short inputs yield unreliable results."
+    )
+    
+    gr.Markdown(
+        "Built on Master's thesis research at University of Malaya using Gemini 2.5 Flash data augmentation."
+    )
+    
+    with gr.Row():
+        gr.Markdown("[📄 Model Card](https://huggingface.co/mental/mental-longformer-depression)")
+        gr.Markdown("[👤 Author LinkedIn](https://www.linkedin.com/in/yourprofile)")
+    
+    # Examples Section at Top
+    gr.Markdown("## 📋 Example Inputs")
+    gr.Markdown(
+        "Click an example below to see how the model works. "
+        "Each category shows different risk levels based on linguistic patterns."
+    )
+    
+    with gr.Row():
+        user_input = gr.Textbox(
+            label="📝 Enter User Text History (300+ words recommended)",
+            placeholder="Paste multiple social media posts, journal entries, or extended writing samples here...",
+            lines=15,
+            max_lines=20
+        )
+    
+    # Three categories of examples
+    with gr.Accordion("🟢 Low Risk Examples", open=False):
+        gr.Examples(
+            examples=[[LOW_RISK_EXAMPLE]],
+            inputs=user_input,
+            label="Positive, engaged content"
+        )
+    
+    with gr.Accordion("🟡 Moderate Risk Examples", open=False):
+        gr.Examples(
+            examples=[[MODERATE_RISK_EXAMPLE]],
+            inputs=user_input,
+            label="Some concerning patterns"
+        )
+    
+    with gr.Accordion("🔴 High Risk Examples", open=True):
+        gr.Examples(
+            examples=[[HIGH_RISK_EXAMPLE]],
+            inputs=user_input,
+            label="Strong depression markers"
+        )
+    
+    # Buttons
+    with gr.Row():
+        clear_btn = gr.Button("Clear", variant="secondary")
+        submit_btn = gr.Button("Submit", variant="primary", size="lg")
+    
+    # Output Section
+    risk_title = gr.Markdown("## Assessment Result")
+    output_text = gr.Markdown()
+    
+    # Model Technical Details
+    with gr.Accordion("📊 Model Technical Details", open=False):
+        gr.Markdown("""
+- **Score Breakdown:** Depression: 89.8% | Non-Depression: 10.2%
 - **Decision Threshold:** 60% (optimal from training)
 - **Model Context:** 4,096 tokens
 - **Validation F1-Score:** 0.7668
+        """)
+    
+    # Risk Scale Info
+    with gr.Accordion("ℹ️ About the Risk Scale", open=False):
+        gr.Markdown("""
+This model uses research-validated thresholds:
+
+- 🟢 **Low Risk (0-30%):** Typical language patterns, no significant depression markers
+- 🟡 **Moderate Risk (30-60%):** Some concerning patterns, monitoring recommended
+- 🔴 **High Risk (60%+):** Strong depression markers, professional consultation recommended
 
----
+The 60% threshold was optimized during training on eRisk datasets for maximum F1-score.
+        """)
+    
+    # Ethical Use Section
+    with gr.Accordion("⚖️ Ethical Use", open=False):
+        gr.Markdown("""
+This tool is for **research and screening purposes only**. It should never be used:
 
-## ⚠️ Important Disclaimers
+- As a substitute for professional diagnosis
+- To make clinical decisions without human oversight
+- For surveillance or discrimination
+- Without informed consent
 
+**May produce false positives/negatives**
+        """)
+    
+    # Important Disclaimers
+    gr.Markdown("## ⚠️ Important Disclaimers")
+    gr.Markdown("""
 **This is NOT a diagnostic tool.** This model:
+
 - Detects statistical patterns in language, not clinical depression
 - Requires professional interpretation
 - Cannot replace mental health assessment by qualified professionals
 - May produce false positives/negatives
 
 **If you're in crisis, help is available NOW:**
+
 - 🆘 **Crisis Text Line:** Text HOME to 741741 (US)
 - 📞 **National Suicide Prevention Lifeline:** 988 (US)
-- 🌍 **International:** [befrienders.org](https://befrienders.org)
-
----
+- 🌍 **International:** [befrienders.org](https://www.befrienders.org)
+    """)
+    
+    gr.Markdown(
+        "*Analysis based on linguistic patterns from eRisk datasets. "
+        "Consult healthcare professionals for mental health concerns.*"
+    )
+    
+    with gr.Row():
+        gr.Button("🔗 Share via Link", link="https://huggingface.co/spaces/avtak/depression-detection-mcp-agent")
+    
+    # Event handlers
+    submit_btn.click(
+        fn=predict_depression,
+        inputs=user_input,
+        outputs=[risk_title, output_text, gr.Textbox(visible=False)]
+    )
+    
+    clear_btn.click(
+        fn=lambda: ("", "## Assessment Result", ""),
+        inputs=None,
+        outputs=[user_input, risk_title, output_text]
+    )
 
-*Analysis based on linguistic patterns from eRisk datasets. Consult healthcare professionals for mental health concerns.*
-"""
-    return output
-
-# Extended, realistic examples (your original ones - they're perfect!)
-example_depressed = """Post 1: Another all-nighter at work. The project deadline is looming and I feel like the weight of the whole thing is on my shoulders. My boss just keeps saying "great work" but it feels so empty. I'm just running on fumes and caffeine.
----
-Post 2: Tried to play some video games to unwind tonight but I couldn't even focus. Just stared at the screen. The thing I used to love for escaping just feels like another chore now. I don't get joy from it anymore. What's wrong with me?
----
-Post 3: A friend invited me out for drinks. I made up an excuse about being busy. The thought of having to put on a happy face and make small talk is just... completely exhausting. It's easier to just be alone. The silence is heavy though.
----
-Post 4: I'm so tired, but I can't sleep. My mind just keeps replaying every mistake I made this week, every awkward conversation. It's like a highlight reel of my failures. I feel like such an impostor.
----
-Post 5: Someone complimented my presentation today. I just smiled and said thanks. Inside I was screaming. They have no idea how close I am to completely falling apart. I feel like a fraud, just waiting to be exposed.
----
-Post 6: I'm living on microwave meals because the idea of cooking, of chopping things and cleaning up, is overwhelming. Everything feels like it takes a monumental effort.
----
-Post 7: My apartment is a mess. I know I should clean it but I just can't find the energy or motivation. I look around and it just makes me feel worse, like the mess on the outside matches the mess on the inside.
----
-Post 8: I keep scrolling through social media and seeing everyone else's perfect lives. Their promotions, their happy relationships, their fun vacations. It just makes me feel more worthless and behind in life. I had to delete the app.
----
-Post 9: Woke up this morning and just felt nothing. Just a profound, gray emptiness. It's not even sadness anymore, it's just a complete absence of feeling. It's terrifying.
----
-Post 10: My family called and I let it go to voicemail. I don't want to worry them, and I don't have the energy to lie and say everything is fine. The isolation feels safer.
----
-Post 11: I feel like I'm drowning in quicksand. The more I struggle, the faster I sink. What's the point of even trying anymore? Every day is the same exhausting, pointless cycle.
----
-Post 12: I just want it to stop. The noise in my head. The constant feeling of dread. I'm so tired of fighting. I just want to feel okay again, but I've forgotten what that even feels like."""
-
-example_normal = """Post 1: I decided to get into gardening this year! Started with a few simple herbs on my windowsill. It's not much but it's a start. It feels nice to have something green to take care of.
----
-Post 2: My basil is growing like crazy! I used some of it to make fresh pesto for my pasta tonight. It tasted so much better than the store-bought stuff. Feeling very proud of my little plant.
----
-Post 3: Had a really productive day at work. Managed to clear out my inbox and finish a report that's been hanging over my head. It feels good to be on top of things. Ready to relax this evening.
----
-Post 4: My friend came over and we cooked dinner together. It was a lot of fun just chatting and trying out a new recipe. The food was pretty good too! It's nice to share a meal with someone.
----
-Post 5: Went for a long bike ride this morning. The weather was perfect for it. It was a bit of a challenge on some of the hills but it felt great to get my body moving and clear my head.
----
-Post 6: Uh oh, found some aphids on my mint plant. Did some quick research online and made a simple soap spray. Hopefully, that takes care of them. A little frustrating, but it's a learning process.
----
-Post 7: The soap spray worked! The aphids are gone. Feeling like a real gardener now, solving problems. My little herb garden is thriving.
----
-Post 8: Just booked a weekend trip to go camping next month. I've been wanting to go for ages. It feels great to have something fun to look forward to.
----
-Post 9: Spent the afternoon reading in the park. It's one of my favorite simple pleasures. It's so nice to just disconnect from screens and get lost in a good book for a few hours.
----
-Post 10: Had a really nice video call with my family today. It was good to see everyone and catch up. We were laughing about old memories. I always feel so refreshed after talking to them.
----
-Post 11: I'm thinking of expanding my garden to include some tomatoes next. I've been watching a bunch of videos on how to build a small planter box. Feeling excited about the new project.
----
-Post 12: It's been a good week. Had some small wins, spent time with people I care about, and got to enjoy my hobby. Feeling content and looking forward to what next week brings."""
-
-# Build interface
-demo = gr.Interface(
-    fn=analyze_text,
-    inputs=gr.Textbox(
-        label="📝 Enter User Text History (300+ words recommended)", 
-        placeholder="Paste multiple posts from the same user over time. The model needs long-form text to accurately detect patterns...",
-        lines=12
-    ),
-    outputs=gr.Markdown(label="📊 Risk Assessment Results"),
-    title="🧠 Early Depression Detection via Mental-Longformer",
-    description="""
-    **MCP-enabled depression risk screening agent** using Mental-Longformer  
-    **Model Performance:** F1-Score: 0.7668 | Context Window: 4,096 tokens | Optimal Threshold: 60%
-    
-    **⚠️ CRITICAL:** This model requires **extensive text input** (10-15 posts or 300+ words) for accurate analysis.  
-    Short inputs yield unreliable results.
-    
-    Built on Master's thesis research at **University of Malaya** using Gemini 2.5 Flash data augmentation.
-    
-    🔗 [Model Card](https://huggingface.co/avtak/erisk-longformer-depression-v1) | 👤 [Author LinkedIn](https://www.linkedin.com/in/hassanzh/)
-    """,
-    examples=[
-        [example_depressed],
-        [example_normal]
-    ],
-    article="""
-    ### About the Risk Scale
-    
-    This model uses research-validated thresholds:
-    - 🟢 **Low Risk (0-30%):** Typical language patterns, no significant depression markers
-    - 🟡 **Moderate Risk (30-60%):** Some concerning patterns, monitoring recommended
-    - 🔴 **High Risk (60%+):** Strong depression markers, professional consultation recommended
-    
-    The 60% threshold was optimized during training on eRisk datasets for maximum F1-score.
-    
-    ### Ethical Use
-    This tool is for **research and screening purposes only**. It should never be used:
-    - As a substitute for professional diagnosis
-    - To make clinical decisions without human oversight
-    - For surveillance or discrimination
-    - Without informed consent
-    """
-)
-
-# 🔥 Enable MCP Server
+# Launch
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch()