Add filter for links

app.py

@@ -16,9 +16,21 @@ def main():
 
     with gr.Blocks(css='style.css') as demo:
         gr.Markdown(DESCRIPTION)
+        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=names_with_link, outputs=table)
+        names_with_link.change(paper_list.render,
+                               inputs=names_with_link,
+                               outputs=table)
+
     demo.launch(enable_queue=True, share=False)
 
 

papers.py

@@ -5,7 +5,7 @@ import pandas as pd
 
 class PaperList:
     def __init__(self):
-        self.table = pd.read_csv('papers.csv').fillna('')
+        self.table = pd.read_csv('papers.csv')
         self.table_header = '''
             <tr>
                 <td width="50%">Paper</td>
@@ -17,6 +17,27 @@ class PaperList:
                 <td width="5%">Hugging Face Spaces</td>
             </tr>'''
 
+    def render(self, names_with_link: list[str]) -> str:
+        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,
+                               has_hf_space)
+        return self.to_html(df)
+
+    def filter_table(self, df: pd.DataFrame, has_supp: bool, has_arxiv: bool,
+                     has_github: bool, has_hf_space: bool) -> pd.DataFrame:
+        if has_supp:
+            df = df[~df.supp.isna()]
+        if has_arxiv:
+            df = df[~df.arxiv.isna()]
+        if has_github:
+            df = df[~df.github.isna()]
+        if has_hf_space:
+            df = df[~df.hf_space.isna()]
+        return df
+
     def to_html(self, df: pd.DataFrame) -> str:
         table_rows = self.generate_table_rows(df)
         table_data = ''.join(table_rows)
@@ -31,10 +52,14 @@ class PaperList:
         for row in df.itertuples():
             paper = f'<a href="{row.url}">{row.title}</a>'
             pdf = f'<a href="{row.pdf}">pdf</a>'
-            supp = f'<a href="{row.supp}">supp</a>' if row.supp else ''
-            arxiv = f'<a href="{row.arxiv}">arXiv</a>' if row.arxiv else ''
-            github = f'<a href="{row.github}">GitHub</a>' if row.github else ''
-            hf_space = f'<a href="{row.hf_space}">Space</a>' if row.hf_space else ''
+            supp = f'<a href="{row.supp}">supp</a>' if isinstance(
+                row.supp, str) else ''
+            arxiv = f'<a href="{row.arxiv}">arXiv</a>' if isinstance(
+                row.arxiv, str) else ''
+            github = f'<a href="{row.github}">GitHub</a>' if isinstance(
+                row.github, str) else ''
+            hf_space = f'<a href="{row.hf_space}">Space</a>' if isinstance(
+                row.hf_space, str) else ''
             row = f'''
                 <tr>
                     <td>{paper}</td>