inspirations

models/Asset.py

@@ -8,10 +8,12 @@ from pydantic import BaseModel, computed_field, Field, model_validator
 class AssetContentType(str, Enum):
     IMAGE = 'image'
     PROMPT = 'prompt'
+    VIDEO = 'video'
 
 class AssetType(str, Enum):
     UPLOADED = 'uploaded'
     GENERATED = 'generated'
+    INSPIRATION = 'inspiration'
 
 
 class Asset(BaseModel):

models/Idea.py

@@ -7,6 +7,7 @@ class Idea(BaseModel):
     name: str = "New Idea"
     description: Optional[str] = None
     project_id: Optional[str] = None
+    inspiration_id: Optional[str] = None  # Link to Inspiration
     created_by: str  # User ID
     is_deleted: bool = False
     created_at: datetime = Field(default_factory=datetime.now)

models/Inspiration.py

@@ -0,0 +1,16 @@
+from datetime import datetime, UTC
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class Inspiration(BaseModel):
+    id: str | None = None
+    source_url: str
+    caption: str | None = None
+    asset_id: str
+    is_completed: bool = False
+    created_by: str
+    project_id: str | None = None
+    created_at: datetime = Field(default_factory=lambda: datetime.now(UTC))
+    updated_at: datetime = Field(default_factory=lambda: datetime.now(UTC))