Add search by title

app.py

@@ -24,6 +24,7 @@ with gr.Blocks(css="style.css") as demo:
         with gr.Row():
             start_date = Calendar(label="Start date", type="datetime", value="2023-05-05")
             end_date = Calendar(label="End date", type="datetime")
+        search_title = gr.Textbox(label="Search by title")
     df = gr.Dataframe(
         value=paper_list.df_prettified,
         datatype=paper_list.column_datatype,
@@ -36,12 +37,12 @@ with gr.Blocks(css="style.css") as demo:
     )
 
     gr.on(
-        triggers=[start_date.change, end_date.change],
-        fn=paper_list.filter_by_date,
-        inputs=[start_date, end_date],
+        triggers=[start_date.change, end_date.change, search_title.change],
+        fn=paper_list.search,
+        inputs=[start_date, end_date, search_title],
         outputs=df,
         api_name=False,
     )
 
 if __name__ == "__main__":
-    demo.queue(api_open=False).launch()
+    demo.queue(api_open=False).launch(show_api=False)

papers.py

@@ -98,10 +98,16 @@ class PaperList:
     def column_datatype(self):
         return list(map(operator.itemgetter(1), self.COLUMN_INFO))
 
-    def filter_by_date(self, start_date: datetime.datetime, end_date: datetime.datetime) -> pd.DataFrame:
+    def search(self, start_date: datetime.datetime, end_date: datetime.datetime, title: str) -> pd.DataFrame:
         df = self.df_raw.copy()
         df["date"] = pd.to_datetime(df["date"])
+
+        # Filter by date
         df = df[(df["date"] >= start_date) & (df["date"] <= end_date)]
         df["date"] = df["date"].dt.strftime("%Y-%m-%d")
+
+        # Filter by title
+        df = df[df["title"].str.contains(title, case=False)]
+
         df_prettified = self._prettifier(df).loc[:, self.column_names]
         return df_prettified