| from fastapi import FastAPI, HTTPException |
| import urllib.request |
| from urllib.error import HTTPError |
| from habanero import Crossref |
| from fastapi.responses import HTMLResponse |
|
|
|
|
| app = FastAPI() |
|
|
| BASE_URL = 'https://doi.org/' |
|
|
| @app.get("/", response_class=HTMLResponse) |
| def read_root(): |
| html_content = """ |
| <html> |
| <head> |
| <title>Citation Helper</title> |
| </head> |
| <body> |
| <h1>Hello!</h1> |
| <p>As Citation Helper, I specialize in generating citations for academic papers. You can provide me with the DOI (Digital Object Identifier) or the title of an academic paper, and I will create a citation for it.</p> |
| <p>Here's how you can use my service:</p> |
| <ul> |
| <li><b>Provide a DOI or Title:</b> Share the DOI or the title of the paper you need a citation for.</li> |
| <li><b>Choose a Citation Style:</b> After I generate the citation in APA format, you can request it in another style like Chicago, MLA, or Harvard if needed.</li> |
| <li><b>Receive Your Citation:</b> I'll provide you with the correctly formatted citation.</li> |
| </ul> |
| <p>Feel free to start by giving me a DOI or a paper title!</p> |
| </body> |
| </html> |
| """ |
| return HTMLResponse(content=html_content) |
|
|
| @app.get("/title2ref/") |
| async def title2doi(title: str): |
| cr = Crossref() |
|
|
| result = cr.works(query=title) |
| |
| |
| if result['message']['items']: |
| for item in result['message']['items']: |
| if 'title' in item and item['title'][0] == title: |
| doi = item['DOI'] |
| 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") |
|
|
| |
| raise HTTPException(status_code=404, detail="Title not found, please enter DOI") |
| |
| @app.get("/doi2ref/") |
| async def doi2ref(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=7860) |
