Update

app.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python
 
+import datetime
+
 import gradio as gr
 import pandas as pd
 from apscheduler.schedulers.background import BackgroundScheduler
@@ -38,6 +40,16 @@ def update_num_papers(df: pd.DataFrame) -> str:
     return f"{len(df)} / {len(paper_list.df_raw)}"
 
 
+def search(
+    start_date: datetime.datetime,
+    end_date: datetime.datetime,
+    search_title: str,
+    search_abstract: str,
+    max_num_to_retrieve: int,
+) -> pd.DataFrame:
+    return paper_list.search(start_date, end_date, search_title, search_abstract, max_num_to_retrieve)
+
+
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Group():
@@ -77,7 +89,7 @@ with gr.Blocks(css="style.css") as demo:
 
     gr.on(
         triggers=[start_date.change, end_date.change, search_title.submit, search_abstract.submit],
-        fn=paper_list.search,
+        fn=search,
         inputs=[start_date, end_date, search_title, search_abstract, max_num_to_retrieve],
         outputs=df,
         api_name=False,