Update app.py

app.py

@@ -365,8 +365,11 @@ def change_sort_method_ui(method: str) -> str:
     Changes the sort method based on user selection.
     """
     logger.info(f"Changing sort method to: {method}")
-    paper_manager.set_sort_method(method.lower())
-    return paper_manager.get_current_page_papers()
+    success = paper_manager.set_sort_method(method.lower())
+    if success:
+        return paper_manager.get_current_page_papers()
+    else:
+        return "<div class='no-papers'>Failed to change sort method.</div>"
 
 
 # --- CSS Styling ---
@@ -545,6 +548,7 @@ with demo:
 
             Once your paper is submitted, it will automatically appear in this demo.
             """)
+        
         # Hacker News-like Header with "Hot" and "New" sort options
         with gr.Row():
             # Left side: Site title
@@ -557,44 +561,54 @@ with demo:
                             </span>
                         </td>
                         <td align="right" class="sort-buttons">
-                            <button onclick="gradioApp().getElementById('hot_sort').click()">Hot</button> | 
-                            <button onclick="gradioApp().getElementById('new_sort').click()">New</button>
+                            <!-- Removed custom HTML buttons -->
                         </td>
                     </tr>
                 </table>
             """, show_label=False)
-        # Hidden buttons to trigger sort methods
-        hot_sort = gr.Button("Hot Sort", visible=False, elem_id="hot_sort")
-        new_sort = gr.Button("New Sort", visible=False, elem_id="new_sort")
+            # Right side: Gradio Buttons for "Hot" and "New"
+            with gr.Column(elem_classes=["sort-buttons"]):
+                hot_button = gr.Button("Hot", elem_id="hot_button")
+                new_button = gr.Button("New", elem_id="new_button")
+        
         # Paper list
         paper_list = gr.HTML()
+        
         # Navigation Buttons
         with gr.Row():
             prev_button = gr.Button("Prev")
             next_button = gr.Button("Next")
-
+    
     # Load papers on app start
     demo.load(
         fn=lambda: paper_manager.get_current_page_papers(),
         outputs=[paper_list]
     )
-
+    
     # Button clicks for pagination
-    prev_button.click(paper_manager.prev_page, outputs=[paper_list])
-    next_button.click(paper_manager.next_page, outputs=[paper_list])
-
-    # Hidden buttons trigger sort methods
-    hot_sort.click(
+    prev_button.click(
+        fn=lambda: paper_manager.prev_page(),
+        inputs=[],
+        outputs=[paper_list]
+    )
+    next_button.click(
+        fn=lambda: paper_manager.next_page(),
+        inputs=[],
+        outputs=[paper_list]
+    )
+    
+    # Gradio Buttons trigger sort methods directly
+    hot_button.click(
         fn=lambda: change_sort_method_ui("hot"),
         inputs=[],
         outputs=[paper_list]
     )
-    new_sort.click(
+    new_button.click(
         fn=lambda: change_sort_method_ui("new"),
         inputs=[],
         outputs=[paper_list]
     )
-
+    
     # Footer - Removed as per request
     # Removed the footer markdown section
 
@@ -602,4 +616,4 @@ with demo:
 # --- Launch the App ---
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()