api/main.py

import logging
import os
from contextlib import asynccontextmanager

import uvicorn
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from api.auth import router as auth_router
from api.enhance import router as enhance_router
from api.generate import router as generate_router, shutdown_executor
from api.integrations import router as integrations_router
from api.intake import router as intake_router
from api.oncall import router as oncall_router
from api.projects import router as projects_router
from api.admin import router as admin_router
from api.uxmaster import router as uxmaster_router

logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(name)s: %(message)s")
logger = logging.getLogger(__name__)


@asynccontextmanager
async def lifespan(app: FastAPI):
    try:
        from db.database import create_tables
        create_tables()
        logger.info("DB tables ensured")
    except Exception as exc:  # noqa: BLE001  ponytail: DB hiccup must not stop boot
        logger.error("create_tables failed (non-fatal): %s", exc)
    yield
    shutdown_executor()


app = FastAPI(title="CodyBuddy Studio API", version="0.1.0", lifespan=lifespan)

# ponytail: strip each origin so "a, b" (stray spaces) doesn't break the CORS match.
_origins = [o.strip() for o in os.environ.get("ALLOWED_ORIGINS", "http://localhost:3000").split(",") if o.strip()]
app.add_middleware(
    CORSMiddleware,
    allow_origins=_origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(auth_router, prefix="/api")
app.include_router(generate_router, prefix="/api")
app.include_router(enhance_router, prefix="/api")
app.include_router(uxmaster_router, prefix="/api")
app.include_router(oncall_router, prefix="/api")
app.include_router(intake_router, prefix="/api")
app.include_router(projects_router, prefix="/api")
app.include_router(admin_router, prefix="/api")
app.include_router(integrations_router, prefix="/api")


@app.get("/api/health", tags=["health"])
@app.get("/healthz", tags=["health"])
@app.get("/ping", tags=["health"])
@app.get("/", tags=["health"])
def health() -> dict:
    # Lightweight, DB-free — safe for an uptime pinger to hit frequently to keep the
    # HF Space warm (free Spaces sleep after ~48h idle).
    return {"status": "ok", "service": "AppSmith API"}


if __name__ == "__main__":
    uvicorn.run("api.main:app", host="0.0.0.0", port=8000, reload=False)