AI Image Studio, AI podcast Maker, AI product Marketing

backend/models/product_asset_models.py

@@ -0,0 +1,155 @@
+"""
+Product Asset Models
+Database models for storing product-specific assets (separate from campaign assets).
+These models are for the Product Marketing Suite (product asset creation).
+"""
+
+from sqlalchemy import Column, Integer, String, DateTime, Float, Boolean, JSON, Text, ForeignKey, Index
+from sqlalchemy.orm import relationship
+from datetime import datetime
+import enum
+
+from models.subscription_models import Base
+
+
+class ProductAssetType(enum.Enum):
+    """Product asset type enum."""
+    IMAGE = "image"
+    VIDEO = "video"
+    AUDIO = "audio"
+    ANIMATION = "animation"
+
+
+class ProductImageStyle(enum.Enum):
+    """Product image style enum."""
+    STUDIO = "studio"
+    LIFESTYLE = "lifestyle"
+    OUTDOOR = "outdoor"
+    MINIMALIST = "minimalist"
+    LUXURY = "luxury"
+    TECHNICAL = "technical"
+
+
+class ProductAsset(Base):
+    """
+    Product asset model.
+    Stores product-specific assets (images, videos, audio) generated for product marketing.
+    """
+    
+    __tablename__ = "product_assets"
+    
+    # Primary fields
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    product_id = Column(String(255), nullable=False, index=True)  # User-defined product ID
+    user_id = Column(String(255), nullable=False, index=True)  # Clerk user ID
+    
+    # Product information
+    product_name = Column(String(500), nullable=False)
+    product_description = Column(Text, nullable=True)
+    
+    # Asset details
+    asset_type = Column(String(50), nullable=False, index=True)  # image, video, audio, animation
+    variant = Column(String(100), nullable=True)  # color, size, angle, etc.
+    style = Column(String(50), nullable=True)  # studio, lifestyle, minimalist, etc.
+    environment = Column(String(50), nullable=True)  # studio, lifestyle, outdoor, etc.
+    
+    # Link to ContentAsset (unified asset library)
+    content_asset_id = Column(Integer, ForeignKey('content_assets.id', ondelete='SET NULL'), nullable=True, index=True)
+    
+    # Generation details
+    provider = Column(String(100), nullable=True)
+    model = Column(String(100), nullable=True)
+    cost = Column(Float, default=0.0)
+    generation_time = Column(Float, nullable=True)
+    prompt_used = Column(Text, nullable=True)
+    
+    # E-commerce integration
+    ecommerce_exported = Column(Boolean, default=False)
+    exported_to = Column(JSON, nullable=True)  # Array of platform names
+    
+    # Status
+    status = Column(String(50), default="completed", nullable=False)  # completed, processing, failed
+    
+    # Metadata
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False, index=True)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    
+    # Additional metadata
+    metadata = Column(JSON, nullable=True)  # Additional product-specific metadata
+    
+    # Composite indexes
+    __table_args__ = (
+        Index('idx_user_product', 'user_id', 'product_id'),
+        Index('idx_user_type', 'user_id', 'asset_type'),
+        Index('idx_product_type', 'product_id', 'asset_type'),
+    )
+
+
+class ProductStyleTemplate(Base):
+    """
+    Brand style template for products.
+    Stores reusable brand style configurations for product asset generation.
+    """
+    
+    __tablename__ = "product_style_templates"
+    
+    # Primary fields
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    user_id = Column(String(255), nullable=False, index=True)
+    template_name = Column(String(255), nullable=False)
+    
+    # Style configuration
+    color_palette = Column(JSON, nullable=True)  # Array of brand colors
+    background_style = Column(String(50), nullable=True)  # white, transparent, lifestyle, branded
+    lighting_preset = Column(String(50), nullable=True)  # natural, studio, dramatic, soft
+    preferred_style = Column(String(50), nullable=True)  # photorealistic, minimalist, luxury, technical
+    preferred_environment = Column(String(50), nullable=True)  # studio, lifestyle, outdoor
+    
+    # Brand integration
+    use_brand_colors = Column(Boolean, default=True)
+    use_brand_logo = Column(Boolean, default=False)
+    
+    # Metadata
+    is_default = Column(Boolean, default=False)  # Default template for user
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    
+    # Composite indexes
+    __table_args__ = (
+        Index('idx_user_template', 'user_id', 'template_name'),
+    )
+
+
+class EcommerceExport(Base):
+    """
+    E-commerce platform export tracking.
+    Tracks product asset exports to e-commerce platforms.
+    """
+    
+    __tablename__ = "product_ecommerce_exports"
+    
+    # Primary fields
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    user_id = Column(String(255), nullable=False, index=True)
+    product_id = Column(String(255), nullable=False, index=True)
+    
+    # Platform information
+    platform = Column(String(50), nullable=False)  # shopify, amazon, woocommerce
+    platform_product_id = Column(String(255), nullable=True)  # Product ID on the platform
+    
+    # Export details
+    exported_assets = Column(JSON, nullable=False)  # Array of asset IDs exported
+    export_status = Column(String(50), default="pending", nullable=False)  # pending, completed, failed
+    error_message = Column(Text, nullable=True)
+    
+    # Metadata
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    exported_at = Column(DateTime, nullable=True)
+    
+    # Composite indexes
+    __table_args__ = (
+        Index('idx_user_platform', 'user_id', 'platform'),
+        Index('idx_product_platform', 'product_id', 'platform'),
+    )
+