Add fetched bibtex to output

app.py

@@ -3,7 +3,8 @@
 
 import streamlit as st
 
-from refcheck.doi import get_dois
+from refcheck.doi import get_dois, doi_to_url
+from refcheck.bibtex import fetch_bibtex
 
 DOI_REGEX = "10.\\d{4,9}/[-._;()/:a-z0-9A-Z]+"
 FORMAT_TO_EXT = {
@@ -28,7 +29,10 @@
     with open(path, "wb") as f:
         f.write(uploaded_file.getvalue())
     results = get_dois(path, format)
+    urls = [doi_to_url(doi) for doi in results]
+    bibtexes = [fetch_bibtex(url) for url in urls]
 
     st.write("Here are the dois in your document. Please check if they are correct!")
-    for result in results:
-        st.write(f"[{result}](https://doi.org/{result})")
+    for doi, url, bibtex in zip(results, urls, bibtexes):
+        st.write(f"[**{doi}**]({url}):", bibtex)
+        st.write("")

src/refcheck/bibtex.py

@@ -0,0 +1,7 @@
+import requests
+
+
+def fetch_bibtex(doi_url: str) -> str | None:
+    headers = {"accept": "application/x-bibtex"}
+    r = requests.get(doi_url, headers=headers)
+    return r.text if r is not None else None

src/refcheck/doi.py

@@ -10,3 +10,7 @@ def get_dois(path: Path, format) -> list[str]:
     txt = pypandoc.convert_file(path, "plain", format=format)
     matches = re.findall(DOI_REGEX, txt)
     return [r if not r.endswith(".") else r[:-1] for r in matches]
+
+
+def doi_to_url(doi: str) -> str:
+    return "http://dx.doi.org/" + doi