Spaces:
Running
Running
File size: 917 Bytes
831e906 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import streamlit as st
from components.generate_chat_stream import generate_chat_stream
from components.stream_handler import stream_handler
from components.show_source import show_source
def chat_loop(session_state, config):
if prompt := st.chat_input("Search the web..."):
st.chat_message("user").markdown(prompt)
session_state.messages.append({"role": "user", "content": prompt})
chat_stream, links = generate_chat_stream(session_state, prompt, config)
with st.chat_message("assistant"):
placeholder = st.empty()
full_response = stream_handler(
session_state, chat_stream, prompt, placeholder
)
if session_state.rag_enabled:
show_source(links)
session_state.history.append([prompt, full_response])
session_state.messages.append({"role": "assistant", "content": full_response})
|