Create main.py

main.py

@@ -0,0 +1,28 @@
+from fastapi import FastAPI, HTTPException
+import urllib.request
+from urllib.error import HTTPError
+
+app = FastAPI()
+
+BASE_URL = 'https://doi.org/'
+
+@app.get("/get_bibtex/")
+async def get_bibtex(doi: str):
+    url = BASE_URL + doi
+    req = urllib.request.Request(url)
+    req.add_header('Accept', 'application/x-bibtex')
+
+    try:
+        with urllib.request.urlopen(req) as f:
+            bibtex = f.read().decode()
+        return {"bibtex": bibtex}
+    except HTTPError as e:
+        if e.code == 404:
+            raise HTTPException(status_code=404, detail="DOI not found")
+        else:
+            raise HTTPException(status_code=503, detail="Service unavailable")
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run("main:app", host="0.0.0.0", port=80)