Add user input

This commit is contained in:
Shaquille Soekhlal 2024-06-23 14:08:42 +00:00
parent 86f8adb104
commit 2169d321fb

8
app.py
View File

@ -10,3 +10,11 @@ if "messages" not in st.session_state:
for message in st.session_state.messages: for message in st.session_state.messages:
with st.chat_message(message["role"]): with st.chat_message(message["role"]):
st.markdown(message["content"]) st.markdown(message["content"])
# React to user input
if prompt := st.chat_input("What is up?"):
# Display user message in chat message container
with st.chat_message("user"):
st.markdown(prompt)
# Add user message to chat history
st.session_state.messages.append({"role": "user", "content": prompt})