Add search box

app.py

@@ -16,6 +16,11 @@ def main():
 
     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',
@@ -26,9 +31,26 @@ def main():
         table = gr.HTML(html_text, show_label=False)
         gr.Markdown(FOOTER)
 
-        demo.load(paper_list.render, inputs=names_with_link, outputs=table)
+        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=names_with_link,
+                               inputs=[
+                                   search_box,
+                                   case_sensitive,
+                                   names_with_link,
+                               ],
                                outputs=table)
 
     demo.launch(enable_queue=True, share=False)

papers.py

@@ -6,6 +6,8 @@ import pandas as pd
 class PaperList:
     def __init__(self):
         self.table = pd.read_csv('papers.csv')
+        self.table['title_lowercase'] = self.table.title.str.lower()
+
         self.table_header = '''
             <tr>
                 <td width="50%">Paper</td>
@@ -17,12 +19,19 @@ class PaperList:
                 <td width="5%">Hugging Face Spaces</td>
             </tr>'''
 
-    def render(self, names_with_link: list[str]) -> str:
+    def render(self, search_query: str, case_sensitive: bool,
+               names_with_link: list[str]) -> str:
+        df = self.table
+        if search_query:
+            if case_sensitive:
+                df = df[df.title.str.contains(search_query)]
+            else:
+                df = df[df.title_lowercase.str.contains(search_query.lower())]
         has_supp = 'Supp' in names_with_link
         has_arxiv = 'arXiv' in names_with_link
         has_github = 'GitHub' in names_with_link
         has_hf_space = 'HF Space' in names_with_link
-        df = self.filter_table(self.table, has_supp, has_arxiv, has_github,
+        df = self.filter_table(df, has_supp, has_arxiv, has_github,
                                has_hf_space)
         return self.to_html(df)