File size: 3,938 Bytes
42d88ae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 | from contextlib import asynccontextmanager
from pathlib import Path
from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse, FileResponse
from fastapi.staticfiles import StaticFiles
from Backend.core.config import settings
from Backend.core.events import event_bus
from Backend.core.logging import setup_logging, get_logger
from Backend.core.security import SecurityHeadersMiddleware, RateLimitMiddleware, RequestValidationMiddleware
from Backend.database.connection import init_db, close_db
from Backend.api.routes import api_router
logger = get_logger(__name__)
STATIC_DIR = Path("static")
@asynccontextmanager
async def lifespan(app: FastAPI):
setup_logging(debug=settings.debug)
logger.info("Starting City Issue Resolution Agent")
await init_db()
logger.info("Database initialized")
await event_bus.start()
logger.info("Event bus started")
from Backend.agents.vision import VisionAgent
try:
VisionAgent.load_model()
logger.info("Vision model loaded")
except Exception as e:
logger.warning(f"Vision model failed to load: {e}. Running in mock mode.")
import asyncio
from Backend.database.connection import get_db_context
from Backend.agents.escalation.agent import EscalationAgent
from Backend.agents.sla.agent import SLAAgent
async def run_periodic_checks():
while True:
try:
logger.info("Running periodic SLA and Escalation checks...")
async with get_db_context() as db:
esc_agent = EscalationAgent(db)
await esc_agent.check_all_pending()
sla_agent = SLAAgent(db)
await sla_agent.check_all_active()
except Exception as e:
logger.error(f"Error in background task: {e}")
await asyncio.sleep(900)
task = asyncio.create_task(run_periodic_checks())
yield
task.cancel()
await event_bus.stop()
await close_db()
logger.info("Shutdown complete")
def create_app() -> FastAPI:
app = FastAPI(
title="City Issue Resolution Agent",
description="Autonomous urban issue detection and resolution platform",
version="1.0.0",
lifespan=lifespan,
root_path="",
)
# CORS must be added first
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=False,
allow_methods=["*"],
allow_headers=["*"],
expose_headers=["*"],
)
app.add_middleware(SecurityHeadersMiddleware)
app.add_middleware(RateLimitMiddleware, requests_per_minute=120, burst_limit=20)
app.add_middleware(RequestValidationMiddleware)
settings.local_temp_dir.mkdir(parents=True, exist_ok=True)
STATIC_DIR.mkdir(parents=True, exist_ok=True)
app.mount("/static", StaticFiles(directory=str(STATIC_DIR)), name="static")
app.include_router(api_router)
@app.get("/")
async def root():
return FileResponse(STATIC_DIR / "flow.html")
@app.get("/dashboard")
async def dashboard():
return FileResponse(STATIC_DIR / "flow.html")
@app.exception_handler(ValueError)
async def value_error_handler(request: Request, exc: ValueError):
return JSONResponse(
status_code=400,
content={"detail": str(exc)}
)
@app.exception_handler(Exception)
async def general_exception_handler(request: Request, exc: Exception):
logger.error(f"Unhandled exception: {exc}", exc_info=True)
return JSONResponse(
status_code=500,
content={"detail": "Internal server error"}
)
return app
|