gcontext/api/app/main.py

from contextlib import asynccontextmanager

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from slowapi import _rate_limit_exceeded_handler
from slowapi.errors import RateLimitExceeded

from .db import init_db
from .ratelimit import limiter
from .routes_admin import router as admin_router
from .routes_moderation import router as moderation_router
from .routes_public import router as public_router


@asynccontextmanager
async def lifespan(app: FastAPI):
    init_db()
    yield


app = FastAPI(title="gcontext workflows API", lifespan=lifespan)
app.state.limiter = limiter
app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
app.add_middleware(
    CORSMiddleware,
    allow_origins=[
        "https://gcontext.ai",
        "https://www.gcontext.ai",
        "http://localhost:3000",
        "http://localhost:3001",
    ],
    allow_methods=["*"],
    allow_headers=["*"],
)
app.include_router(public_router)
app.include_router(moderation_router)
app.include_router(admin_router)


@app.get("/health")
def health():
    return {"status": "ok"}