#!/usr/bin/env python

from __future__ import annotations

import gradio as gr

from paper_list import PaperList

DESCRIPTION = '# ICLR 2023 Paper Submissions'
NOTES = '''
- [ICLR 2023](https://openreview.net/group?id=ICLR.cc/2023/Conference)
- [List of submitted papers](https://docs.google.com/spreadsheets/d/1dQMjjetud2edTEREdLiuD4giC244lxY67ZxaL7NiMUc/edit#gid=1277917086)
'''


def main():
    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',
            max_lines=1)

        case_sensitive = gr.Checkbox(label='Case Sensitive')

        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
                  ],
                  outputs=[
                      number_of_papers,
                      table,
                  ])
        search_box.submit(fn=paper_list.render,
                          inputs=[
                              search_box,
                              case_sensitive
                          ],
                          outputs=[
                              number_of_papers,
                              table,
                          ])

        search_button.click(fn=paper_list.render,
                            inputs=[
                                search_box,
                                case_sensitive
                            ],
                            outputs=[
                                number_of_papers,
                                table,
                            ])

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


if __name__ == '__main__':
    main()