Update app.py

app.py

@@ -1,16 +1,12 @@
 #!/usr/bin/env python
 
 import datetime
-import operator
 import pandas as pd
 import tqdm.auto
 from apscheduler.schedulers.background import BackgroundScheduler
 from huggingface_hub import HfApi
 
 import gradio as gr
-from gradio_calendar import Calendar
-import datasets
-import requests
 
 from datetime import timezone  # Ensure timezone is imported
 
@@ -95,7 +91,7 @@ class Prettifier:
 
 class PaperList:
     """
-    Manages the list of papers, including search functionality.
+    Manages the list of papers.
     """
     COLUMN_INFO = [
         ["arxiv_id", "str"],          # Added arxiv_id
@@ -121,27 +117,11 @@ class PaperList:
     def column_datatype(self):
         return [col[1] for col in self.COLUMN_INFO]
 
-    def search(
-        self,
-        title_search_query: str,
-        max_num_to_retrieve: int = 1000,  # Set a high default to include all if not specified
-    ) -> pd.DataFrame:
+    def get_prettified_df(self) -> pd.DataFrame:
         """
-        Filters the DataFrame based on the title search query and limits the number of results.
+        Returns the prettified DataFrame.
         """
-        df = self.df_raw.copy()
-
-        # Filter by title if search query is provided
-        if title_search_query:
-            df = df[df["title"].str.contains(title_search_query, case=False, na=False)]
-        
-        # Limit the number of papers to retrieve if max_num_to_retrieve is set
-        if max_num_to_retrieve:
-            df = df.head(max_num_to_retrieve)
-
-        # Prettify the DataFrame
-        df_prettified = self._prettifier(df).loc[:, self.column_names]
-        return df_prettified
+        return self.df_prettified
 
 
 # --- Sorting and Pagination Management ---
@@ -154,7 +134,6 @@ class PaperManager:
         self.paper_list = paper_list
         self.papers_per_page = papers_per_page
         self.sort_method = "hot"              # Default sort method
-        self.current_search_query = ""        # Initialize with no search query
         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()'
@@ -181,14 +160,10 @@ class PaperManager:
 
     def sort_papers(self):
         """
-        Sorts the papers based on the current sort method and search query.
+        Sorts the papers based on the current sort method.
         """
         df = self.paper_list.df_raw.copy()
 
-        # Apply search filter if a search query exists
-        if self.current_search_query:
-            df = df[df["title"].str.contains(self.current_search_query, case=False, na=False)]
-
         if self.sort_method == "hot":
             if not df.empty:
                 df = df.drop(columns=['score'], errors='ignore')  # Remove existing 'score' column if present
@@ -242,15 +217,6 @@ class PaperManager:
         self.sort_papers()
         return True  # Assume success
 
-    def set_search_query(self, query: str):
-        """
-        Sets the current search query and re-sorts the papers.
-        """
-        print(f"Setting search query to: {query}")
-        self.current_search_query = query
-        self.sort_papers()
-        return True  # Assume success
-
     def get_current_page_papers(self) -> str:
         """
         Retrieves the HTML string of the current page's papers.
@@ -399,22 +365,6 @@ def refresh_papers_ui() -> str:
     return paper_manager.refresh()
 
 
-def search_papers_ui(query: str) -> str:
-    """
-    Searches for papers based on the title search query.
-    """
-    paper_manager.set_search_query(query)
-    return paper_manager.get_current_page_papers()
-
-
-def clear_search_ui() -> str:
-    """
-    Clears the current search query and refreshes the paper list.
-    """
-    paper_manager.set_search_query("")
-    return paper_manager.get_current_page_papers()
-
-
 # --- CSS Styling ---
 
 css = """
@@ -592,16 +542,6 @@ with demo:
                 </tr>
             </table>
             """)
-        # Search Bar and Clear Search Button
-        with gr.Row():
-            search_box = gr.Textbox(
-                label="Search Papers by Title",
-                placeholder="Enter keywords to search...",
-                lines=1,
-                interactive=True
-            )
-            search_button = gr.Button("Search")
-            clear_search_button = gr.Button("Clear Search")
         # Sort Options and Time Frame (conditionally visible)
         with gr.Row():
             sort_radio = gr.Radio(
@@ -667,20 +607,13 @@ with demo:
         outputs=[paper_list]
     )
 
-    # Search functionality
-    search_button.click(
-        fn=search_papers_ui,
-        inputs=[search_box],
-        outputs=[paper_list]
-    )
-
-    # Clear search functionality
-    clear_search_button.click(
-        fn=clear_search_ui,
-        inputs=None,
-        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)
+    """)
 
 
 # --- Launch the App ---