New component: Chatbox

Custom Components
1

Chatbox component for streamlit

A Streamlit component to show chat messages.

Features

  • user can custom the bg_color and icon of message senders.
  • support streaming output.
  • support image/video/audio messages

This make it easy to chat with LLMs in streamlit.

Install

just pip install streamlit-chatbox

Usage examples

import streamlit as st
from streamlit_chatbox import st_chatbox, MsgType
import time


st.write('start to chat')
streaming = st.checkbox('streaming', False)
chat_box = st_chatbox(
	greetings=['welcome to chat', '\n```streamlit``` is a great tool!'])
# use help(st_chatbox) to see costom params

q = st.text_input('input', placeholder='input your question here')
if q:
    chat_box.user_say(q)
    text = f'my answer to:\n\n{q}\n\n```this is some code```'
    if streaming:
        chat_box.robot_stream(text)
    else:
        chat_box.robot_say(text)

    chat_box.robot_say('https://tse4-mm.cn.bing.net/th/id/OIP-C.cy76ifbr2oQPMEs2H82D-QHaEv?w=284&h=181&c=7&r=0&o=5&dpr=1.5&pid=1.7', MsgType.IMAGE)

    chat_box.robot_say('https://sample-videos.com/video123/mp4/720/big_buck_bunny_720p_1mb.mp4', MsgType.VIDEO, format='viedo/mp4')

    chat_box.robot_say('https://sample-videos.com/video123/mp4/720/big_buck_bunny_720p_1mb.mp4', MsgType.AUDIO, format='audio/mp4')

chat_box.output_messages()

demo

2 Likes
2

Hi @liunux could you post some screenshots?

3

demo
I did upload a gif demo to github, but not shown.

1 Like
4

fix an error in the example code:

import streamlit as st
from streamlit_chatbox import st_chatbox
import time


st.write('start to chat')
streaming = st.checkbox('streaming', False)
chat_box = st_chatbox(greetings=['welcome to chat', '\n```streamlit``` is a great tool!'])


q = st.text_input('input', placeholder='input your question here')
if q:
    chat_box.user_say(q)
    text = f'my answer to:\n\n{q}\n\n```this is some code```\n\n'
    if streaming:
        chat_box.robot_say('')
		chat_box.output_messages() # you need output the msg first for streaming output
        for i in range(1, len(text)+10, 5):
            chat_box.update_last_box_text(text[:i])
            time.sleep(0.3)
    else:
        chat_box.robot_say(text)

chat_box.output_messages()
5

Glad to share that I have upgrade the package to support new chat elements. Some new futures also be added, take a look at streamlit-chatbox · PyPI.

demo

6

This topic was automatically closed 180 days after the last reply. New replies are no longer allowed.