Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 from gtts import gTTS
 import torch
 
-# Load GPT-2 model and tokenizer from Hugging Face
-tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-model = GPT2LMHeadModel.from_pretrained("gpt2")
+# Load DialoGPT model and tokenizer from Hugging Face
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
 # Set up Streamlit page configuration
 st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🌿", layout="centered")
@@ -31,46 +31,27 @@ user_input = st.text_input("Share what's on your mind...", placeholder="Type her
 if 'previous_responses' not in st.session_state:
     st.session_state.previous_responses = []
 
-# Function to generate a more empathetic and focused response
+# Function to generate a more empathetic and focused response using DialoGPT
 def generate_response(user_input):
-    # Tokenize input and set parameters for text generation
-    inputs = tokenizer.encode(user_input, return_tensors="pt")
-    outputs = model.generate(
-        inputs, 
-        max_length=150, 
-        temperature=0.7, 
-        top_k=50, 
-        repetition_penalty=1.2,  # Adds a penalty for repeated phrases
-        num_return_sequences=1,
-        pad_token_id=tokenizer.eos_token_id
-    )
-    response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    # Encode the input text and generate a response
+    new_user_input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors='pt')
+    bot_input_ids = new_user_input_ids
+    chat_history_ids = model.generate(bot_input_ids, max_length=150, pad_token_id=tokenizer.eos_token_id, temperature=0.7, top_k=50, repetition_penalty=1.2)
     
-    # Trim the output to remove extraneous content
-    response_text = response_text.split("Here's something")[0].strip()
+    # Decode the response to text
+    chat_history_ids = chat_history_ids[:, bot_input_ids.shape[-1]:]  # remove the input from the response
+    bot_output = tokenizer.decode(chat_history_ids[0], skip_special_tokens=True)
     
-    # Add a suggestion for coping activity based on keywords in user input
-    if "crying" in user_input.lower() or "overload" in user_input.lower():
-        activity_suggestion = (
-            "Work overload can be overwhelming, and it's completely understandable to feel like you're reaching your limit. "
-            "Sometimes, taking a short break can help you gain some clarity and calm your mind. "
-            "If possible, try stepping outside for a few minutes to breathe deeply or do some light stretching to release tension."
-        )
-    elif "angry" in user_input.lower() or "frustrated" in user_input.lower():
-        activity_suggestion = (
-            "When feeling overwhelmed, anger can sometimes mask deeper emotions like stress or exhaustion. "
-            "Taking deep breaths or doing a short meditation can help you ground yourself and regain focus. "
-            "Even small moments of calm can make a huge difference."
-        )
-    else:
-        activity_suggestion = (
-            "Sometimes, a creative outlet can be a great way to cope with overwhelming feelings. "
-            "Art, whether it’s drawing, painting, or even coloring, allows you to express what you might not be able to say out loud."
-        )
-    
-    # Add a closing empathetic sentence
-    response = f"{response_text}\n\nHere's something you could try to help cope with how you're feeling:\n{activity_suggestion}"
+    # Add empathetic and coping suggestions
+    response = f"{bot_output}\n\nHere's something you could try to help cope with how you're feeling:\n"
     
+    if "crying" in user_input.lower():
+        response += "Sometimes letting yourself cry is a necessary release. It’s okay to feel what you're feeling. Also, trying to reach out to a friend or family member for support can be really comforting."
+    elif "overwhelmed" in user_input.lower():
+        response += "Overwhelm can often feel like too much to handle, but try taking it one step at a time. Break your tasks down into small pieces and remember to take breaks. You don't have to do it all at once."
+    else:
+        response += "Taking a moment to breathe deeply, meditate, or even take a short walk can help you regain some balance in this difficult moment."
+
     return response
 
 # Check if the user has typed something