Added Upstage ChatBot

main.py

@@ -1,57 +1,63 @@
 from fasthtml.common import *
-from fasthtml_hf import setup_hf_backup
-
-app,rt,todos,Todo = fast_app(
-    'data/todos.db',
-    hdrs=[Style(':root { --pico-font-size: 100%; }')],
-    id=int, title=str, done=bool, pk='id')
-
-id_curr = 'current-todo'
-def tid(id): return f'todo-{id}'
-
-@patch
-def __ft__(self:Todo):
-    show = AX(self.title, f'/todos/{self.id}', id_curr)
-    edit = AX('edit',     f'/edit/{self.id}' , id_curr)
-    dt = ' ✅' if self.done else ''
-    return Li(show, dt, ' | ', edit, id=tid(self.id))
-
-def mk_input(**kw): return Input(id="new-title", name="title", placeholder="New Todo", required=True, **kw)
-
-@rt("/")
-async def get():
-    add = Form(Group(mk_input(), Button("Add")),
-               hx_post="/", target_id='todo-list', hx_swap="beforeend")
-    card = Card(Ul(*todos(), id='todo-list'),
-                header=add, footer=Div(id=id_curr)),
-    title = 'Todo list'
-    return Title(title), Main(H1(title), card, cls='container')
-
-@rt("/todos/{id}")
-async def delete(id:int):
-    todos.delete(id)
-    return clear(id_curr)
-
-@rt("/")
-async def post(todo:Todo): return todos.insert(todo), mk_input(hx_swap_oob='true')
-
-@rt("/edit/{id}")
-async def get(id:int):
-    res = Form(Group(Input(id="title"), Button("Save")),
-        Hidden(id="id"), CheckboxX(id="done", label='Done'),
-        hx_put="/", target_id=tid(id), id="edit")
-    return fill_form(res, todos.get(id))
-
-@rt("/")
-async def put(todo: Todo): return todos.upsert(todo), clear(id_curr)
-
-@rt("/todos/{id}")
-async def get(id:int):
-    todo = todos.get(id)
-    btn = Button('delete', hx_delete=f'/todos/{todo.id}',
-                 target_id=tid(todo.id), hx_swap="outerHTML")
-    return Div(Div(todo.title), btn)
-
-setup_hf_backup(app)
+from openai import OpenAI # openai==1.2.0
+from dotenv import load_dotenv, find_dotenv
+_ = load_dotenv(find_dotenv())
+
+upstage_token  = os.getenv('UPSTAGE_TOKEN')
+
+# Set up the app, including daisyui and tailwind for the chat component
+hdrs = (picolink, Script(src="https://cdn.tailwindcss.com"),
+    Link(rel="stylesheet", href="https://cdn.jsdelivr.net/npm/[email protected]/dist/full.min.css"))
+app = FastHTML(hdrs=hdrs, cls="p-4 max-w-lg mx-auto")
+
+client = OpenAI(
+    api_key=upstage_token,
+    base_url="https://api.upstage.ai/v1/solar"
+)
+
+def get_completion_from_messages(messages, model="solar-1-mini-chat", temperature=0):
+    response = client.ChatCompletion.create(
+        model=model,
+        messages=messages,
+        temperature=temperature, # this is the degree of randomness of the model's output
+    )
+
+# Chat message component (renders a chat bubble)
+def ChatMessage(msg, user):
+    bubble_class = "chat-bubble-primary" if user else 'chat-bubble-secondary'
+    chat_class = "chat-end" if user else 'chat-start'
+    return Div(cls=f"chat {chat_class}")(
+               Div('user' if user else 'assistant', cls="chat-header"),
+               Div(msg, cls=f"chat-bubble {bubble_class}"),
+               Hidden(msg, name="messages")
+           )
+
+# The input field for the user message. Also used to clear the
+# input field after sending a message via an OOB swap
+def ChatInput():
+    return Input(name='msg', id='msg-input', placeholder="Type a message",
+                 cls="input input-bordered w-full", hx_swap_oob='true')
+
+# The main screen
+@app.get
+def index():
+    page = Form(hx_post=send, hx_target="#chatlist", hx_swap="beforeend")(
+           Div(id="chatlist", cls="chat-box h-[73vh] overflow-y-auto"),
+               Div(cls="flex space-x-2 mt-2")(
+                   Group(ChatInput(), Button("Send", cls="btn btn-primary"))
+               )
+           )
+    return Titled('Chatbot Demo', page)
+
+# Handle the form submission
+@app.post
+def send(msg:str, messages:list[str]=None):
+    if not messages: messages = []
+    messages.append(msg.rstrip())
+    r = contents(get_completion_from_messages(messages)) # get response from chat model
+    return (ChatMessage(msg, True),    # The user's message
+            ChatMessage(r.rstrip(), False), # The chatbot's response
+            ChatInput()) # And clear the input field via an OOB swap
 
 serve()
+

requirements.txt

@@ -3,3 +3,4 @@ huggingface-hub
 fasthtml-hf
 python-multipart
 sqlite-utils
+claudette