Update app.py

app.py

@@ -1,16 +1,15 @@
+# coding:utf-8
 import gradio as gr
-import rebiber 
+import rebiber
 import os
 import uuid
- 
 
-# Load Bib Database 
-filepath = os.path.abspath(rebiber.__file__).replace("__init__.py","")
+# Load Bib Database
+filepath = os.path.abspath(rebiber.__file__).replace("__init__.py", "")
 bib_list_path = os.path.join(filepath, "bib_list.txt")
 abbr_tsv_path = "abbr.tsv"
 
 bib_db = rebiber.construct_bib_db(bib_list_path, start_dir=filepath)
-
 abbr_dict = rebiber.normalize.load_abbr_tsv(abbr_tsv_path)
 
 
@@ -23,55 +22,54 @@ def process(input_bib, shorten, remove_keys, deduplicate, sort):
     with open(f"input_{random_id}.bib", "w") as f:
         f.write(input_bib.replace("\t", "    "))
     all_bib_entries = rebiber.load_bib_file(f"input_{random_id}.bib")
-    print("# Input Bib Entries:", len(all_bib_entries))
+    # print("# Input Bib Entries:", len(all_bib_entries))
     abbr_dict_pass = []
     if shorten:
         abbr_dict_pass = abbr_dict
-    rebiber.normalize_bib(bib_db, all_bib_entries, f"output_{random_id}.bib",
-                          abbr_dict=abbr_dict_pass,
-                          deduplicate=deduplicate,
-                          sort=sort,
-                          removed_value_names=remove_keys)
+    rebiber.normalize_bib(
+        bib_db,
+        all_bib_entries,
+        f"output_{random_id}.bib",
+        abbr_dict=abbr_dict_pass,
+        deduplicate=deduplicate,
+        sort=sort,
+        removed_value_names=remove_keys,
+    )
     with open(f"output_{random_id}.bib") as f:
         output_bib = f.read().replace("\n ", "\n    ")
     # delete both files
-    # print(output_bib)
     return output_bib, random_id, gr.update(visible=True)
 
 
+def download_file(ex_uuid):
+    global download_content
+    # Replace this with your code to generate/download the file
+    file_path = f"output_{ex_uuid}.bib"
+    download_content.update(visible=False)
+    return file_path, gr.update(visible=True)
+    
+
 example_input = """
-@article{lin2020birds,
-    title={Birds have four legs?! NumerSense: Probing Numerical Commonsense Knowledge of Pre-trained Language Models},
-    author={Lin, Bill Yuchen and Lee, Seyeon and Khanna, Rahul and Ren, Xiang},
-    journal={arXiv preprint arXiv:2005.00683},
+    @article{lin2020birds,
+        title={Birds have four legs?! NumerSense: Probing Numerical Commonsense Knowledge of Pre-trained Language Models},
+        author={Lin, Bill Yuchen and Lee, Seyeon and Khanna, Rahul and Ren, Xiang},
+        journal={arXiv preprint arXiv:2005.00683},
+        year={2020}
+    } 
+    @inproceedings{Lin2020CommonGenAC,
+    title={CommonGen: A Constrained Text Generation Challenge for Generative Commonsense Reasoning},
+    author={Bill Yuchen Lin and Minghan Shen and Wangchunshu Zhou and Pei Zhou and Chandra Bhagavatula and Yejin Choi and Xiang Ren},
+    booktitle={Findings},
     year={2020}
-} 
-@inproceedings{Lin2020CommonGenAC,
-  title={CommonGen: A Constrained Text Generation Challenge for Generative Commonsense Reasoning},
-  author={Bill Yuchen Lin and Minghan Shen and Wangchunshu Zhou and Pei Zhou and Chandra Bhagavatula and Yejin Choi and Xiang Ren},
-  booktitle={Findings},
-  year={2020}
-}
-""" 
+    }
+"""
 
 examples = [[example_input]]
 
-
-# iface = gr.Interface(fn=process,
-#     inputs=gr.inputs.Textbox(lines=30, label="Input BIB"),
-#     outputs=gr.outputs.Textbox(label="Output BIB").style(show_copy_button=True),
-#     examples=examples,
-#     allow_flagging="never"
-#     )
-
-
-
-
-
 with gr.Blocks() as demo:
-    
+
     gr.Markdown(
-            '''# Rebiber: A tool for normalizing bibtex with official info.
+        """# Rebiber: A tool for normalizing bibtex with official info.
                 <table> 
                 <tr>
                 <td>
@@ -98,16 +96,34 @@ with gr.Blocks() as demo:
                 Apart from handling outdated arXiv citations, __Rebiber__ also normalizes citations in a unified way (DBLP-style), supporting abbreviation and value selection.  
                 
                 </span>
-            '''
+            """
     )
-    
+
     with gr.Row():
         with gr.Column(scale=3):
-            input_bib = gr.Textbox(lines=15, label="Input BIB", value=example_input, interactive=True)
-            removekeys = gr.CheckboxGroup(["url", "biburl", "address", "publisher", "pages", "doi", "volume", "bibsource"], 
-                                value=[False, False, False, False, False, False, False, False],
-                                label="Remove Keys", info="Which keys to remove?")
-            shorten = gr.Checkbox(label="Abbreviation", info="Shorten the conference/journal names (e.g., `Proceedings of the 2020 International Conference of ...` --> `Proc. of ICML')", value=False)            
+            input_bib = gr.Textbox(
+                lines=15, label="Input BIB", value=example_input, interactive=True
+            )
+            removekeys = gr.CheckboxGroup(
+                [
+                    "url",
+                    "biburl",
+                    "address",
+                    "publisher",
+                    "pages",
+                    "doi",
+                    "volume",
+                    "bibsource",
+                ],
+                # value=[False, False, False, False, False, False, False, False],
+                label="Remove Keys",
+                info="Which keys to remove?",
+            )
+            shorten = gr.Checkbox(
+                label="Abbreviation",
+                info="Shorten the conference/journal names (e.g., `Proceedings of the 2020 International Conference of ...` --> `Proc. of ICML')",
+                value=False,
+            )
             dedup = gr.Checkbox(label="Deduplicate entries.", value=False)
             sort = gr.Checkbox(label="Sort alphabetically by ID.", value=False)
             with gr.Row():
@@ -115,22 +131,32 @@ with gr.Blocks() as demo:
                 button = gr.Button("Submit")
             ex_uuid = gr.Text(label="UUID")
             ex_uuid.visible = False
+            
         with gr.Column(scale=3):
-            output=gr.Textbox(label="Output BIB (Note that you can copy the output bib file by clicking the top-right button.)").style(show_copy_button=True, interactive=False)        
+            output = gr.Textbox(
+                label="Output BIB (Note that you can copy the output bib file by clicking the top-right button.)",
+                show_copy_button=True,
+                interactive=False,
+            )
             download_btn = gr.Button("Generate Bib File")
             download_btn.visible = False
-            download_content = gr.outputs.File()
-            download_content.visible = False 
-    def download_file(ex_uuid):
-        global download_content
-        # Replace this with your code to generate/download the file
-        file_path = f"output_{ex_uuid}.bib"
-        download_content.update(visible=False)
-        return file_path, gr.update(visible=True)
-    download_btn.click(download_file, inputs=ex_uuid, outputs=[download_content,download_content])  
-    button.click(process, inputs=[input_bib, shorten, removekeys, dedup, sort], outputs=[output, ex_uuid, download_btn], api_name = "process")
+            download_content = gr.File()
+            download_content.visible = False
+
+    download_btn.click(
+        download_file, inputs=ex_uuid, outputs=[download_content, download_content]
+    )
+    
+    button.click(
+        process,
+        inputs=[input_bib, shorten, removekeys, dedup, sort],
+        outputs=[output, ex_uuid, download_btn],
+        api_name="process",
+    )
+
     def clean(text):
         return ""
+
     clr_button.click(clean, input_bib, input_bib)
     # gr.Interface(fn=process,
     # outputs=gr.outputs.Textbox(label="Output BIB").style(show_copy_button=True),
@@ -167,4 +193,4 @@ Ren, Xiang},
  url = {https://aclanthology.org/2020.emnlp-main.557},
  year = {2020}
 }   
-"""
+"""