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/models/content_asset_models.py

@@ -0,0 +1,145 @@
+"""
+Content Asset Models
+Unified database models for tracking all AI-generated content assets across all modules.
+"""
+
+from sqlalchemy import Column, Integer, String, DateTime, Float, Boolean, JSON, Text, ForeignKey, Enum, Index, func
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship
+from datetime import datetime
+import enum
+
+# Use the same Base as subscription models for consistency
+from models.subscription_models import Base
+
+
+class AssetType(enum.Enum):
+    """Types of content assets."""
+    TEXT = "text"
+    IMAGE = "image"
+    VIDEO = "video"
+    AUDIO = "audio"
+
+
+class AssetSource(enum.Enum):
+    """Source module/tool that generated the asset - covers ALL ALwrity tools."""
+    # Image Studio modules
+    IMAGE_STUDIO_CREATE = "image_studio_create"
+    IMAGE_STUDIO_EDIT = "image_studio_edit"
+    IMAGE_STUDIO_UPSCALE = "image_studio_upscale"
+    IMAGE_STUDIO_TRANSFORM = "image_studio_transform"
+    IMAGE_STUDIO_CONTROL = "image_studio_control"
+    IMAGE_STUDIO_SOCIAL = "image_studio_social"
+    IMAGE_STUDIO_BATCH = "image_studio_batch"
+    
+    # Content Writers
+    STORY_WRITER = "story_writer"
+    BLOG_WRITER = "blog_writer"
+    LINKEDIN_WRITER = "linkedin_writer"
+    FACEBOOK_WRITER = "facebook_writer"
+    
+    # Content Planning
+    CONTENT_PLANNING = "content_planning"
+    CONTENT_STRATEGY = "content_strategy"
+    
+    # SEO Tools
+    SEO_DASHBOARD = "seo_dashboard"
+    SEO_TOOLS = "seo_tools"
+    
+    # Research
+    RESEARCH = "research"
+    
+    # Scheduler
+    SCHEDULER = "scheduler"
+    
+    # Main Generation (legacy/fallback)
+    MAIN_TEXT_GENERATION = "main_text_generation"
+    MAIN_IMAGE_GENERATION = "main_image_generation"
+    MAIN_VIDEO_GENERATION = "main_video_generation"
+    MAIN_AUDIO_GENERATION = "main_audio_generation"
+
+
+class ContentAsset(Base):
+    """
+    Unified model for tracking all AI-generated content assets.
+    Similar to subscription tracking, this provides a centralized way to manage all content.
+    """
+    
+    __tablename__ = "content_assets"
+    
+    # Primary fields
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(255), nullable=False, index=True)  # Clerk user ID
+    
+    # Asset identification
+    asset_type = Column(Enum(AssetType), nullable=False, index=True)
+    source_module = Column(Enum(AssetSource), nullable=False, index=True)
+    
+    # File information
+    filename = Column(String(500), nullable=False)
+    file_path = Column(String(1000), nullable=True)  # Server file path
+    file_url = Column(String(1000), nullable=False)  # Public URL
+    file_size = Column(Integer, nullable=True)  # Size in bytes
+    mime_type = Column(String(100), nullable=True)  # MIME type
+    
+    # Asset metadata
+    title = Column(String(500), nullable=True)
+    description = Column(Text, nullable=True)
+    prompt = Column(Text, nullable=True)  # Original prompt used for generation
+    tags = Column(JSON, nullable=True)  # Array of tags for search/filtering
+    metadata = Column(JSON, nullable=True)  # Additional module-specific metadata
+    
+    # Generation details
+    provider = Column(String(100), nullable=True, index=True)  # AI provider used (e.g., "stability", "gemini")
+    model = Column(String(200), nullable=True, index=True)  # Model used (full model path/name)
+    cost = Column(Float, nullable=True, default=0.0)  # Generation cost in USD
+    generation_time = Column(Float, nullable=True)  # Time taken in seconds
+    
+    # Status tracking
+    status = Column(String(50), default='completed', index=True)  # completed, processing, failed, pending
+    error_message = Column(Text, nullable=True)  # Error details if failed
+    
+    # Organization
+    is_favorite = Column(Boolean, default=False, index=True)
+    collection_id = Column(Integer, ForeignKey('asset_collections.id'), nullable=True)
+    
+    # Usage tracking
+    download_count = Column(Integer, default=0)
+    share_count = Column(Integer, default=0)
+    last_accessed = Column(DateTime, nullable=True)
+    
+    # Timestamps
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False, index=True)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    
+    # Relationships
+    collection = relationship("AssetCollection", back_populates="assets", cascade="all, delete-orphan")
+    
+    # Composite indexes for common query patterns
+    __table_args__ = (
+        Index('idx_user_type_source', 'user_id', 'asset_type', 'source_module'),
+        Index('idx_user_favorite_created', 'user_id', 'is_favorite', 'created_at'),
+        Index('idx_user_tags', 'user_id', 'tags'),
+    )
+
+
+class AssetCollection(Base):
+    """
+    Collections/albums for organizing assets.
+    """
+    
+    __tablename__ = "asset_collections"
+    
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(255), nullable=False, index=True)
+    name = Column(String(255), nullable=False)
+    description = Column(Text, nullable=True)
+    is_public = Column(Boolean, default=False)
+    cover_asset_id = Column(Integer, ForeignKey('content_assets.id'), nullable=True)
+    
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    
+    # Relationships
+    assets = relationship("ContentAsset", back_populates="collection")
+