from pydantic import BaseModel
from typing import Optional, List, Dict
from app.models.enums import NoteType, NoteStatus

class AudioMetadata(BaseModel):
    duration: Optional[float] = None
    chunks: Optional[list] = None
    sample_rate: Optional[int] = None
    url: Optional[str] = None
    asr_model: Optional[str] = None

class NoteMetadata(BaseModel):
    audio: Optional[AudioMetadata] = None
    client: Optional[Dict] = None

class Note(BaseModel):
    note_id: str
    type: NoteType
    
    title: Optional[str] = None
    raw_text: Optional[str] = None
    normalized_text: Optional[str] = None
    keywords: Optional[List[str]] = None
    summary: Optional[str] = None
    mindmap: Optional[Dict] = None

    folder_id: Optional[str] = None
    metadata: Optional[NoteMetadata] = None

    status: NoteStatus
    created_at: int
    updated_at: int