import streamlit as st

st.set_page_config(
    page_title="Grascii Search",
    menu_items={
        "About": """
        Web interface for [grascii](https://github.com/grascii/grascii)'s
        search utility

        Image search powered by [gregg-vision-v0.2.1](https://huggingface.co/grascii/gregg-vision-v0.2.1)
        """
    },
)

import pandas as pd  # noqa E402
from search import write_grascii_search, write_reverse_search  # noqa E402

pd.options.mode.copy_on_write = True

if "report_submitted" not in st.session_state:
    st.session_state["report_submitted"] = False

if "grascii" not in st.session_state:
    st.session_state["grascii"] = ""

if "alternatives" not in st.session_state:
    st.session_state["alternatives"] = {}

if "save_image" in st.session_state:
    st.session_state["save_image"] = st.session_state["save_image"]

if st.session_state["report_submitted"]:
    st.toast("Thanks for the report!")
    st.session_state["report_submitted"] = False

tab1, tab2 = st.tabs(["Grascii", "Reverse"])

with tab1:
    write_grascii_search()

with tab2:
    write_reverse_search()