#!/usr/bin/env python

from __future__ import annotations

import gradio as gr

from papers import PaperList

DESCRIPTION = '# CVPR 2022 papers'
FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=hysts.cvpr2022_papers" />'


def main():
    paper_list = PaperList()
    html_text = paper_list.to_html(paper_list.table)

    with gr.Blocks(css='style.css') as demo:
        gr.Markdown(DESCRIPTION)
        search_box = gr.Textbox(
            label='Keywords',
            placeholder='You can search for titles with regular expressions')
        case_sensitive = gr.Checkbox(label='Case Sensitive')
        search_button = gr.Button('Search')
        names_with_link = gr.CheckboxGroup(choices=[
            'Supp',
            'arXiv',
            'GitHub',
            'HF Space',
        ],
                                           label='With')
        table = gr.HTML(html_text, show_label=False)
        gr.Markdown(FOOTER)

        demo.load(paper_list.render,
                  inputs=[
                      search_box,
                      case_sensitive,
                      names_with_link,
                  ],
                  outputs=table)
        search_button.click(paper_list.render,
                            inputs=[
                                search_box,
                                case_sensitive,
                                names_with_link,
                            ],
                            outputs=table)
        names_with_link.change(paper_list.render,
                               inputs=[
                                   search_box,
                                   case_sensitive,
                                   names_with_link,
                               ],
                               outputs=table)

    demo.launch(enable_queue=True, share=False)


if __name__ == '__main__':
    main()