Add precomputing acl data

app.py

@@ -5,12 +5,13 @@ from reference_string_parsing import *
 from controlled_summarization import *
 from dataset_extraction import *
 
+from controlled_summarization import recommended_kw
 import requests
 
 # Example Usage
-#url = "https://arxiv.org/pdf/2305.14996.pdf"
-#dest_folder = "./examples/"
-#download_pdf(url, dest_folder)
+# url = "https://arxiv.org/pdf/2305.14996.pdf"
+# dest_folder = "./examples/"
+# download_pdf(url, dest_folder)
 
 
 with gr.Blocks(css="#htext span {white-space: pre-line}") as demo:
@@ -31,17 +32,20 @@ with gr.Blocks(css="#htext span {white-space: pre-line}") as demo:
                             gr.Markdown("* Set the length of text used for summarization. Length 0 will exert no control over length.")
                             # ctrlsum_file_beams = gr.Number(label="Number of beams for beam search", value=1, precision=0)
                             # ctrlsum_file_sequences = gr.Number(label="Number of generated summaries", value=1, precision=0)
-                            ctrlsum_file_length = gr.Slider(0,300,step=50, label="Length")
-                            ctrlsum_file_keywords = gr.Textbox(label="Keywords",max_lines=1)
+                            ctrlsum_file_length = gr.Radio(label="Length", value=0, choices=[0, 50, 100, 200])
+                            kw = gr.Radio(visible=False)
+                            ctrlsum_file_keywords = gr.Textbox(label="Keywords", max_lines=1)
                         with gr.Row():
                             ctrlsum_file_btn = gr.Button("Generate")
                     ctrlsum_file_output = gr.Textbox(
                         elem_id="htext",
                         label="Summary",
                     )
-                ctrlsum_file_examples = gr.Examples(examples=[["examples/H01-1042_body.txt", 50, "automatic evaluation technique", "",""],["examples/H01-1042.pdf", 0, "automatic evaluation technique","",""]],
-                                                inputs=[ctrlsum_file, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_str, ctrlsum_url])
-
+                ctrlsum_file_examples = gr.Examples(
+                    examples=[["examples/H01-1042_body.txt", 50, "automatic evaluation technique", "", ""],
+                              ["examples/H01-1042.pdf", 0, "automatic evaluation technique", "", ""]],
+                    inputs=[ctrlsum_file, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_str, ctrlsum_url
+                            ])
 
 
 
@@ -51,13 +55,34 @@ with gr.Blocks(css="#htext span {white-space: pre-line}") as demo:
             outputs=[ctrlsum_file_output, ctrlsum_str, ctrlsum_file]
         )
         def clear():
-            return None,0,None, None
+            return None, 0, None, None, gr.Radio(visible=False)
+
+
+        def update_url(url):
+            if url in recommended_kw.keys():
+                keywords = recommended_kw[url]
+                if keywords != None:
+                    return None, None, gr.Radio(choices=keywords[:3], label="Recommended Keywords", visible=True,
+                                                interactive=True)
 
+            return None, None, gr.Radio(visible=False)
+
+
+        ctrlsum_file.upload(clear, inputs=None,
+                            outputs=[ctrlsum_str, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_url, kw])
+        ctrlsum_url.input(update_url, inputs=ctrlsum_url, outputs=[ctrlsum_str, ctrlsum_file, kw])
 
-        ctrlsum_file.upload(clear, inputs=None,outputs=[ctrlsum_str,ctrlsum_file_length,ctrlsum_file_keywords, ctrlsum_url])
-        ctrlsum_url.input(clear, inputs=None, outputs=[ctrlsum_str, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_file])
         ctrlsum_str.input(clear, inputs=None,
-                          outputs=[ctrlsum_url, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_file])
+                          outputs=[ctrlsum_url, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_file, kw])
+
+
+
+        def select_kw(env: gr.SelectData):
+            return env.value
+
+
+        kw.select(select_kw, None, ctrlsum_file_keywords)
+
         # Reference String Parsing
         with gr.TabItem("Reference String Parsing"):
             gr.Markdown(rsp_title_md)

requirements.txt

@@ -1,4 +1,5 @@
 pip==23.2.1
 torch==1.12.0
-SciAssist==0.0.41
+SciAssist==0.1.3
 nltk~=3.7
+pytest