Add filter by date

README.md

@@ -4,7 +4,7 @@ emoji: 📊
 colorFrom: pink
 colorTo: pink
 sdk: gradio
-sdk_version: 4.13.0
+sdk_version: 4.14.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -3,6 +3,7 @@
 import os
 
 import gradio as gr
+from gradio_calendar import Calendar
 
 from papers import PaperList, get_df
 from update_scheduler import UpdateScheduler
@@ -19,6 +20,10 @@ if (SPACE_ID := os.getenv("SPACE_ID")) is not None:
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
+    with gr.Group():
+        with gr.Row():
+            start_date = Calendar(label="Start date", type="datetime", value="2023-05-05")
+            end_date = Calendar(label="End date", type="datetime")
     df = gr.Dataframe(
         value=paper_list.df_prettified,
         datatype=paper_list.column_datatype,
@@ -30,5 +35,13 @@ with gr.Blocks(css="style.css") as demo:
         wrap=True,
     )
 
+    gr.on(
+        triggers=[start_date.change, end_date.change],
+        fn=paper_list.filter_by_date,
+        inputs=[start_date, end_date],
+        outputs=df,
+        api_name=False,
+    )
+
 if __name__ == "__main__":
     demo.queue(api_open=False).launch()

papers.py

@@ -97,3 +97,11 @@ class PaperList:
     @property
     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:
+        df = self.df_raw.copy()
+        df["date"] = pd.to_datetime(df["date"])
+        df = df[(df["date"] >= start_date) & (df["date"] <= end_date)]
+        df["date"] = df["date"].dt.strftime("%Y-%m-%d")
+        df_prettified = self._prettifier(df).loc[:, self.column_names]
+        return df_prettified

requirements.txt

@@ -1,5 +1,6 @@
 apscheduler==3.10.4
-gradio==4.13.0
+gradio==4.14.0
+gradio_calendar==0.0.2
 huggingface_hub==0.20.2
 pandas==2.1.4
 requests==2.31.0