Update app.py

app.py

@@ -37,7 +37,10 @@ def generate_story(theme):
 
 
 def generate_response(user_input):
-    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."
+    # Update the prompt to be more natural and empathetic.
+    response_prompt = f"User shares: '{user_input}'. Respond with empathy, encouragement, and reassurance. Be kind and understanding, offering support for their situation."
+    
+    # Generate the response using the model (this could be GPT-2 or DialoGPT).
     input_ids = gpt2_tokenizer.encode(response_prompt, return_tensors='pt')
     response_ids = gpt2_model.generate(
         input_ids,
@@ -47,8 +50,13 @@ def generate_response(user_input):
         repetition_penalty=1.2,
         num_return_sequences=1
     )
+    
+    # Decode and return the response
     response = gpt2_tokenizer.decode(response_ids[0], skip_special_tokens=True)
-    return response
+    
+    # Clean up the response to avoid the model repeating the prompt
+    cleaned_response = response.replace(f"You are a compassionate, kind,and empathetic support bot. A user has shared their feelings: '{user_input}'. Respond with empathy, encouragement, and motivation.", "").strip()
+    return cleaned_response