# /// script
# requires-python = ">=3.12"
# dependencies = [
#     "marimo",
#     "openai==1.53.0",
# ]
# ///

import marimo

__generated_with = "0.9.20"
app = marimo.App(width="medium")


@app.cell
def __(mo):
    mo.md(
        r"""
        # Recipe Bot 🤖

        If you upload a screenshot of a recipe, this chatbot will catalog them for you.
        """
    )
    return


@app.cell
def __():
    import marimo as mo
    import os
    return mo, os


@app.cell
def __(mo, os):
    api_key = mo.ui.text(
        label="API Key",
        kind="password",
        value=os.environ.get("OPENAI_API_KEY") or "",
    )
    api_key
    return (api_key,)


@app.cell
def __(api_key, mo):
    chat = mo.ui.chat(
        mo.ai.llm.openai(
            "gpt-4o",
            system_message="""You are a helpful assistant that can
            parse my recipe and summarize them for me.
            Give me a title in the first line.""",
            api_key=api_key,
        ),
        allow_attachments=["image/png", "image/jpeg"],
        prompts=["What is the recipe?"],
    )
    chat
    return (chat,)


@app.cell
def __(chat, mo):
    mo.stop(not chat.value)

    last_message: str = chat.value[-1].content
    title = last_message.split("\n")[0]
    summary = "\n".join(last_message.split("\n")[1:])
    with open(f"{title}.md", "w") as f:
        f.write(summary)
        mo.status.toast("Receipt summary saved!", description=title)
    return f, last_message, summary, title


if __name__ == "__main__":
    app.run()