Added YouTube Creator scene building flow documentation

backend/services/llm_providers/main_image_generation.py

@@ -9,6 +9,7 @@ from .image_generation import (
     HuggingFaceImageProvider,
     GeminiImageProvider,
     StabilityImageProvider,
+    WaveSpeedImageProvider,
 )
 from utils.logger_utils import get_service_logger
 
@@ -26,6 +27,8 @@ def _select_provider(explicit: Optional[str]) -> str:
         return "huggingface"
     if os.getenv("STABILITY_API_KEY"):
         return "stability"
+    if os.getenv("WAVESPEED_API_KEY"):
+        return "wavespeed"
     # Fallback to huggingface to enable a path if configured
     return "huggingface"
 
@@ -37,6 +40,8 @@ def _get_provider(provider_name: str):
         return GeminiImageProvider()
     if provider_name == "stability":
         return StabilityImageProvider()
+    if provider_name == "wavespeed":
+        return WaveSpeedImageProvider()
     raise ValueError(f"Unknown image provider: {provider_name}")
 
 
@@ -56,6 +61,7 @@ def generate_image(prompt: str, options: Optional[Dict[str, Any]] = None, user_i
         from services.subscription.preflight_validator import validate_image_generation_operations
         from fastapi import HTTPException
         
+        logger.info(f"[Image Generation] 🔍 Starting pre-flight validation for user_id={user_id}")
         db = next(get_db())
         try:
             pricing_service = PricingService(db)
@@ -64,14 +70,15 @@ def generate_image(prompt: str, options: Optional[Dict[str, Any]] = None, user_i
                 pricing_service=pricing_service,
                 user_id=user_id
             )
+            logger.info(f"[Image Generation] ✅ Pre-flight validation passed for user_id={user_id} - proceeding with image generation")
         except HTTPException as http_ex:
             # Re-raise immediately - don't proceed with API call
-            logger.error(f"[Image Generation] ❌ Pre-flight validation failed - blocking API call")
+            logger.error(f"[Image Generation] ❌ Pre-flight validation failed for user_id={user_id} - blocking API call: {http_ex.detail}")
             raise
         finally:
             db.close()
-    
-    logger.info(f"[Image Generation] ✅ Pre-flight validation passed - proceeding with image generation")
+    else:
+        logger.warning(f"[Image Generation] ⚠️ No user_id provided - skipping pre-flight validation (this should not happen in production)")
     opts = options or {}
     provider_name = _select_provider(opts.get("provider"))
 
@@ -96,6 +103,10 @@ def generate_image(prompt: str, options: Optional[Dict[str, Any]] = None, user_i
     if provider_name == "huggingface" and not image_options.model:
         # Provide a sensible default HF model if none specified
         image_options.model = "black-forest-labs/FLUX.1-Krea-dev"
+    
+    if provider_name == "wavespeed" and not image_options.model:
+        # Provide a sensible default WaveSpeed model if none specified
+        image_options.model = "ideogram-v3-turbo"
 
     logger.info("Generating image via provider=%s model=%s", provider_name, image_options.model)
     provider = _get_provider(provider_name)