AI story writer enhancements, text to video and voice generation, subscription management, and more.

backend/services/subscription/monitoring_middleware.py

@@ -19,10 +19,18 @@ import re
 
 from models.api_monitoring import APIRequest, APIEndpointStats, SystemHealth, CachePerformance
 from models.subscription_models import APIProvider
-from services.database import get_db
 from .usage_tracking_service import UsageTrackingService
 from .pricing_service import PricingService
 
+
+def _get_db_session():
+    """
+    Get a database session with lazy import to survive hot reloads.
+    Uvicorn's reloader can sometimes clear module-level imports.
+    """
+    from services.database import get_db
+    return next(get_db())
+
 class DatabaseAPIMonitor:
     """Database-backed API monitoring with usage tracking and subscription management."""
     
@@ -145,8 +153,9 @@ async def check_usage_limits_middleware(request: Request, user_id: str, request_
     except Exception:
         pass
     
+    db = None
     try:
-        db = next(get_db())
+        db = _get_db_session()
         api_monitor = DatabaseAPIMonitor()
         
         # Detect if this is an API call that should be rate limited
@@ -203,14 +212,15 @@ async def check_usage_limits_middleware(request: Request, user_id: str, request_
         # Don't block requests if usage checking fails
         return None
     finally:
-        db.close()
+        if db is not None:
+            db.close()
 
 async def monitoring_middleware(request: Request, call_next):
     """Enhanced FastAPI middleware for monitoring API calls with usage tracking."""
     start_time = time.time()
     
     # Get database session
-    db = next(get_db())
+    db = _get_db_session()
     
     # Extract request details - Enhanced user identification
     user_id = None
@@ -340,8 +350,9 @@ async def monitoring_middleware(request: Request, call_next):
 
 async def get_monitoring_stats(minutes: int = 5) -> Dict[str, Any]:
     """Get current monitoring statistics."""
-    db = next(get_db())
+    db = None
     try:
+        db = _get_db_session()
         # Placeholder to match old API; heavy stats handled elsewhere
         return {
             'timestamp': datetime.utcnow().isoformat(),
@@ -354,12 +365,14 @@ async def get_monitoring_stats(minutes: int = 5) -> Dict[str, Any]:
             'system_health': {'status': 'healthy', 'error_rate': 0.0}
         }
     finally:
-        db.close()
+        if db is not None:
+            db.close()
 
 async def get_lightweight_stats() -> Dict[str, Any]:
     """Get lightweight stats for dashboard header."""
-    db = next(get_db())
+    db = None
     try:
+        db = _get_db_session()
         # Minimal viable placeholder values
         now = datetime.utcnow()
         return {
@@ -371,4 +384,5 @@ async def get_lightweight_stats() -> Dict[str, Any]:
             'timestamp': now.isoformat()
         }
     finally:
-        db.close()
+        if db is not None:
+            db.close()

backend/services/subscription/preflight_validator.py

@@ -420,3 +420,54 @@ def validate_video_generation_operations(
                 'message': f"Failed to validate video generation: {str(e)}"
             }
         )
+
+
+def validate_scene_animation_operation(
+    pricing_service: PricingService,
+    user_id: str,
+) -> None:
+    """
+    Validate the per-scene animation workflow before API calls.
+    """
+    try:
+        operations_to_validate = [
+            {
+                'provider': APIProvider.VIDEO,
+                'tokens_requested': 0,
+                'actual_provider_name': 'wavespeed',
+                'operation_type': 'scene_animation',
+            }
+        ]
+
+        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] Scene animation blocked for user {user_id}: {message}")
+            usage_info = error_details.get('usage_info', {}) if error_details else {}
+            provider = usage_info.get('provider', 'video') if usage_info else 'video'
+            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] ✅ Scene animation validated for user {user_id}")
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"[Pre-flight Validator] Error validating scene animation: {e}", exc_info=True)
+        raise HTTPException(
+            status_code=500,
+            detail={
+                'error': f"Failed to validate scene animation: {str(e)}",
+                'message': f"Failed to validate scene animation: {str(e)}",
+            },
+        )

backend/services/subscription/pricing_service.py

@@ -307,6 +307,41 @@ class PricingService:
                 "model_name": "default",
                 "cost_per_request": 0.10,  # $0.10 per video generation (estimated)
                 "description": "AI Video Generation default pricing"
+            },
+            {
+                "provider": APIProvider.VIDEO,
+                "model_name": "kling-v2.5-turbo-std-5s",
+                "cost_per_request": 0.21,
+                "description": "WaveSpeed Kling v2.5 Turbo Std Image-to-Video (5 seconds)"
+            },
+            {
+                "provider": APIProvider.VIDEO,
+                "model_name": "kling-v2.5-turbo-std-10s",
+                "cost_per_request": 0.42,
+                "description": "WaveSpeed Kling v2.5 Turbo Std Image-to-Video (10 seconds)"
+            },
+            {
+                "provider": APIProvider.VIDEO,
+                "model_name": "wavespeed-ai/infinitetalk",
+                "cost_per_request": 0.30,
+                "description": "WaveSpeed InfiniteTalk (image + audio to talking avatar video)"
+            },
+            # Audio Generation Pricing (Minimax Speech 02 HD via WaveSpeed)
+            {
+                "provider": APIProvider.AUDIO,
+                "model_name": "minimax/speech-02-hd",
+                "cost_per_input_token": 0.00005,  # $0.05 per 1,000 characters (every character is 1 token)
+                "cost_per_output_token": 0.0,  # No output tokens for audio
+                "cost_per_request": 0.0,  # Pricing is per character, not per request
+                "description": "AI Audio Generation (Text-to-Speech) - Minimax Speech 02 HD via WaveSpeed"
+            },
+            {
+                "provider": APIProvider.AUDIO,
+                "model_name": "default",
+                "cost_per_input_token": 0.00005,  # $0.05 per 1,000 characters default
+                "cost_per_output_token": 0.0,
+                "cost_per_request": 0.0,
+                "description": "AI Audio Generation default pricing"
             }
         ]
         
@@ -358,6 +393,7 @@ class PricingService:
                 "exa_calls_limit": 100,
                 "video_calls_limit": 0,  # No video generation for free tier
                 "image_edit_calls_limit": 10,  # 10 AI image editing calls/month
+                "audio_calls_limit": 20,  # 20 AI audio generation calls/month
                 "gemini_tokens_limit": 100000,
                 "monthly_cost_limit": 0.0,
                 "features": ["basic_content_generation", "limited_research"],
@@ -381,6 +417,7 @@ class PricingService:
                 "exa_calls_limit": 500,
                 "video_calls_limit": 20,  # 20 videos/month for basic plan
                 "image_edit_calls_limit": 30,  # 30 AI image editing calls/month
+                "audio_calls_limit": 50,  # 50 AI audio generation calls/month
                 "gemini_tokens_limit": 20000,  # Increased from 5000 for better stability
                 "openai_tokens_limit": 20000,  # Increased from 5000 for better stability
                 "anthropic_tokens_limit": 20000,  # Increased from 5000 for better stability
@@ -406,6 +443,7 @@ class PricingService:
                 "exa_calls_limit": 2000,
                 "video_calls_limit": 50,  # 50 videos/month for pro plan
                 "image_edit_calls_limit": 100,  # 100 AI image editing calls/month
+                "audio_calls_limit": 200,  # 200 AI audio generation calls/month
                 "gemini_tokens_limit": 5000000,
                 "openai_tokens_limit": 2500000,
                 "anthropic_tokens_limit": 1000000,
@@ -431,6 +469,7 @@ class PricingService:
                 "exa_calls_limit": 0,  # Unlimited
                 "video_calls_limit": 0,  # Unlimited for enterprise
                 "image_edit_calls_limit": 0,  # Unlimited image editing for enterprise
+                "audio_calls_limit": 0,  # Unlimited audio generation for enterprise
                 "gemini_tokens_limit": 0,
                 "openai_tokens_limit": 0,
                 "anthropic_tokens_limit": 0,
@@ -651,6 +690,7 @@ class PricingService:
                 'stability_calls': plan.stability_calls_limit,
                 'video_calls': getattr(plan, 'video_calls_limit', 0),  # Support missing column
                 'image_edit_calls': getattr(plan, 'image_edit_calls_limit', 0),  # Support missing column
+                'audio_calls': getattr(plan, 'audio_calls_limit', 0),  # Support missing column
                 # Token limits
                 'gemini_tokens': plan.gemini_tokens_limit,
                 'openai_tokens': plan.openai_tokens_limit,

backend/services/subscription/schema_utils.py

@@ -31,6 +31,7 @@ def ensure_subscription_plan_columns(db: Session) -> None:
             "exa_calls_limit": "INTEGER DEFAULT 0",
             "video_calls_limit": "INTEGER DEFAULT 0",
             "image_edit_calls_limit": "INTEGER DEFAULT 0",
+            "audio_calls_limit": "INTEGER DEFAULT 0",
         }
 
         for col_name, ddl in required_columns.items():
@@ -84,6 +85,8 @@ def ensure_usage_summaries_columns(db: Session) -> None:
             "video_cost": "REAL DEFAULT 0.0",
             "image_edit_calls": "INTEGER DEFAULT 0",
             "image_edit_cost": "REAL DEFAULT 0.0",
+            "audio_calls": "INTEGER DEFAULT 0",
+            "audio_cost": "REAL DEFAULT 0.0",
         }
 
         for col_name, ddl in required_columns.items():