Spaces:
Sleeping
Sleeping
| # import openai | |
| import time | |
| import random | |
| import streamlit as st | |
| # from streamlit_chat import message | |
| import os | |
| from utils import set_main_page, message_border | |
| st.set_page_config( | |
| page_title="streamlit-demo", | |
| page_icon="🧊", | |
| layout="wide", | |
| initial_sidebar_state="expanded", | |
| menu_items={ | |
| 'Get Help': 'https://github.com/vickyli777', | |
| 'Report a bug': "https://github.com/vickyli777", | |
| 'About': "## a simple demo!" | |
| } | |
| ) | |
| set_main_page() | |
| st.sidebar.divider() | |
| def clear_chat(): | |
| """ | |
| """ | |
| st.session_state.messages =[] | |
| st.sidebar.button("clear all chats", on_click=clear_chat, type="primary") | |
| st.sidebar.caption("clear histories and start a new chat") | |
| #storing the chat | |
| if 'messages' not in st.session_state: | |
| st.session_state['messages'] = [] | |
| # Display chat messages from history on app rerun | |
| for msg in st.session_state.messages: | |
| with st.chat_message(msg["role"]): | |
| chat_box = message_border(msg["role"],msg["content"]) | |
| st.markdown(chat_box,unsafe_allow_html=True) | |
| # React to user input | |
| if question := st.chat_input("your AI assistant is here, ask me something"): | |
| # Display user message in chat message container | |
| with st.chat_message(name="user"): | |
| cur_border = message_border("user",question) | |
| st.markdown(cur_border,unsafe_allow_html=True) #paragraph | |
| # st.markdown(prompt) | |
| # Add user message to chat history | |
| st.session_state.messages.append({"role": "user", "content": question}) | |
| # response = f"Echo: {prompt}" | |
| # Display assistant response in chat message container | |
| with st.chat_message(name="assistant"): | |
| message_placeholder = st.empty() | |
| full_response = "" | |
| response = random.choice( | |
| [ | |
| "Hello there! How can I assist you today?", | |
| "Hi, human! Is there anything I can help you with?", | |
| "Do you need help?", | |
| ] | |
| ) | |
| # Simulate stream of response with milliseconds delay | |
| for chunk in response.split(): | |
| full_response += chunk + " " | |
| # ans_border = message_border(full_response+ "▌") | |
| time.sleep(0.5) | |
| # Add a blinking cursor to simulate typing | |
| # message_placeholder.markdown(""" | |
| # <p style="border: 1px solid red; | |
| # padding: 5px; | |
| # border-radius: 5px"> | |
| # {} | |
| # </p> | |
| # """.format(full_response +"▌")) | |
| ans_border = message_border("assistant",full_response) | |
| message_placeholder.markdown(ans_border, unsafe_allow_html=True) | |
| # message_placeholder.markdown(ans_border, unsafe_allow_html=True) | |
| # st.markdown(f':blue[{response}]') | |
| # Add assistant response to chat history | |
| st.session_state.messages.append({"role": "assistant", "content": response}) | |