models/studio_project.py

from pydantic import BaseModel
from typing import Dict, Optional
from datetime import datetime


class StudioProjectCreate(BaseModel):
    user_id: str
    prompt: str
    files: Dict[str, str] = {}
    status: str = "building"
    created_at: datetime = datetime.utcnow()
    updated_at: datetime = datetime.utcnow()


class StudioProjectResponse(BaseModel):
    id: str
    user_id: str
    prompt: str
    files: Dict[str, str]
    status: str
    created_at: str
    updated_at: str