models.py

from dataclasses import dataclass, field
from datetime import datetime
from typing import Optional, Dict, List


@dataclass
class NoteSettings:
    summary_length: str = "medium"
    output_format: str = "bullets"  # bullets, outline, flashcards
    tone: str = "simple"  # simple, academic
    quiz_enabled: bool = False
    quiz_difficulty: str = "medium"  # easy, medium, hard


@dataclass
class Quiz:
    multiple_choice: List[Dict] = field(default_factory=list)
    short_answer: List[Dict] = field(default_factory=list)
    answer_key: Dict = field(default_factory=dict)


@dataclass
class NoteSession:
    id: str
    timestamp: datetime
    original_text: str
    summary: Optional[str] = None
    key_terms: Optional[List[Dict]] = None
    quiz: Optional[Quiz] = None
    settings: NoteSettings = field(default_factory=NoteSettings)
    
    def to_dict(self) -> Dict:
        return {
            "id": self.id,
            "timestamp": self.timestamp.isoformat(),
            "original_text": self.original_text[:100] + "..." if len(self.original_text) > 100 else self.original_text,
            "summary": self.summary,
            "key_terms": self.key_terms,
            "settings": {
                "summary_length": self.settings.summary_length,
                "output_format": self.settings.output_format,
                "tone": self.settings.tone,
                "quiz_enabled": self.settings.quiz_enabled
            }
        }