#!/usr/bin/env python

from __future__ import annotations

import gradio as gr

from paper_list import PaperList

DESCRIPTION = "# NAACL 2022 Papers"
NOTES = """
- [NAACL 2022](https://2022.naacl.org/)
- [NAACL'22 Reproducibility Track](https://naacl2022-reproducibility-track.github.io/results/)
"""

paper_list = PaperList()

with gr.Blocks(css="style.css") as demo:
    gr.Markdown(DESCRIPTION)

    search_box = gr.Textbox(
        label="Search Title", placeholder="You can search for titles with regular expressions. e.g. (?<!sur)face"
    )
    case_sensitive = gr.Checkbox(label="Case Sensitive")
    filter_names = gr.CheckboxGroup(
        label="Filter",
        choices=[
            "arXiv",
            "GitHub",
            "HF Space",
            "HF Model",
            "HF Dataset",
        ],
    )
    paper_category_names = [
        "Long Paper",
        "Short Paper",
        "Special Theme Paper",
        "Findings",
        "Industry Track",
    ]
    paper_categories = gr.CheckboxGroup(
        label="Paper Categories", choices=paper_category_names, value=paper_category_names
    )
    search_button = gr.Button("Search")

    number_of_papers = gr.Textbox(label="Number of Papers Found")
    table = gr.HTML(show_label=False)

    gr.Markdown(NOTES)

    demo.load(
        fn=paper_list.render,
        inputs=[
            search_box,
            case_sensitive,
            filter_names,
            paper_categories,
        ],
        outputs=[
            number_of_papers,
            table,
        ],
    )
    search_button.click(
        fn=paper_list.render,
        inputs=[
            search_box,
            case_sensitive,
            filter_names,
            paper_categories,
        ],
        outputs=[
            number_of_papers,
            table,
        ],
    )
demo.queue().launch()