Spaces:
Sleeping
Sleeping
File size: 2,012 Bytes
8b78dce 96cd409 8b78dce 187d834 96cd409 8b78dce 187d834 96cd409 8b78dce 96cd409 8b78dce 96cd409 187d834 |
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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
# ------------------------------------------------------------------------------------
# A simple recipe extractor chatbot that extracts recipes from URLs using the OpenAI API.
# 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,
)
await chat.append_message_stream(response)
app = App(app_ui, server) |