Update app.py

app.py

@@ -36,9 +36,8 @@ def count_nans(row):
     
 gemini_api_key = os.getenv("GEMINI_API_KEY")
 hf_token = os.getenv("HF_TOKEN")
-
-dataset_repo_id = "chansung/auto-paper-qa2"
-request_arxiv_repo_id="chansung/requested-arxiv-ids-3"
+dataset_repo_id = os.getenv("SOURCE_DATA_REPO_ID") # "chansung/auto-paper-qa2"
+request_arxiv_repo_id = os.getenv("REQUEST_DATA_REPO_ID") # "chansung/requested-arxiv-ids-3"
 
 ds = datasets.load_dataset(dataset_repo_id)
 request_ds = datasets.load_dataset(request_arxiv_repo_id)
@@ -387,29 +386,9 @@ with gr.Blocks(css=STYLE, theme=gr.themes.Soft()) as demo:
     with gr.Column(elem_id="control-panel", elem_classes=["group"]):
         with gr.Column():
             with gr.Row():
-                year_dd = gr.Dropdown(
-                    sorted_year,
-                    value=last_year,
-                    label="Year",
-                    interactive=True,
-                    filterable=False
-                )
-    
-                month_dd = gr.Dropdown(
-                    sorted_month,
-                    value=last_month,
-                    label="Month",
-                    interactive=True,
-                    filterable=False
-                )
-    
-                day_dd = gr.Dropdown(
-                    sorted_day,
-                    value=last_day,
-                    label="Day",
-                    interactive=True,
-                    filterable=False
-                )
+                year_dd = gr.Dropdown(sorted_year, value=last_year, label="Year", interactive=True, filterable=False)    
+                month_dd = gr.Dropdown(sorted_month, value=last_month, label="Month", interactive=True, filterable=False)    
+                day_dd = gr.Dropdown(sorted_day, value=last_day, label="Day", interactive=True, filterable=False)
                 
             papers_dd = gr.Dropdown(
                 list(set([paper["title"] for paper in last_papers])),
@@ -444,6 +423,7 @@ with gr.Blocks(css=STYLE, theme=gr.themes.Soft()) as demo:
             hf_paper_link = gr.Markdown(
                 "[![Paper page](https://huggingface.co/datasets/huggingface/badges/resolve/main/paper-page-md.svg)](https://huggingface.co/papers/%s)" % selected_paper['arxiv_id']
             )
+            gr.Button("Chat about the paper", interactive=False)
             
         summary = gr.Markdown(f"{selected_paper['summary']}", elem_classes=["small-font"])