|
import gradio as gr |
|
import rebiber |
|
import os |
|
import uuid |
|
|
|
|
|
|
|
|
|
filepath = os.path.abspath(rebiber.__file__).replace("__init__.py","") |
|
bib_list_path = os.path.join(filepath, "bib_list.txt") |
|
bib_db = rebiber.construct_bib_db(bib_list_path, start_dir=filepath) |
|
|
|
|
|
|
|
|
|
def process(input_bib): |
|
random_id = uuid.uuid4().hex |
|
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)) |
|
rebiber.normalize_bib(bib_db, all_bib_entries, f"output_{random_id}.bib") |
|
with open(f"output_{random_id}.bib") as f: |
|
output_bib = f.read().replace("\n ", "\n ") |
|
|
|
|
|
return output_bib |
|
|
|
|
|
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}, |
|
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" |
|
) |
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
iface.launch() |
|
|
|
|
|
""" |
|
@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{lin2020birds, |
|
address = {Online}, |
|
author = {Lin, Bill Yuchen and |
|
Lee, Seyeon and |
|
Khanna, Rahul and |
|
Ren, Xiang}, |
|
booktitle = {Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP)}, |
|
doi = {10.18653/v1/2020.emnlp-main.557}, |
|
pages = {6862--6868}, |
|
publisher = {Association for Computational Linguistics}, |
|
title = {{B}irds have four legs?! {N}umer{S}ense: {P}robing {N}umerical {C}ommonsense {K}nowledge of {P}re-{T}rained {L}anguage {M}odels}, |
|
url = {https://aclanthology.org/2020.emnlp-main.557}, |
|
year = {2020} |
|
} |
|
""" |
