Update app.py

app.py

@@ -44,7 +44,13 @@ if user_input:
 
     # Generate response from chatbot
     context = [msg['content'] for msg in st.session_state.messages]
-    response = chat_pipeline(context, **generation_args)[0]['generated_text']
+    messages = [ 
+    {"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"}, 
+    {"role": "user", "content": user_input}, 
+    {"role": "assistant", "content": ""}, 
+    {"role": "user", "content": ""}, 
+]
+    response = chat_pipeline(messages, **generation_args)[0]['generated_text']
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})