Spaces:
Sleeping
Sleeping
File size: 691 Bytes
1d77aa6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import warnings
import logging
import streamlit as st
warnings.filterwarnings("ignore")
logging.getLogger("transformers").setLevel(logging.ERROR)
st.title("ChatUI")
if 'messages' not in st.session_state:
st.session_state.messages = []
for message in st.session_state.messages:
st.chat_message(message['role']).markdown(message['content'])
prompt = st.chat_input('enter a message')
if prompt:
st.chat_message('user').markdown(prompt)
st.session_state.messages.append({'role': 'user', 'content': prompt})
response = "Hi! I'm Chatty"
st.chat_message('assistant').markdown(response)
st.session_state.messages.append({'role': 'assistant', 'content': response})
|