Update app.py

app.py

@@ -36,6 +36,7 @@ user_input = st.chat_input("Ask Digital Ink..")
 if user_input:
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": user_input})
+    st.chat_state.chat_message("user").markdown(user_input)
 
     # Generate response from chatbot
     context = [msg['content'] for msg in st.session_state.messages]
@@ -43,6 +44,8 @@ if user_input:
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
+    with st.chat_message("assistant"):
+        st.markdown(response)
 
     # Redisplay chat history
     st.experimental_rerun()