File size: 1,287 Bytes
356ff01 |
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 |
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
import uvicorn
from contextlib import asynccontextmanager
from api.routes import scan, health
from api import dependencies
@asynccontextmanager
async def lifespan(app: FastAPI):
"""Load models on startup, cleanup on shutdown"""
await dependencies.initialize_models()
yield
dependencies.cleanup_models()
app = FastAPI(
title="XSS Detection API",
description="CodeBERT-based XSS vulnerability detection for PHP and JavaScript",
version="1.0.0",
lifespan=lifespan
)
# CORS configuration
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # In production, replace with your frontend URL
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# Include routers
app.include_router(scan.router, prefix="/api/v1", tags=["scan"])
app.include_router(health.router, prefix="/api/v1", tags=["health"])
@app.get("/")
async def root():
return {
"service": "XSS Detection API",
"version": "1.0.0",
"status": "running",
"docs": "/docs"
}
if __name__ == "__main__":
uvicorn.run(
"api.main:app",
host="0.0.0.0",
port=8080,
reload=True,
log_level="info"
)
|