add title search

app.py

@@ -11,6 +11,11 @@ paper_central_df = PaperCentral()
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown("# Paper Central")
 
+    with gr.Accordion(label="⭐Release notes", open=False):
+        gr.Markdown("""
+        4/10/2024 - You can now filter by Title
+        """)
+
     # Create a row for navigation buttons and calendar
     with gr.Row():
         with gr.Column(scale=1):
@@ -69,12 +74,17 @@ with gr.Blocks(css="style.css") as demo:
                 label="Conference options",
                 choices=["In proceedings"] + PaperCentral.CONFERENCES
             )
+            with gr.Row():
+                # Define a Textbox for author search
+                author_search = gr.Textbox(
+                    label="Search Authors",
+                    placeholder="Enter author name",
+                )
+                title_search = gr.Textbox(
+                    label="Search Title",
+                    placeholder="Enter keywords",
+                )
 
-            # Define a Textbox for author search
-            author_search = gr.Textbox(
-                label="Search Authors",
-                placeholder="Enter author name",
-            )
     # Define the Dataframe component to display paper data
     # List of columns in your DataFrame
     columns = paper_central_df.COLUMNS_START_PAPER_PAGE
@@ -90,6 +100,7 @@ with gr.Blocks(css="style.css") as demo:
         interactive=False,
         height=1000,
         elem_id="table",
+        wrap=True,
     )
 
 
@@ -100,6 +111,7 @@ with gr.Blocks(css="style.css") as demo:
             hf_options_list: List[str],
             conference_options_list: List[str],
             author_search_input: str,
+            title_search_input: str,
     ) -> tuple:
         """
         Moves the selected date to the next day and updates the data.
@@ -130,6 +142,7 @@ with gr.Blocks(css="style.css") as demo:
             hf_options=hf_options_list,
             conference_options=conference_options_list,
             author_search_input=author_search_input,
+            title_search_input=title_search_input,
         )
 
         # Return the new date and updated Dataframe
@@ -143,6 +156,7 @@ with gr.Blocks(css="style.css") as demo:
             hf_options_list: List[str],
             conference_options_list: List[str],
             author_search_input: str,
+            title_search_input: str,
     ) -> tuple:
         """
         Moves the selected date to the previous day and updates the data.
@@ -173,6 +187,7 @@ with gr.Blocks(css="style.css") as demo:
             hf_options=hf_options_list,
             conference_options=conference_options_list,
             author_search_input=author_search_input,
+            title_search_input=title_search_input,
         )
 
         # Return the new date and updated Dataframe
@@ -186,6 +201,7 @@ with gr.Blocks(css="style.css") as demo:
             hf_options_list: List[str],
             conference_options_list: List[str],
             author_search_input: str,
+            title_search_input: str,
     ):
         """
         Updates the data displayed in the Dataframe based on the selected date and options.
@@ -205,6 +221,7 @@ with gr.Blocks(css="style.css") as demo:
             hf_options=hf_options_list,
             conference_options=conference_options_list,
             author_search_input=author_search_input,
+            title_search_input=title_search_input,
         )
 
 
@@ -215,6 +232,7 @@ with gr.Blocks(css="style.css") as demo:
             hf_options_list: List[str],
             conference_options_list: List[str],
             author_search_input: str,
+            title_search_input: str,
     ):
 
         cat_options_update = gr.update()
@@ -231,6 +249,7 @@ with gr.Blocks(css="style.css") as demo:
                 hf_options_list,
                 conference_options_list,
                 author_search_input,
+                title_search_input,
             )
             visible = False
 
@@ -248,6 +267,7 @@ with gr.Blocks(css="style.css") as demo:
             hf_options_list: List[str],
             conference_options_list: List[str],
             author_search_input: str,
+            title_search_input: str,
     ):
         conference_options_update = gr.update()
         paper_central_component_update = gr.update()
@@ -263,6 +283,7 @@ with gr.Blocks(css="style.css") as demo:
                 hf_options_list,
                 [],
                 author_search_input,
+                title_search_input,
             )
             visible = True
 
@@ -279,6 +300,7 @@ with gr.Blocks(css="style.css") as demo:
         hf_options,
         conference_options,
         author_search,
+        title_search,
     ]
 
     # Set up the event listener for the author search
@@ -288,6 +310,12 @@ with gr.Blocks(css="style.css") as demo:
         outputs=paper_central_component,
     )
 
+    title_search.submit(
+        fn=update_data,
+        inputs=inputs,
+        outputs=paper_central_component,
+    )
+
     # Set up the event listener for the 'Next Day' button
     next_day_btn.click(
         fn=go_to_next_day,
@@ -331,11 +359,34 @@ with gr.Blocks(css="style.css") as demo:
     )
 
     # Load the initial data when the app starts
+    request = gr.Request()
+
+
+    def echo(request: gr.Request):
+        """
+        Echoes the input text and prints request details.
+
+        Args:
+            text (str): The input text from the user.
+            request (gr.Request): The HTTP request object.
+
+        Returns:
+            str: The echoed text.
+        """
+        if request:
+            print("Request headers dictionary:", dict(request.headers))
+            print("IP address:", request.client.host)
+            print("Query parameters:", dict(request.query_params))
+            print("Session hash:", request.session_hash)
+
+
     demo.load(
         fn=update_data,
         inputs=inputs,
         outputs=paper_central_component,
         api_name=False,
+    ).then(
+        fn=echo,
     )
 
 

constants.py

@@ -5,4 +5,4 @@ DATASET_DAILY_PAPERS = "hysts-bot-data/daily-papers"
 DATASET_DAILY_PAPERS_STATS = "hysts-bot-data/daily-papers-stats"
 DATASET_COMMUNITY_SCIENCE = "huggingface/community-science-paper-v2"
 # DATASET_PAPER_CENTRAL = "huggingface/paper-central-data"
-DATASET_PAPER_CENTRAL = "huggingface/paper-central-data-2"
+DATASET_PAPER_CENTRAL = "huggingface/paper-central-data"

df/PaperCentral.py

@@ -276,6 +276,7 @@ class PaperCentral:
             hf_options: Optional[List[str]] = None,
             conference_options: Optional[List[str]] = None,
             author_search_input: Optional[str] = None,
+            title_search_input: Optional[str] = None,
     ) -> gr.update:
         """
         Filter the DataFrame based on selected date and options, and prepare it for display.
@@ -293,6 +294,22 @@ class PaperCentral:
         # Start with the initial columns to display
         columns_to_show: List[str] = PaperCentral.COLUMNS_START_PAPER_PAGE.copy()
 
+        if title_search_input:
+            if 'title' not in columns_to_show:
+                columns_to_show.append('authors')
+
+            search_string = title_search_input.lower()
+
+            def title_match(title):
+                if isinstance(title, str):
+                    # If authors_list is a single string
+                    return search_string in title.lower()
+                else:
+                    # Handle unexpected data types
+                    return False
+
+            filtered_df = filtered_df[filtered_df['title'].apply(title_match)]
+
         if author_search_input:
             if 'authors' not in columns_to_show:
                 columns_to_show.append('authors')