Improve the frontend

services/training/app/api/training_operations.py

@@ -9,6 +9,7 @@ import structlog
 from datetime import datetime, timezone
 import uuid
 import shared.redis_utils
+from sqlalchemy.ext.asyncio import AsyncSession
 
 from shared.routing import RouteBuilder
 from shared.monitoring.decorators import track_execution_time
@@ -39,6 +40,7 @@ from app.services.training_events import (
     publish_training_failed
 )
 from app.core.config import settings
+from app.core.database import get_db
 
 logger = structlog.get_logger()
 route_builder = RouteBuilder('training')
@@ -86,7 +88,8 @@ async def start_training_job(
     request_obj: Request = None,
     current_user: Dict[str, Any] = Depends(get_current_user_dep),
     enhanced_training_service: EnhancedTrainingService = Depends(get_enhanced_training_service),
-    rate_limiter = Depends(get_rate_limiter)
+    rate_limiter = Depends(get_rate_limiter),
+    db: AsyncSession = Depends(get_db)
 ):
     """
     Start a new training job for all tenant products (Admin+ only, quota enforced).
@@ -169,9 +172,7 @@ async def start_training_job(
         # We don't know exact product count yet, so use historical average or estimate
         try:
             # Try to get historical average for this tenant
-            from app.core.database import get_db
-            db = next(get_db())
-            historical_avg = get_historical_average_estimate(db, tenant_id)
+            historical_avg = await get_historical_average_estimate(db, tenant_id)
 
             # If no historical data, estimate based on typical product count (10-20 products)
             estimated_products = 15  # Conservative estimate