import streamlit as st
# Initialize history
if "messages" not in st.session_state:
st.session_state.messages = []
# Display past messages
for msg in st.session_state.messages:
with st.chat_message(msg["role"]):
st.markdown(msg["content"])
# Chat input
if prompt := st.chat_input("Say something"):
# Add user message to history and display it
st.session_state.messages.append({"role": "user", "content": prompt})
with st.chat_message("user"):
st.markdown(prompt)
# Generate assistant response (echo in this case)
response = f"Echo: {prompt}"
# Display assistant response and add to history
with st.chat_message("assistant"):
st.markdown(response)
st.session_state.messages.append({"role": "assistant", "content": response})