Update app.py

app.py

@@ -18,15 +18,18 @@ generation_args = {
 # Initialize the model pipeline
 chat_pipeline = pipeline("text-generation", model="microsoft/Phi-3-mini-128k-instruct")
 
-# Define system content
-SYSTEM_CONTENT = "You are a helpful assistant named Digital Ink. Your purpose is to provide creative engaging and effective marketing content.You can introduce your self as follows: I'm Digital Ink, a marketing content generation model. I'm designed to assist you in creating engaging and effective marketing content, such as blog posts, social media posts, and product descriptions"
 
 # Streamlit app
 st.title("Digital Ink")
 
 # Initialize the chat history
-if 'messages' not in st.session_state:
-    st.session_state.messages = [{"role": "system", "content": SYSTEM_CONTENT}]
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []
+
+#display chat history
+for message in st.session_state.chat_history:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
 
 
@@ -35,6 +38,7 @@ user_input = st.chat_input("Ask Digital Ink..")
 
 if user_input:
     # Add user message to chat history
+    st.session_state.messages.append({"role": "system", "content": "You are a helpful assistant named Digital Ink. Your purpose is to provide creative engaging and effective marketing content.You can introduce your self as follows: I'm Digital Ink, a marketing content generation model. I'm designed to assist you in creating engaging and effective marketing content, such as blog posts, social media posts, and product descriptions"})
     st.session_state.messages.append({"role": "user", "content": user_input})
     st.chat_state.chat_message("user").markdown(user_input)