Skip to content

Commit 9aded25

Browse files
feat: 集成一键防护路由和页面入口
1 parent 60c0dd2 commit 9aded25

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

app.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from networksecurity.stats.api import router as stats_router
2929
from networksecurity.firewall.api import firewall_router
3030
from networksecurity.models.api import model_router
31+
from networksecurity.protection.api import protection_router
3132

3233
# --- 可视化库导入 ---
3334
import matplotlib
@@ -130,6 +131,7 @@ async def lifespan(app: FastAPI):
130131
app.include_router(stats_router)
131132
app.include_router(firewall_router)
132133
app.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)
248256
async def serve_model_select(request: Request):
249257
"""模型选择页面"""

0 commit comments

Comments
 (0)