Update app.py

app.py

@@ -44,50 +44,56 @@ st.markdown(
 # Add the blurred background div
 st.markdown('<div class="blurred-background"></div>', unsafe_allow_html=True)
 
+# Load dataset for context
 @st.cache_resource
-def load_pipeline():
-    return pipeline("text-generation", model="QuantFactory/Mental-Health-FineTuned-Mistral-7B-Instruct-v0.2-GGUF")
+def load_counseling_dataset():
+    return load_dataset("Amod/mental_health_counseling_conversations")
 
-text_generation_pipe = load_pipeline()
+dataset = load_counseling_dataset()
 
-# Load dataset for reference (optional)
+# Load a Hugging Face-compatible text-generation model
 @st.cache_resource
-def load_dataset_reference():
-    return load_dataset("Amod/mental_health_counseling_conversations")
+def load_text_generation_model():
+    return pipeline("text-generation", model="meta-llama/Llama-2-7b-chat-hf")
 
-dataset = load_dataset_reference()
+text_generator = load_text_generation_model()
 
-# Streamlit App Configuration
-st.title("Mental Well-Being Support Application")
+# Streamlit app
+st.title("Mental Health Counseling Chat")
 st.markdown("""
-Welcome to the Mental Well-Being Support Application. This platform is designed to provide positive, supportive, and encouraging responses to your mental health concerns. Our responses are powered by a fine-tuned AI model based on expert psychologists' answers.
+Welcome to the Mental Health Counseling Chat application. 
+This platform is designed to provide supportive, positive, and encouraging responses based on mental health counseling expertise.
 """)
 
-# User Input Section
-st.header("Your Mental Health Journey")
-user_query = st.text_area("Please share your thoughts or questions:", placeholder="Write here...")
+# Explore dataset for additional context or resources (optional)
+if st.checkbox("Show Example Questions and Answers from Dataset"):
+    sample = dataset["train"].shuffle(seed=42).select(range(3))  # Display 3 random samples
+    for example in sample:
+        st.markdown(f"**Question:** {example['context']}")
+        st.markdown(f"**Answer:** {example['response']}")
+        st.markdown("---")
 
-# Generate AI Response
-if st.button("Get Supportive Response"):
-    if user_query.strip():
-        # Generate response
-        inputs = tokenizer(f"User: {user_query}\nAI:", return_tensors="pt")
-        outputs = model.generate(inputs.input_ids, max_length=200, temperature=0.7, num_return_sequences=1)
-        ai_response = tokenizer.decode(outputs[0], skip_special_tokens=True).split("AI:")[-1].strip()
+# User input for mental health concerns
+user_input = st.text_area("Your question or concern:", placeholder="Type here...")
 
-        # Display the response
-        st.subheader("Your Supportive Response:")
-        st.write(ai_response)
+if st.button("Get Supportive Response"):
+    if user_input.strip():
+        # Generate response using the model
+        prompt = f"User: {user_input}\nCounselor:"
+        response = text_generator(prompt, max_length=200, num_return_sequences=1)
+        counselor_reply = response[0]["generated_text"].split("Counselor:")[-1].strip()
+        
+        st.subheader("Counselor's Response:")
+        st.write(counselor_reply)
     else:
-        st.error("Please enter a question or concern to get a response.")
+        st.error("Please enter a question or concern to receive a response.")
 
-# Additional Resources Section
-st.sidebar.header("Resources")
+# Sidebar resources
+st.sidebar.header("Additional Mental Health Resources")
 st.sidebar.markdown("""
 - [Mental Health Foundation](https://www.mentalhealth.org)
 - [Mind](https://www.mind.org.uk)
 - [National Suicide Prevention Lifeline](https://suicidepreventionlifeline.org)
 """)
 
-# Footer
-st.sidebar.info("This application is not a replacement for professional help. If you're in crisis, please contact a mental health professional.")
+st.sidebar.info("This application is not a replacement for professional counseling. If you are in crisis, please seek professional help immediately.")