Update app.py

app.py

@@ -43,45 +43,44 @@
 
 
 import streamlit as st
+from transformers import pipeline, AutoTokenizer
 
+# ✅ Load Emotion Recognition Model
+emotion_pipeline = pipeline("text-classification", model="distilbert-base-uncased-finetuned-emotion")
+
+# ✅ Load Stress Detection Model
+stress_pipeline = pipeline("text-classification", model="mental-health-roberta")
+
+# ✅ Load Mental Disorder Detection Model
+mental_bert_pipeline = pipeline("text-classification", model="mental-bert")
+
+# ✅ Load PHQ-9 Depression Severity Classifier
+phq9_pipeline = pipeline("text-classification", model="PHQ-9 Depression Classifier")
+
+# ✅ Load Chatbot Model (DeepSeek)
+deepseek_model = "deepseek-ai/deepseek-llm-7b"
+deepseek_tokenizer = AutoTokenizer.from_pretrained(deepseek_model)
+deepseek_pipeline = pipeline("text-generation", model=deepseek_model, tokenizer=deepseek_tokenizer)
+
+# 🏥 Streamlit UI
 st.title("🧠 Mental Health Assistant Bot")
 
-# User Input
 user_input = st.text_input("How are you feeling today?", "")
 
 if st.button("Submit"):
     if user_input:
-        # Get Emotion Analysis
+        # ✅ Emotion Analysis
         emotion_result = emotion_pipeline(user_input)[0]
         st.write(f"**Emotion Detected:** {emotion_result['label']} ({emotion_result['score']:.2f})")
 
-        # Get Mental Health Condition Analysis
-        mental_health_result = mental_bert_pipeline(user_input)[0]
-        st.write(f"**Possible Mental Health Condition:** {mental_health_result['label']} ({mental_health_result['score']:.2f})")
-
-        # Get Stress Level Analysis
+        # ✅ Stress Level Analysis
         stress_result = stress_pipeline(user_input)[0]
         st.write(f"**Stress Level:** {stress_result['label']} ({stress_result['score']:.2f})")
 
-        # Chatbot Response using DeepSeek AI
+        # ✅ Mental Health Condition Detection
+        mental_health_result = mental_bert_pipeline(user_input)[0]
+        st.write(f"**Possible Mental Health Condition:** {mental_health_result['label']} ({mental_health_result['score']:.2f})")
+
+        # ✅ AI Chatbot Response
         deepseek_response = deepseek_pipeline(user_input, max_length=100, do_sample=True)[0]['generated_text']
         st.write(f"🤖 **Chatbot:** {deepseek_response}")
-
-# Question Answering Section
-st.subheader("Ask Mental Health Questions")
-user_question = st.text_input("Ask me anything about mental health:", "")
-
-if st.button("Ask"):
-    if user_question:
-        answer = qa_pipeline(question=user_question, context="Mental health is important for overall well-being.")
-        st.write(f"**Answer:** {answer['answer']}")
-
-# PHQ-9 Depression Assessment
-st.subheader("Depression Severity Assessment (PHQ-9)")
-phq9_question = st.text_input("Describe your mood over the last two weeks:", "")
-
-if st.button("Analyze Depression Level"):
-    if phq9_question:
-        phq9_result = phq9_pipeline(phq9_question)[0]
-        st.write(f"**PHQ-9 Score Suggests:** {phq9_result['label']} ({phq9_result['score']:.2f})")
-