Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import arxiv
+from datasets import load_dataset
 
 search = arxiv.Search(
     query = "cs.LG",
@@ -7,9 +8,17 @@ search = arxiv.Search(
     sort_by = arxiv.SortCriterion.SubmittedDate
 )
 
+#HF_TOKEN = os.getenv("HF_TOKEN") # might use in the future
+
+def hf_data_upload(user_id, paper_id, feedback):
+  ds = load_dataset("CShorten/Last-Week-on-ArXiv")
+  new_row = {"user_id": user_id, "paper_id": paper_id, "feedback": feedback}
+  ds = ds.add_item(new_row)
+  ds.push_to_hub("CShorten/Last-Week-on-ArXiv") 
+
 with gr.Blocks() as demo:
   gr.Markdown("<center><h1>My ArXiv</h1></center>")
-  gr.Textbook(placeholder="Enter user id for personalization: ")
+  user_id = gr.Textbox(placeholder="Enter user id for personalization: ")
   with gr.Column():
     for arxiv_paper in search.results():
       with gr.Column():
@@ -17,8 +26,10 @@ with gr.Blocks() as demo:
           gr.Markdown("<center><h3>" + arxiv_paper.title + "</h3></center>")
           gr.Markdown(arxiv_paper.summary)
         with gr.Row():
-          button = gr.Button("More like this! 😎")
+          more_button = gr.Button("More like this! 😎")
+          more_button.click(hf_data_upload(user_id, arxiv_paper.title, 1)
           #button.click(flip_image, inputs=image_input, outputs=image_output)
-          gr.Button("Less like this! 😕")
+          less_button = gr.Button("Less like this! 😕")
+          less_button.click(hf_data_upload(user_id, arxiv_paper.title, 0)
 
 demo.launch()