AI/crawler_platform/app/main.py

from __future__ import annotations

import os
from pathlib import Path

from fastapi import FastAPI, Request
from fastapi.responses import FileResponse, JSONResponse

from crawler_platform.app.api.routes import register_routes
from crawler_platform.app.core.database.session import init_db


DATABASE_URL = os.getenv("CRAWLER_DATABASE_URL", "sqlite:///crawler_platform.db")
STATIC_DIR = Path(__file__).parent / "web" / "static"

app = FastAPI(title="Ontology Crawler Platform", version="0.1.0")
init_db(DATABASE_URL)
register_routes(app, DATABASE_URL)


@app.exception_handler(KeyError)
def _handle_key_error(_request: Request, exc: KeyError) -> JSONResponse:
    return JSONResponse(status_code=404, content={"detail": str(exc.args[0]) if exc.args else "not found"})

@app.get("/static/{asset_path:path}", include_in_schema=False)
def static_or_spa(asset_path: str):
    static_root = STATIC_DIR.resolve()
    target = (static_root / asset_path).resolve()
    try:
        target.relative_to(static_root)
    except ValueError:
        return JSONResponse(status_code=404, content={"detail": "not found"})

    if target.is_file():
        return FileResponse(target)
    return FileResponse(STATIC_DIR / "index.html")


@app.get("/", include_in_schema=False)
def admin_ui():
    return FileResponse(STATIC_DIR / "index.html")