Update app.py

app.py

@@ -30,17 +30,28 @@ st.subheader("Your compassionate companion in tough times 💚")
 # Get user input
 user_input = st.text_input("Share what's on your mind...", placeholder="Type here...", max_chars=500)
 
+# Store previous response to check for repetition
+if 'previous_response' not in st.session_state:
+    st.session_state.previous_response = ""
+
 # Check if user has entered text
 if user_input:
     # Run sentiment analysis to check for distress
     sentiment = sentiment_analysis(user_input)[0]
 
     # Generate empathetic response (model generates responses with empathy)
-    response = conversational_bot(user_input, max_length=150)[0]['generated_text']
+    response = conversational_bot(user_input, max_length=150, num_return_sequences=1, temperature=0.9, top_k=50)[0]['generated_text']
 
     # Ensure response does not repeat what the user said, and is supportive
     if user_input.lower() in response.lower():
         response = "I understand how you're feeling. You're not alone in this. I'm here to listen and help."
+    
+    # Check if the response is too similar to the previous one, if so, generate a new one
+    if response == st.session_state.previous_response:
+        response = conversational_bot(user_input, max_length=150, num_return_sequences=1, temperature=0.9, top_k=50)[0]['generated_text']
+    
+    # Store the response for future comparison
+    st.session_state.previous_response = response
 
     # Display response
     st.text_area("Bot's Response:", response, height=150)
@@ -59,7 +70,7 @@ if user_input:
         
         # Search YouTube for videos related to the selected activity
         search = Search(activity)
-        search_results = search.results[:2]  # limit results to 2 videos
+        search_results = search.results[:3]  # limit results to 3 videos
         for video in search_results:
             st.write(f"[{video.title}]({video.watch_url})")