Update app.py

app.py

@@ -36,11 +36,10 @@ def generate_story(theme):
     return story
 
 
-# Function to generate an empathetic response
 def generate_response(user_input):
-    response_prompt = f"You are a compassionate support bot. A user has shared: '{user_input}'. Respond with empathy and encouragement."
-    input_ids = tokenizer.encode(response_prompt, return_tensors='pt')
-    chat_history_ids = model.generate(
+    response_prompt = f"You are a compassionate, kind ,and empathetic support bot. A user has shared their feelings: '{user_input}'. Respond with empathy, encouragement, and motivation."
+    input_ids = gpt2_tokenizer.encode(response_prompt, return_tensors='pt')
+    response_ids = gpt2_model.generate(
         input_ids,
         max_length=300,
         temperature=0.85,
@@ -48,10 +47,11 @@ def generate_response(user_input):
         repetition_penalty=1.2,
         num_return_sequences=1
     )
-    response = tokenizer.decode(chat_history_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
+    response = gpt2_tokenizer.decode(response_ids[0], skip_special_tokens=True)
     return response
 
 
+
 # Analyze user input for emotional tone
 def get_emotion(user_input):
     emotions = emotion_classifier(user_input)