Update app.py

app.py

@@ -47,25 +47,20 @@ def get_df() -> pd.DataFrame:
         
         # Prepare the DataFrame by removing 'abstract'
         logger.info("Removing 'abstract' column if present.")
-        paper_info = []
-        for _, row in tqdm.auto.tqdm(df.iterrows(), total=len(df)):
-            info = row.copy()
-            if "abstract" in info:
-                del info["abstract"]
-            paper_info.append(info)
-        df_prepared = pd.DataFrame(paper_info)
+        if 'abstract' in df.columns:
+            df = df.drop(columns=['abstract'])
         
         # Add 'paper_page' links
         logger.info("Adding 'paper_page' links.")
-        df_prepared["paper_page"] = df_prepared["arxiv_id"].apply(lambda x: f"https://huggingface.co/papers/{x}")
+        df["paper_page"] = df["arxiv_id"].apply(lambda x: f"https://huggingface.co/papers/{x}")
         
         # Verify that 'date' column exists
-        if 'date' not in df_prepared.columns:
-            logger.error("'date' column is missing from the DataFrame.")
-            df_prepared["date"] = datetime.datetime.now(timezone.utc).strftime("%Y-%m-%d")
+        if 'date' not in df.columns:
+            logger.error("'date' column is missing from the DataFrame. Filling with current date.")
+            df["date"] = datetime.datetime.now(timezone.utc).strftime("%Y-%m-%d")
         
         logger.info("DataFrame preparation complete.")
-        return df_prepared
+        return df
     except Exception as e:
         logger.error(f"Error in get_df: {e}")
         return pd.DataFrame()  # Return empty DataFrame on error
@@ -157,7 +152,6 @@ class PaperManager:
         self.paper_list = paper_list
         self.papers_per_page = papers_per_page
         self.sort_method = "hot"              # Default sort method
-        self.top_time_frame = "all time"      # Default time frame for "Top" sorting
         self.sort_papers()
         # 'current_page' and 'total_pages' are set in 'sort_papers()'
 
@@ -552,7 +546,7 @@ with demo:
 
             Once your paper is submitted, it will automatically appear in this demo.
             """)
-        # Hacker News-like Header
+        # Hacker News-like Header with "Hot" and "New" sort options
         with gr.Row():
             gr.HTML("""
             <table border="0" cellpadding="0" cellspacing="0" class="header-table">
@@ -563,33 +557,26 @@ with demo:
                         </span>
                     </td>
                     <td align="right">
-                        <!-- Future Navigation Links Can Be Added Here -->
+                        <span class="pagetop">
+                            <a href="#" id="hot_sort">Hot</a> | 
+                            <a href="#" id="new_sort">New</a>
+                        </span>
                     </td>
                 </tr>
             </table>
             """)
-        # Sort Options and Time Frame (conditionally visible)
-        with gr.Row():
-            sort_radio = gr.Radio(
-                choices=["Hot", "New", "Top"],
-                value="Hot",
-                label="Sort By",
-                interactive=True
-            )
-            time_frame_dropdown = gr.Dropdown(
-                choices=["day", "week", "month", "year", "all time"],
-                value="all time",
-                label="Time Frame for Top",
-                visible=False,
-                interactive=True
-            )
+        # 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
         # 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(),
@@ -600,30 +587,51 @@ with demo:
     prev_button.click(paper_manager.prev_page, outputs=[paper_list])
     next_button.click(paper_manager.next_page, outputs=[paper_list])
 
-    # Sort option change: Toggle visibility of time_frame_dropdown based on sort method
-    sort_radio.change(
-        fn=lambda method: gr.update(visible=True) if method.lower() == "top" else gr.update(visible=False),
-        inputs=[sort_radio],
-        outputs=[time_frame_dropdown]
+    # 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();
+            });
+        }
+        """
     )
 
-    # Sort option change: Apply sorting method with time frame if applicable
-    sort_radio.change(
-        fn=change_sort_method_ui,
-        inputs=[sort_radio, time_frame_dropdown],
+    # Assign hidden buttons to trigger sort methods
+    hot_sort.click(
+        fn=lambda: change_sort_method_ui("hot"),
+        inputs=[],
+        outputs=[paper_list]
+    )
+    new_sort.click(
+        fn=lambda: change_sort_method_ui("new"),
+        inputs=[],
         outputs=[paper_list]
     )
 
-    # Footer
-    gr.Markdown("""
-    Related useful Spaces:
-    - [Semantic Scholar Paper Recommender](https://huggingface.co/spaces/librarian-bots/recommend_similar_papers) by [davanstrien](https://huggingface.co/davanstrien)
-    - [ArXiv CS RAG](https://huggingface.co/spaces/bishmoy/Arxiv-CS-RAG) by [bishmoy](https://huggingface.co/bishmoy)
-    - [Paper Q&A](https://huggingface.co/spaces/chansung/paper_qa) by [chansung](https://huggingface.co/chansung)
-    """)
+    # Footer - Removed as per request
+    # Removed the footer markdown section
 
 
 # --- Launch the App ---
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()