import gradio as gr
from clean_bibtex.clean_bibtex import get_url, get_dblp_bibtext, parse_bibtext_file_titles

DEFAULT_TEXT = """@inproceedings{DBLP:conf/naacl/DevlinCLT19,
  author    = {Jacob Devlin and
               Ming{-}Wei Chang and
               Kenton Lee and
               Kristina Toutanova},
  editor    = {Jill Burstein and
               Christy Doran and
               Thamar Solorio},
  title     = {{BERT:} Pre-training of Deep Bidirectional Transformers for Language
               Understanding},
  booktitle = {Proceedings of the 2019 Conference of the North American Chapter of
               the Association for Computational Linguistics: Human Language Technologies,
               {NAACL-HLT} 2019, Minneapolis, MN, USA, June 2-7, 2019, Volume 1 (Long
               and Short Papers)},
  pages     = {4171--4186},
  publisher = {Association for Computational Linguistics},
  year      = {2019},
  url       = {https://doi.org/10.18653/v1/n19-1423},
  doi       = {10.18653/v1/n19-1423},
  timestamp = {Fri, 06 Aug 2021 00:41:31 +0200},
  biburl    = {https://dblp.org/rec/conf/naacl/DevlinCLT19.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}
"""


def parse_titles(bibtex):
    titles = []
    lines = bibtex.split("\n")
    print(lines)
    for line in lines:
        if line.strip().startswith("title"):
            title = "".join(line.split("=")[1:])
            title_clean = title.replace("{", "").replace("}", "").replace(",\n", "").strip()
            titles.append(title_clean)
    return titles


def cleaner(bibtex, keep_keys, file_obj):
    dblp_citations = []
    errors = []

    if file_obj:
        titles = parse_bibtext_file_titles(file_obj.name)

    elif bibtex:
        titles = parse_titles(bibtex)

    for publication in titles:
        if site_url := get_url(publication):
            if dblp_citation := get_dblp_bibtext(site_url):
                dblp_citations.append(dblp_citation)
            else:
                errors.append(" - " + publication)
        else:
            errors.append(" - " + publication)

    if dblp_citations:
        filenaem = "cleaned.bib"
        with open(filenaem, "w") as outFile:
            outFile.write("\n".join(dblp_citations))

    return filenaem, "\n".join(errors)


iface = gr.Interface(
    fn=cleaner,
    title="Bibtex cleaner",
    description="Clean a bibtex file or string simply by dragging the incomplete or broken bibtex file into the file box or pasting a bibtex string into the string field. The titles are extrected, searched at the DBLP and a clean bibtexfile is created.",
    inputs=[
        gr.inputs.Textbox(label="Paste a string here:", lines=1),
        gr.inputs.Checkbox(label="Keep original keys:"),
        gr.inputs.File(label="Drag a Bibtex file here:", file_count="single", type="file", optional=True),
    ],
    outputs=[
        gr.outputs.File(label="Cleaned bibtext file:"),
        gr.outputs.Textbox(type="auto", label="Errors:"),
    ],
)
iface.launch()