add manual and timer for reset project

app.py

@@ -24,13 +24,24 @@ def category_radio(cat):
         return 'cs.RO'
 
 def comment_radio(com):
-    if com == 'CVPR':
-        return 'CVPR'
-    else:
+    if com == 'None':
         return None
+    else:
+        return com
     
-def recommend_link(recs):
-    return recs
+def reset_project():
+    file_path = 'arxiv-scrape.csv'
+    if os.path.exists(file_path):
+        os.remove(file_path)
+        logging.info(f"{file_path} has been deleted. Delete reset_project() if you want to persist recommended papers.")
+
+    api_key = os.getenv('PINECONE_API_KEY')
+    index = os.getenv('INDEX_NAME')
+    pc = Pinecone(api_key = api_key)
+    if index in pc.list_indexes().names():
+        pc.delete_index(index)
+        logging.info(f"{index} index has been deleted from the vectordb. Delete reset_project() if you want to persist recommended papers.")
+    return f"{file_path} has been deleted.<br />{index} index has been deleted from the vectordb.<br />"
 
 with gr.Blocks() as demo:
 
@@ -45,7 +56,7 @@ with gr.Blocks() as demo:
     radio_arxiv_category_name.change(fn = category_radio, inputs= radio_arxiv_category_name, outputs= arxiv_category_name)
 
     arxiv_comment_query = gr.State([])
-    radio_arxiv_comment_query = gr.Radio(['CVPR', 'None'], label="ArXiv Comment Query")
+    radio_arxiv_comment_query = gr.Radio(['CVPR', 'ACL', 'TACL', 'JAIR', 'IJRR', 'None'], label="ArXiv Comment Query")
     radio_arxiv_comment_query.change(fn = comment_radio, inputs= radio_arxiv_comment_query, outputs= arxiv_comment_query)
 
     threshold = gr.Slider(minimum= 0.70, maximum= 0.99, label="Similarity Score Threshold")
@@ -54,10 +65,19 @@ with gr.Blocks() as demo:
 
     rec_output = gr.Markdown(label = "Recommended Papers")
 
+    reset_output = gr.Markdown(label = "Reset Declaration")
+
     init_btn = gr.Button("Initialize")
 
     rec_btn = gr.Button("Recommend")
 
+    reset_btn = gr.Button("Reset")
+
+    timer = gr.Timer(interval=600)
+    timer.tick(reset_project)
+
+    reset_btn.click(fn = reset_project, inputs= [], outputs= [reset_output])
+
     @init_btn.click(inputs= [zotero_api_key, zotero_library_id, zotero_tag], outputs= [init_output])
     def init(zotero_api_key, zotero_library_id, zotero_tag, hf_api_key = HF_API_KEY,  pinecone_api_key = PINECONE_API_KEY, index_name = INDEX_NAME,  namespace_name = NAMESPACE_NAME):
 
@@ -95,16 +115,4 @@ with gr.Blocks() as demo:
 
         return results
 
-file_path = 'arxiv-scrape.csv'
-if os.path.exists(file_path):
-    os.remove(file_path)
-    logging.info(f"{file_path} has been deleted. Delete this part of the code if you want to persist recommended papers.")
-
-api_key = os.getenv('PINECONE_API_KEY')
-index = os.getenv('INDEX_NAME')
-pc = Pinecone(api_key = api_key)
-if index in pc.list_indexes().names():
-    pc.delete_index(index)
-    logging.info(f"{index} index has been deleted from the vectordb. Delete this part of the code if you want to persist recommended papers.")
-
 demo.launch(share = True)