src/time_machine/domain/models.py

from __future__ import annotations

from datetime import datetime, timezone
from typing import Any, Literal
from uuid import uuid4

from pydantic import BaseModel, ConfigDict, Field, computed_field


def utc_now() -> datetime:
    return datetime.now(timezone.utc)


class StrictModel(BaseModel):
    model_config = ConfigDict(extra="forbid", validate_assignment=True)


class VisualMotif(StrictModel):
    key: str
    label: str
    palette: list[str] = Field(default_factory=list)
    motion: str | None = None


class Destination(StrictModel):
    year: str
    place: str
    mode: Literal["past", "future", "strange"]
    atmosphere: str
    visual_preset_key: str
    visual_motifs: list[VisualMotif] = Field(default_factory=list)
    temporal_oddity: str | None = None


class Persona(StrictModel):
    name: str
    local_identifier: str | None = None
    role: str
    immediate_situation: str
    daily_concern: str
    secret_or_desire: str
    worldview_constraints: list[str] = Field(default_factory=list)
    voice_theory: str
    speaking_style: str
    safety_notes: list[str] = Field(default_factory=list)


class VoiceProfile(StrictModel):
    voice_id: str
    description: str
    pace: Literal["slow", "medium", "fast"] = "medium"
    emotion: str = "curious"
    accent_hint: str | None = None


class ConversationTurn(StrictModel):
    role: Literal["user", "character", "machine"]
    text: str
    audio_path: str | None = None
    metadata: dict[str, Any] = Field(default_factory=dict)
    created_at: datetime = Field(default_factory=utc_now)


class ConversationHistory(StrictModel):
    turns: list[ConversationTurn] = Field(default_factory=list)

    def append(self, turn: ConversationTurn) -> None:
        self.turns.append(turn)

    def character_turns(self) -> list[ConversationTurn]:
        return [turn for turn in self.turns if turn.role == "character"]


class ConversationResponse(StrictModel):
    text: str
    prosody_hint: str | None = None
    ui_signal_event: str | None = None


class Transcript(StrictModel):
    text: str
    confidence: float | None = Field(default=None, ge=0, le=1)
    language: str | None = None
    is_final: bool = True


class AudioResult(StrictModel):
    path: str | None = None
    mime_type: str = "audio/wav"
    duration_seconds: float | None = Field(default=None, ge=0)
    description: str | None = None


class ImmersiveScene(StrictModel):
    prompt: str
    image_b64: str | None = None
    mime_type: str = "image/png"
    ambient_key: str = "temporal_storm"
    visual_style: str = "surreal cinematic time portal"
    description: str | None = None


class CharacterPortrait(StrictModel):
    prompt: str
    image_b64: str | None = None
    mime_type: str = "image/png"
    description: str | None = None


class Narration(StrictModel):
    text: str
    audio: AudioResult | None = None
    voice_profile: VoiceProfile | None = None


class VisualArtifact(StrictModel):
    title: str
    kind: str
    description: str
    prompt: str
    image_b64: str | None = None
    mime_type: str = "image/png"


class Souvenir(StrictModel):
    destination: str
    contact: str
    quote: str
    artifact: str
    stamp_name: str
    summary: str


class EncounterSession(StrictModel):
    session_id: str
    encounter_id: str = Field(default_factory=lambda: str(uuid4()))
    destination: Destination | None = None
    persona: Persona | None = None
    voice_profile: VoiceProfile | None = None
    scene: ImmersiveScene | None = None
    portrait: CharacterPortrait | None = None
    narration: Narration | None = None
    artifact: VisualArtifact | None = None
    history: ConversationHistory = Field(default_factory=ConversationHistory)
    souvenir: Souvenir | None = None
    created_at: datetime = Field(default_factory=utc_now)
    updated_at: datetime = Field(default_factory=utc_now)

    def touch(self) -> None:
        self.updated_at = utc_now()


class SavedEncounter(StrictModel):
    encounter_id: str
    session_id: str
    path: str
    saved_at: datetime = Field(default_factory=utc_now)


class ModelSpec(StrictModel):
    role: str
    provider: str
    model_id: str
    parameters_billion: float = Field(ge=0)
    enabled: bool = True
    runtime: str
    license: str | None = None
    notes: str | None = None


class ModelBudget(StrictModel):
    models: list[ModelSpec] = Field(default_factory=list)
    parameter_limit_billion: float = 32.0

    @computed_field
    @property
    def enabled_parameters_billion(self) -> float:
        return round(
            sum(model.parameters_billion for model in self.models if model.enabled),
            3,
        )

    @computed_field
    @property
    def largest_enabled_parameters_billion(self) -> float:
        enabled = [model.parameters_billion for model in self.models if model.enabled]
        return round(max(enabled, default=0.0), 3)

    @computed_field
    @property
    def is_within_limit(self) -> bool:
        return all(
            model.parameters_billion <= self.parameter_limit_billion
            for model in self.models
            if model.enabled
        )