| import os |
| import tempfile |
| from fastapi import FastAPI, File, UploadFile, HTTPException |
| from fastapi.responses import JSONResponse |
|
|
| app = FastAPI(title="Scholarform Docling Service") |
|
|
| try: |
| from docling.document_converter import DocumentConverter |
| _converter = DocumentConverter() |
| DOCLING_OK = True |
| except Exception as e: |
| _converter = None |
| DOCLING_OK = False |
| DOCLING_ERR = str(e) |
|
|
| @app.get("/") |
| def root(): |
| return {"ok": True, "service": "docling", "docling_available": DOCLING_OK} |
|
|
| @app.get("/health") |
| def health(): |
| payload = {"ok": True, "service": "docling", "docling_available": DOCLING_OK} |
| if not DOCLING_OK: |
| payload["error"] = DOCLING_ERR |
| return payload |
|
|
| @app.post("/analyze") |
| async def analyze(file: UploadFile = File(...)): |
| if not DOCLING_OK: |
| raise HTTPException(status_code=503, detail=f"Docling unavailable: {DOCLING_ERR}") |
| if not file.filename.lower().endswith(".pdf"): |
| raise HTTPException(status_code=400, detail="Only PDF is supported") |
|
|
| with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp: |
| tmp.write(await file.read()) |
| path = tmp.name |
|
|
| try: |
| doc = _converter.convert(path).document |
| elements = [] |
|
|
| for item in getattr(doc, "texts", []): |
| text = getattr(item, "text", "") or "" |
| label = getattr(item, "label", "text") |
| bbox = None |
| prov = getattr(item, "prov", None) |
| if prov and len(prov) > 0 and getattr(prov[0], "bbox", None): |
| b = prov[0].bbox |
| bbox = {"x0": b.l, "y0": b.t, "x1": b.r, "y1": b.b, "page": getattr(prov[0], "page_no", 0)} |
| elements.append({"type": str(label), "text": text, "bbox": bbox}) |
|
|
| pages = getattr(doc, "num_pages", 0) |
| if callable(pages): |
| pages = pages() |
|
|
| return {"ok": True, "pages": pages, "elements": elements[:1000]} |
| except Exception as e: |
| raise HTTPException(status_code=500, detail=f"Docling analyze failed: {e}") |
| finally: |
| try: |
| os.remove(path) |
| except Exception: |
| pass |
|
|
