Py.Cafe

alonsosilvaallende/

solara-streambot

Solara StreamBot

DocsPricing
  • app.py
  • requirements.txt
app.py
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
26
27
28
29
30
31
32
33
34
35
import solara
from typing import List
from typing_extensions import TypedDict

class MessageDict(TypedDict):
    role: str
    content: str

messages: solara.Reactive[List[MessageDict]] = solara.reactive([])

messages.value = [
    {"role": "user", "content": "Hello!"}, 
    {"role": "assistant",  "content": "Hello! How can I assist you today?"},
]

@solara.component
def Page():
    user_message_count = len([m for m in messages.value if m["role"] == "user"])
    def send(message):
        messages.value = [*messages.value, {"role": "user", "content": message}]
    def response(message):
        messages.value = [*messages.value, {"role": "assistant", "content": message}]
    def result(user_message_count):
        if messages.value != []:
            response(messages.value[-1]["content"])
    result = solara.lab.use_task(result, dependencies=[user_message_count])
    with solara.lab.ChatBox():
        for item in messages.value:
            with solara.lab.ChatMessage(
                user=item["role"] == "user",
                name="User" if item["role"] == "user" else "Assistant"
            ):
                solara.Markdown(item["content"])
    solara.lab.ChatInput(send_callback=send)
Page()