Update app.py

app.py

@@ -95,6 +95,7 @@ class Prettifier:
         new_rows = []
         for _, row in df.iterrows():
             new_row = {
+                "arxiv_id": row["arxiv_id"],  # Include arxiv_id
                 "date": Prettifier.create_link(row.date, f"https://huggingface.co/papers?date={row.date}"),
                 "paper_page": Prettifier.create_link(row.arxiv_id, row.paper_page),
                 "title": row["title"],
@@ -108,6 +109,7 @@ class Prettifier:
 
 class PaperList:
     COLUMN_INFO = [
+        ["arxiv_id", "str"],        # Added arxiv_id
         ["date", "markdown"],
         ["paper_page", "markdown"],
         ["title", "str"],
@@ -176,8 +178,8 @@ class PaperManager:
         self.papers_per_page = papers_per_page
         self.current_page = 1
         self.sort_method = "hot"  # Default sort method
-        self.total_pages = max((len(self.paper_list.df_raw) + self.papers_per_page - 1) // self.papers_per_page, 1)
         self.sort_papers()
+        self.total_pages = max((len(self.paper_list.df_raw) + self.papers_per_page - 1) // self.papers_per_page, 1)
 
     def calculate_score(self, row):
         """
@@ -486,6 +488,7 @@ table {
 }
 """
 
+
 # --- Initialize Gradio Blocks ---
 
 demo = gr.Blocks(css=css)
@@ -500,7 +503,6 @@ with demo:
 
             Once your paper is submitted, it will automatically appear in this demo.
             """)
-
         # Header with Refresh Button
         with gr.Row():
             gr.HTML("""
@@ -517,7 +519,6 @@ with demo:
                 </tr>
             </table>
             """)
-
         # Sort Options
         with gr.Row():
             sort_radio = gr.Radio(
@@ -526,10 +527,8 @@ with demo:
                 label="Sort By",
                 interactive=True
             )
-
         # Paper list
         paper_list = gr.HTML()
-
         # Navigation Buttons
         with gr.Row():
             prev_button = gr.Button("Prev")
@@ -576,4 +575,4 @@ with demo:
 # --- Launch the App ---
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()