import streamlit as st
from humanize import naturaltime
from loguru import logger
from streamlit_monaco import st_monaco  # type: ignore

from open_notebook.domain.notebook import Note
from open_notebook.graphs.multipattern import graph as pattern_graph
from open_notebook.utils import surreal_clean

from .consts import context_icons


@st.dialog("Write a Note", width="large")
def add_note(session_id):
    note_title = st.text_input("Title")
    note_content = st.text_area("Content")
    if st.button("Save", key="add_note"):
        logger.debug("Adding note")
        note = Note(title=note_title, content=note_content, note_type="human")
        note.save()
        note.add_to_notebook(st.session_state[session_id]["notebook"].id)
        st.rerun()


@st.dialog("Add a Source", width="large")
def note_panel(session_id=None, note_id=None):
    if note_id:
        note: Note = Note.get(note_id)
    else:
        note: Note = Note()

    t_preview, t_edit = st.tabs(["Preview", "Edit"])
    with t_preview:
        st.subheader(note.title)
        st.markdown(note.content)
    with t_edit:
        note.title = st.text_input("Title", value=note.title)
        note.content = st_monaco(
            value=note.content, height="600px", language="markdown"
        )
        if st.button("Save", key=f"pn_edit_note_{note_id}"):
            logger.debug("Editing note")
            note.save()
            if not note.id:
                note.add_to_notebook(st.session_state[session_id]["notebook"].id)
            st.rerun()
    if st.button("Delete", type="primary", key=f"delete_note_{note_id}"):
        logger.debug("Deleting note")
        note.delete()
        st.rerun()


def make_note_from_chat(content, notebook_id=None):
    # todo: make this more efficient
    transformations = [
        "Based on the Note below, please provide a Title for this content, with max 15 words"
    ]
    output = pattern_graph.invoke(
        dict(content_stack=[content], transformations=transformations)
    )
    title = surreal_clean(output["output"])

    note = Note(
        title=title,
        content=content,
        note_type="ai",
    )
    note.save()
    if notebook_id:
        note.add_to_notebook(notebook_id)

    st.rerun()


def note_card(session_id, note):
    if note.note_type == "human":
        icon = "🤵"
    else:
        icon = "🤖"

    with st.container(border=True):
        st.markdown((f"{icon} **{note.title if note.title else 'No Title'}**"))
        context_state = st.selectbox(
            "Context",
            label_visibility="collapsed",
            options=context_icons,
            index=0,
            key=f"note_{note.id}",
        )
        st.caption(f"Updated: {naturaltime(note.updated)}")

        if st.button("Expand", icon="📝", key=f"edit_note_{note.id}"):
            note_panel(session_id, note.id)

    st.session_state[session_id]["context_config"][note.id] = context_state


def note_list_item(note_id, score=None):
    logger.debug(note_id)
    note: Note = Note.get(note_id)
    if note.note_type == "human":
        icon = "🤵"
    else:
        icon = "🤖"

    with st.expander(
        f"{icon} [{score:.2f}] **{note.title}** {naturaltime(note.updated)}"
    ):
        st.write(note.content)
        if st.button("Edit Note", icon="📝", key=f"x_edit_note_{note.id}"):
            note_panel(note_id=note.id)