Update app.py

app.py

@@ -17,7 +17,7 @@ class PaperManager:
             response.raise_for_status()
             data = response.json()
 
-            # Sort papers primarily by 'publishedAt' descending, then by 'upvotes' descending
+            # Sort papers by 'publishedAt' descending, then by 'upvotes' descending
             self.papers = sorted(
                 data,
                 key=lambda x: (
@@ -26,10 +26,10 @@ class PaperManager:
                     ),
                     x.get('paper', {}).get('upvotes', 0)
                 ),
-                reverse=True  # Ensures descending order
+                reverse=True
             )
 
-            self.total_pages = (len(self.papers) + self.papers_per_page - 1) // self.papers_per_page
+            self.total_pages = max((len(self.papers) + self.papers_per_page - 1) // self.papers_per_page, 1)
             self.current_page = 1
             return True
         except requests.RequestException as e:
@@ -43,21 +43,21 @@ class PaperManager:
         title = paper.get('title', 'No title')
         paper_id = paper.get('paper', {}).get('id', '')
         url = f"https://huggingface.co/papers/{paper_id}"
-        authors = ', '.join([author.get('name', '') for author in paper.get('paper', {}).get('authors', [])])
+        authors = ', '.join([author.get('name', '') for author in paper.get('paper', {}).get('authors', [])]) or 'Unknown'
         upvotes = paper.get('paper', {}).get('upvotes', 0)
         comments = paper.get('numComments', 0)
         published_time = datetime.fromisoformat(
             paper.get('publishedAt', datetime.now(timezone.utc).isoformat()).replace('Z', '+00:00')
         )
-        time_ago_days = (datetime.now(timezone.utc) - published_time).days
+        time_diff = datetime.now(timezone.utc) - published_time
+        time_ago_days = time_diff.days
         time_ago = f"{time_ago_days} days ago" if time_ago_days > 0 else "today"
 
-        # Return HTML similar to Hacker News
         return f"""
         <tr class="athing">
             <td align="right" valign="top" class="title"><span class="rank">{rank}.</span></td>
             <td valign="top" class="votelinks">
-                <center><a href="#"><div class="votearrow"></div></a></center>
+                <center><div class="votearrow"></div></center>
             </td>
             <td class="title">
                 <a href="{url}" class="storylink" target="_blank">{title}</a>
@@ -90,9 +90,12 @@ class PaperManager:
     def next_page(self):
         if self.current_page < self.total_pages:
             self.current_page += 1
-            return self.render_papers()
-        else:
-            return self.render_papers() + "<center class='more'><span>No more papers.</span></center>"
+        return self.render_papers()
+
+    def prev_page(self):
+        if self.current_page > 1:
+            self.current_page -= 1
+        return self.render_papers()
 
 paper_manager = PaperManager()
 
@@ -102,6 +105,12 @@ def initialize_app():
     else:
         return "<div class='no-papers'>Failed to fetch papers. Please try again later.</div>"
 
+def refresh_papers():
+    if paper_manager.fetch_papers():
+        return paper_manager.render_papers()
+    else:
+        return "<div class='no-papers'>Failed to refresh papers. Please try again later.</div>"
+
 css = """
 body {
     background-color: white;
@@ -177,8 +186,14 @@ table {
     text-decoration: none;
 }
 
-.more {
-    padding: 10px;
+.more-link button {
+    background: none;
+    border: none;
+    color: #0000ff;
+    text-decoration: underline;
+    cursor: pointer;
+    font-size: 10pt;
+    padding: 0;
 }
 
 .no-papers {
@@ -207,7 +222,7 @@ demo = gr.Blocks(css=css)
 
 with demo:
     with gr.Column(elem_classes=["container"]):
-        # Header
+        # Header with Refresh Button
         gr.HTML("""
         <table border="0" cellpadding="0" cellspacing="0" width="100%" bgcolor="#ff6600">
             <tr>
@@ -216,28 +231,40 @@ with demo:
                         <b class="hnname"><a href="#" style="color: black; text-decoration: none;">Daily Papers</a></b>
                     </span>
                 </td>
+                <td align="right" style="padding: 8px;">
+                    <button id="refresh-button">Refresh</button>
+                </td>
             </tr>
         </table>
         """)
         # Paper list
         paper_list = gr.HTML()
-        # Footer with 'More' link and embedded JavaScript
-        more_button = gr.HTML("""
-        <center class='more'>
-            <a href="#" id="more-button">More</a>
-        </center>
-        <script>
-        document.getElementById('more-button').addEventListener('click', function(e) {
-            e.preventDefault();
-            document.getElementById('hidden-next-button').click();
-        });
-        </script>
-        """)
-
+        # Navigation Buttons
+        with gr.Row():
+            prev_button = gr.Button("Prev")
+            next_button = gr.Button("Next")
+        # "More" Button Styled as Link
+        with gr.Row():
+            more_button = gr.Button("More", elem_classes="more-link")
+
+    # Load papers on app start
     demo.load(initialize_app, outputs=[paper_list])
 
-    # Hidden button to trigger next page
-    hidden_next_button = gr.Button(visible=False, elem_id='hidden-next-button')
-    hidden_next_button.click(paper_manager.next_page, outputs=[paper_list])
+    # Button clicks
+    prev_button.click(paper_manager.prev_page, outputs=[paper_list])
+    next_button.click(paper_manager.next_page, outputs=[paper_list])
+    more_button.click(paper_manager.next_page, outputs=[paper_list])
+    # Bind the refresh button
+    refresh_button = gr.Button(visible=False, elem_id="refresh-hidden")
+    refresh_button.click(refresh_papers, outputs=[paper_list])
+
+    # JavaScript to bind the click event to the refresh button
+    demo.load(None, _js="""
+    function() {
+        document.getElementById('refresh-button').addEventListener('click', function() {
+            document.querySelector('button#refresh-hidden').click();
+        });
+    }
+    """)
 
 demo.launch()