Improve the traffic fetching system

services/data/app/api/traffic.py

@@ -110,4 +110,60 @@ async def get_historical_traffic(
         raise
     except Exception as e:
         logger.error("Unexpected error in historical traffic API", error=str(e))
+        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
+
+@router.post("/tenants/{tenant_id}/traffic/stored")
+async def get_stored_traffic_for_training(
+    request: HistoricalTrafficRequest,
+    db: AsyncSession = Depends(get_db),
+    tenant_id: UUID = Path(..., description="Tenant ID"),
+    current_user: Dict[str, Any] = Depends(get_current_user_dep),
+):
+    """Get stored traffic data specifically for training/re-training purposes"""
+    try:
+        # Validate date range
+        if request.end_date <= request.start_date:
+            raise HTTPException(status_code=400, detail="End date must be after start date")
+        
+        # Allow longer date ranges for training (up to 3 years)
+        if (request.end_date - request.start_date).days > 1095:
+            raise HTTPException(status_code=400, detail="Date range cannot exceed 3 years for training data")
+        
+        logger.info("Retrieving stored traffic data for training",
+                   tenant_id=str(tenant_id),
+                   location=f"{request.latitude},{request.longitude}",
+                   date_range=f"{request.start_date} to {request.end_date}")
+        
+        # Use the dedicated method for training data retrieval
+        stored_data = await traffic_service.get_stored_traffic_for_training(
+            request.latitude, request.longitude, request.start_date, request.end_date, db
+        )
+        
+        # Log retrieval for audit purposes
+        logger.info("Stored traffic data retrieved for training",
+                   records_count=len(stored_data),
+                   tenant_id=str(tenant_id),
+                   purpose="model_training")
+        
+        # Publish event for monitoring
+        try:
+            await publish_traffic_updated({
+                "type": "stored_data_retrieved_for_training",
+                "latitude": request.latitude,
+                "longitude": request.longitude,
+                "start_date": request.start_date.isoformat(),
+                "end_date": request.end_date.isoformat(),
+                "records_count": len(stored_data),
+                "tenant_id": str(tenant_id),
+                "timestamp": datetime.utcnow().isoformat()
+            })
+        except Exception as pub_error:
+            logger.warning("Failed to publish stored traffic retrieval event", error=str(pub_error))
+        
+        return stored_data
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error("Unexpected error in stored traffic retrieval API", error=str(e))
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")