main.py

"""OCR API Server."""

import datetime
import logging

from fastapi import FastAPI, status
from fastapi.responses import JSONResponse, RedirectResponse

from .routers import docling, kreuzberg, marker, paddleocr

_today = datetime.datetime.now(datetime.UTC).strftime("%Y_%m_%d")  # type: ignore[attr-defined] # mypy complains that 'Module has no attribute "UTC"'
logging.basicConfig(
    filename=f"pyonb-{_today}.log",
    format="%(asctime)s %(message)s",
    filemode="a",
)

logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

app = FastAPI(swagger_ui_parameters={"tryItOutEnabled": True})

app.include_router(marker.router)
app.include_router(paddleocr.router)
app.include_router(docling.router)
app.include_router(kreuzberg.router)


@app.get("/", include_in_schema=False)
async def root() -> RedirectResponse:
    """Redirect localhost:<PORT> to Swagger at localhost:<PORT>/docs."""
    return RedirectResponse(url="/docs")


@app.get("/")
async def health_check() -> JSONResponse:
    """
    Health check endpoint to verify API is accessible.

    Returns 200 OK status if API is running properly.
    """
    return JSONResponse(
        status_code=status.HTTP_200_OK,
        content={"service": "pyonb-ocr-api", "status": "healthy"},
    )