Improve the frontend 5

services/training/app/main.py

@@ -11,7 +11,7 @@ from fastapi import FastAPI, Request
 from sqlalchemy import text
 from app.core.config import settings
 from app.core.database import initialize_training_database, cleanup_training_database, database_manager
-from app.api import training_jobs, training_operations, models, health, monitoring, websocket_operations
+from app.api import training_jobs, training_operations, models, health, monitoring, websocket_operations, audit
 from app.services.training_events import setup_messaging, cleanup_messaging
 from app.websocket.events import setup_websocket_event_consumer, cleanup_websocket_consumers
 from shared.service_base import StandardFastAPIService
@@ -163,6 +163,8 @@ service.setup_custom_middleware()
 service.setup_custom_endpoints()
 
 # Include API routers
+# IMPORTANT: Register audit router FIRST to avoid route matching conflicts
+service.add_router(audit.router)
 service.add_router(training_jobs.router, tags=["training-jobs"])
 service.add_router(training_operations.router, tags=["training-operations"])
 service.add_router(models.router, tags=["models"])