AI Image Studio Progress Review

- Added new router for content assets
- Added new service for content assets
- Added new model for content assets
- Added new utils for content assets
- Added new docs for content assets
- Added new tests for content assets
- Added new examples for content assets
- Added new guides for content assets

backend/services/subscription/preflight_validator.py

@@ -415,6 +415,75 @@ def validate_image_editing_operations(
         )
 
 
+def validate_image_control_operations(
+    pricing_service: PricingService,
+    user_id: str,
+    num_images: int = 1
+) -> None:
+    """
+    Validate image control operations (sketch-to-image, structure control, style transfer) before making API calls.
+    
+    Control operations use Stability AI for image generation with control inputs, so they use
+    the same validation as image generation operations.
+    
+    Args:
+        pricing_service: PricingService instance
+        user_id: User ID for subscription checking
+        num_images: Number of images to generate (for multiple variations)
+        
+    Returns:
+        None - raises HTTPException with 429 status if validation fails
+    """
+    try:
+        # Control operations use Stability AI, same as image generation
+        operations_to_validate = [
+            {
+                'provider': APIProvider.STABILITY,
+                'tokens_requested': 0,
+                'actual_provider_name': 'stability',
+                'operation_type': 'image_generation'  # Control ops use image generation limits
+            }
+            for _ in range(num_images)
+        ]
+        
+        logger.info(f"[Pre-flight Validator] 🚀 Validating {num_images} image control operation(s) for user {user_id}")
+        
+        can_proceed, message, error_details = pricing_service.check_comprehensive_limits(
+            user_id=user_id,
+            operations=operations_to_validate
+        )
+        
+        if not can_proceed:
+            logger.error(f"[Pre-flight Validator] Image control blocked for user {user_id}: {message}")
+            
+            usage_info = error_details.get('usage_info', {}) if error_details else {}
+            provider = usage_info.get('provider', 'stability') if usage_info else 'stability'
+            
+            raise HTTPException(
+                status_code=429,
+                detail={
+                    'error': message,
+                    'message': message,
+                    'provider': provider,
+                    'usage_info': usage_info if usage_info else error_details
+                }
+            )
+        
+        logger.info(f"[Pre-flight Validator] ✅ Image control validated for user {user_id}")
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"[Pre-flight Validator] Error validating image control: {e}", exc_info=True)
+        raise HTTPException(
+            status_code=500,
+            detail={
+                'error': f"Failed to validate image control: {str(e)}",
+                'message': f"Failed to validate image control: {str(e)}"
+            }
+        )
+
+
 def validate_video_generation_operations(
     pricing_service: PricingService,
     user_id: str