Update app.py

app.py

@@ -190,27 +190,6 @@ class PaperManager:
                 df_sorted = df
         elif self.sort_method == "new":
             df_sorted = df.sort_values(by='date', ascending=False)  # Sort by 'date'
-        elif self.sort_method == "top":
-            # Filter based on the selected time frame
-            now = datetime.datetime.now(timezone.utc)
-            if self.top_time_frame == "day":
-                time_threshold = now - datetime.timedelta(days=1)
-            elif self.top_time_frame == "week":
-                time_threshold = now - datetime.timedelta(weeks=1)
-            elif self.top_time_frame == "month":
-                time_threshold = now - datetime.timedelta(days=30)
-            elif self.top_time_frame == "year":
-                time_threshold = now - datetime.timedelta(days=365)
-            elif self.top_time_frame == "all time":
-                time_threshold = datetime.datetime.min.replace(tzinfo=timezone.utc)
-            else:
-                time_threshold = datetime.datetime.min.replace(tzinfo=timezone.utc)
-            
-            # Convert 'date' column to datetime
-            df_sorted = df.copy()
-            df_sorted['date_parsed'] = pd.to_datetime(df_sorted['date'], errors='coerce').dt.tz_localize(timezone.utc, ambiguous='NaT', nonexistent='NaT')
-            df_sorted = df_sorted[df_sorted['date_parsed'] >= time_threshold]
-            df_sorted = df_sorted.sort_values(by='upvotes', ascending=False).drop(columns=['date_parsed'])
         else:
             df_sorted = df
 
@@ -222,16 +201,12 @@ class PaperManager:
 
     def set_sort_method(self, method, time_frame=None):
         """
-        Sets the sort method ('hot', 'new', 'top') and re-sorts the papers.
-        If 'top' is selected, also sets the time frame.
+        Sets the sort method ('hot', 'new') and re-sorts the papers.
         """
-        if method not in ["hot", "new", "top"]:
+        if method not in ["hot", "new"]:
             method = "hot"
         logger.info(f"Setting sort method to: {method}")
         self.sort_method = method
-        if method == "top" and time_frame:
-            self.top_time_frame = time_frame.lower()
-            logger.info(f"Setting top time frame to: {self.top_time_frame}")
         self.sort_papers()
         return True  # Assume success
 
@@ -385,15 +360,12 @@ logger.info("Scheduler shutdown registered.")
 
 # --- Gradio Interface Functions ---
 
-def change_sort_method_ui(method: str, time_frame: str = "all time") -> str:
+def change_sort_method_ui(method: str) -> str:
     """
-    Changes the sort method and, if 'top' is selected, sets the time frame.
+    Changes the sort method based on user selection.
     """
-    logger.info(f"Changing sort method to: {method} with time frame: {time_frame}")
-    if method.lower() == "top":
-        paper_manager.set_sort_method(method.lower(), time_frame)
-    else:
-        paper_manager.set_sort_method(method.lower())
+    logger.info(f"Changing sort method to: {method}")
+    paper_manager.set_sort_method(method.lower())
     return paper_manager.get_current_page_papers()
 
 
@@ -440,6 +412,14 @@ table {
     text-decoration: none;
 }
 
+.header-table .sort-links a {
+    color: black;
+    font-weight: normal;
+    font-size: 14pt;
+    margin-left: 15px;
+    cursor: pointer;
+}
+
 .itemlist .athing {
     background-color: #f6f6ef;
 }
@@ -477,6 +457,11 @@ table {
         font-size: 12pt;
     }
 
+    .sort-links a {
+        font-size: 12pt;
+        margin-left: 10px;
+    }
+
     .storylink {
         font-size: 9pt;
     }
@@ -509,6 +494,10 @@ table {
         color: black;
     }
 
+    .header-table .sort-links a {
+        color: black;
+    }
+
     .itemlist .athing {
         background-color: #1e1e1e;
     }
@@ -556,20 +545,24 @@ with demo:
                             <b class="hnname"><a href="#">Daily Papers</a></b>
                         </span>
                     </td>
-                    <td align="right">
-                        <span class="pagetop">
-                            <a href="#" id="hot_sort">Hot</a> | 
-                            <a href="#" id="new_sort">New</a>
-                        </span>
+                    <td align="right" class="sort-links">
+                        <a href="#" onclick="HotSort()">Hot</a> | 
+                        <a href="#" onclick="NewSort()">New</a>
                     </td>
                 </tr>
             </table>
+            <script>
+                function HotSort() {
+                    gradioApp().getElementById('hot_sort_button').click();
+                }
+                function NewSort() {
+                    gradioApp().getElementById('new_sort_button').click();
+                }
+            </script>
             """)
-        # Hidden components for Gradio to capture sort clicks
-        hot_sort = gr.Button("Hot", visible=False)
-        new_sort = gr.Button("New", visible=False)
-        # Sort Options and Time Frame (conditionally visible) - Removed "Top" sort
-        # Removed the time_frame_dropdown as "Top" sort is removed
+        # Hidden buttons to trigger sort methods
+        hot_sort = gr.Button("Hot Sort", visible=False, elem_id="hot_sort_button")
+        new_sort = gr.Button("New Sort", visible=False, elem_id="new_sort_button")
         # Paper list
         paper_list = gr.HTML()
         # Navigation Buttons
@@ -587,35 +580,7 @@ with demo:
     prev_button.click(paper_manager.prev_page, outputs=[paper_list])
     next_button.click(paper_manager.next_page, outputs=[paper_list])
 
-    # Sort option changes via header links
-    # Since Gradio cannot capture clicks on HTML directly, use hidden buttons and trigger them via JavaScript
-    # Add JavaScript to trigger Gradio button clicks when header links are clicked
-    demo.load(
-        fn=lambda: None,
-        outputs=[],
-        _js="""
-        const hotLink = document.getElementById('hot_sort');
-        const newLink = document.getElementById('new_sort');
-        const hotButton = document.querySelector('button#component-__block-0--hot_sort');
-        const newButton = document.querySelector('button#component-__block-0--new_sort');
-
-        if (hotLink && hotButton) {
-            hotLink.addEventListener('click', (e) => {
-                e.preventDefault();
-                hotButton.click();
-            });
-        }
-
-        if (newLink && newButton) {
-            newLink.addEventListener('click', (e) => {
-                e.preventDefault();
-                newButton.click();
-            });
-        }
-        """
-    )
-
-    # Assign hidden buttons to trigger sort methods
+    # Hidden buttons trigger sort methods
     hot_sort.click(
         fn=lambda: change_sort_method_ui("hot"),
         inputs=[],