Update app.py

app.py

@@ -1,63 +1,40 @@
-# ------------------------------------------------------------------------------------
-# A simple recipe extractor chatbot that extracts recipes from URLs using the OpenAI API.
+# -----------------------------------------------------------------------------
+# A basic example of dynamically re-rendering a Shiny Chat instance with different models.
 # To run it, you'll need an OpenAI API key.
 # To get one, follow the instructions at https://platform.openai.com/docs/quickstart
-# ------------------------------------------------------------------------------------
+# -----------------------------------------------------------------------------
 import os
 
-from openai import AsyncOpenAI
-from utils import recipe_prompt, scrape_page_with_url
-
-from shiny.express import ui
-
-# Provide your API key here (or set the environment variable)
-llm = AsyncOpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
-
-# Set some Shiny page options
-ui.page_opts(
-    title="Recipe Extractor Chat",
-    fillable=True,
-    fillable_mobile=True,
-)
-
-# Initialize the chat (with a system prompt and starting message)
-chat = ui.Chat(
-    id="chat",
-    messages=[
-        {"role": "system", "content": recipe_prompt},
-        {
-            "role": "assistant",
-            "content": "Hello! I'm a recipe extractor. Please enter a URL to a recipe page. For example, <https://www.thechunkychef.com/epic-dry-rubbed-baked-chicken-wings/>",
-        },
-    ],
-)
-
-chat.ui(placeholder="Enter a recipe URL...")
-
-
-# A function to transform user input
-# Note that, if an exception occurs, the function will return a message to the user
-# "short-circuiting" the conversation and asking the user to try again.
-@chat.transform_user_input
-async def try_scrape_page(input: str) -> str | None:
-    try:
-        return await scrape_page_with_url(input)
-    except Exception:
-        await chat.append_message(
-            "I'm sorry, I couldn't extract content from that URL. Please try again. "
-        )
-        return None
-
-
-@chat.on_user_submit
-async def _():
-    response = await llm.chat.completions.create(
-        model="gpt-4o",
-        messages=chat.messages(format="openai"),
-        temperature=0,
-        stream=True,
+from langchain_openai import ChatOpenAI
+
+from shiny.express import input, render, ui
+
+ui.input_select("model", "Model", choices=["gpt-4o", "gpt-3.5-turbo"])
+
+
+@render.express
+def chat_ui():
+
+    chat = ui.Chat(
+        id="chat",
+        messages=[
+            {
+                "content": f"Hi! I'm a {input.model()} model. How can I help you today?",
+                "role": "assistant",
+            }
+        ],
     )
-    await chat.append_message_stream(response)
 
+    chat.ui()
+
+    llm = ChatOpenAI(
+        model=input.model(),
+        # Provide your API key here (or set the environment variable)
+        api_key=os.environ.get("OPENAI_API_KEY"),  # type: ignore
+    )
 
-app = App(app_ui, server)
+    @chat.on_user_submit
+    async def _():
+        messages = chat.messages(format="langchain")
+        response = llm.astream(messages)
+        await chat.append_message_stream(response)