File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 2828from networksecurity .stats .api import router as stats_router
2929from networksecurity .firewall .api import firewall_router
3030from networksecurity .models .api import model_router
31+ from networksecurity .protection .api import protection_router
3132
3233# --- 可视化库导入 ---
3334import matplotlib
@@ -130,6 +131,7 @@ async def lifespan(app: FastAPI):
130131app .include_router (stats_router )
131132app .include_router (firewall_router )
132133app .include_router (model_router )
134+ app .include_router (protection_router )
133135
134136
135137# --- WebSocket 端点 ---
@@ -244,6 +246,12 @@ async def serve_train(request: Request):
244246 return templates .TemplateResponse ("training.html" , {"request" : request , "page" : "training" })
245247
246248
249+ @app .get ("/protection" , tags = ["Frontend" ], response_class = HTMLResponse )
250+ async def serve_protection (request : Request ):
251+ """一键保护页面"""
252+ return templates .TemplateResponse ("protection.html" , {"request" : request , "page" : "protection" })
253+
254+
247255@app .get ("/model-select" , tags = ["Frontend" ], response_class = HTMLResponse )
248256async def serve_model_select (request : Request ):
249257 """模型选择页面"""
You can’t perform that action at this time.
0 commit comments