Update app.py

app.py

@@ -5,8 +5,7 @@ from pytube import Search
 import os
 
 # Initialize conversational models
-empathy_bot = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-empathy")
-general_bot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
+conversational_bot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
 sentiment_analysis = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
 
 # Set up Streamlit page
@@ -34,11 +33,8 @@ if user_input:
     # Run sentiment analysis for crisis detection
     sentiment = sentiment_analysis(user_input)[0]
     
-    # Use empathy model for more specific support if the user is distressed
-    if sentiment['label'] == 'NEGATIVE' and sentiment['score'] > 0.8:
-        response = empathy_bot(user_input, max_length=150)[0]['generated_text']
-    else:
-        response = general_bot(user_input, max_length=150)[0]['generated_text']
+    # Use conversational bot for responses
+    response = conversational_bot(user_input, max_length=150)[0]['generated_text']
     
     # Display response
     st.text_area("Bot's Response:", response, height=150)