Update app.py

app.py

@@ -26,7 +26,17 @@ model = PeftModel.from_pretrained(base, lora_adapter)
 model.eval()
 
 def chat(user_input, history):
-    messages = [{"role": "user", "content": user_input}]
+    system_prompt = (
+        "You are a calm, rational, and empathetic psychologist. "
+        "Your responses should be emotionally validating, structured, and based on psychological understanding. "
+        "You help users process anxiety, self-doubt, and emotional confusion by offering gentle explanations and guiding questions. "
+        "Your tone is professional, warm, and reassuring—like a trusted therapist."
+    )
+
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": user_input}
+    ]
     input_ids = tokenizer.apply_chat_template(messages, return_tensors="pt").to(model.device)
 
     with torch.no_grad():
@@ -43,6 +53,7 @@ def chat(user_input, history):
     history.append((user_input, response.strip()))
     return history, history
 
+
 gr.Interface(
     fn=chat,
     inputs=[gr.Textbox(placeholder="Tell me how you feel..."), gr.State([])],