Update page1.py

page1.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from PyPDF2 import PdfReader
 from langchain_core.messages import HumanMessage, AIMessage
+from langchain_core.messages import SystemMessage
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
@@ -92,7 +93,15 @@ def text():
         max_retries=6
 
     )
-
+        # Add system prompt (psychological assistant)
+    system_prompt = (
+        "You are a compassionate and emotionally intelligent AI assistant trained in cognitive behavioral therapy (CBT), "
+        "mindfulness, and active listening. You provide supportive, empathetic responses without making medical diagnoses. "
+        "Use a warm tone and guide users to explore their feelings, reframe thoughts, and reflect gently."
+    )
+    
+    # Add SystemMessage to chat history
+    st.session_state.chat_history.add_message(SystemMessage(content=system_prompt))
     # Display chat messages
     chat_container = st.container()
     with chat_container: