Refactor. Pre-generate HTML rows

papers.py

@@ -6,7 +6,7 @@ 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._preprcess_table()
 
         self.table_header = '''
             <tr>
@@ -19,6 +19,34 @@ class PaperList:
                 <td width="5%">Hugging Face Spaces</td>
             </tr>'''
 
+    def _preprcess_table(self) -> None:
+        self.table['title_lowercase'] = self.table.title.str.lower()
+
+        rows = []
+        for row in self.table.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 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>
+                    <td>{row.authors}</td>
+                    <td>{pdf}</td>
+                    <td>{supp}</td>
+                    <td>{arxiv}</td>
+                    <td>{github}</td>
+                    <td>{hf_space}</td>
+                </tr>'''
+            rows.append(row)
+        self.table['html_table_content'] = rows
+
     def render(self, search_query: str, case_sensitive: bool,
                names_with_link: list[str]) -> tuple[int, str]:
         df = self.table
@@ -48,36 +76,10 @@ class PaperList:
         return df
 
     def to_html(self, df: pd.DataFrame) -> str:
-        table_rows = self.generate_table_rows(df)
+        table_rows = df.html_table_content
         table_data = ''.join(table_rows)
         html = f'''<table>
             {self.table_header}
             {table_data}
         </table>'''
         return html
-
-    def generate_table_rows(self, df: pd.DataFrame) -> list[str]:
-        rows = []
-        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 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>
-                    <td>{row.authors}</td>
-                    <td>{pdf}</td>
-                    <td>{supp}</td>
-                    <td>{arxiv}</td>
-                    <td>{github}</td>
-                    <td>{hf_space}</td>
-                </tr>'''
-            rows.append(row)
-        return rows