import uuid
from datetime import datetime

from sqlalchemy import Float, ForeignKey, Integer, JSON, String, func
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.models.base import Base


class Analysis(Base):
    __tablename__ = "analyses"

    id: Mapped[str] = mapped_column(
        String(36), primary_key=True, default=lambda: str(uuid.uuid4())
    )
    project_id: Mapped[str] = mapped_column(ForeignKey("projects.id"), index=True)
    user_id: Mapped[str] = mapped_column(String(36), index=True)
    name: Mapped[str] = mapped_column(String(255))
    model_used: Mapped[str] = mapped_column(String(50), default="deepgaze_iie")
    status: Mapped[str] = mapped_column(String(20), default="pending")

    original_filename: Mapped[str] = mapped_column(String(255))
    image_width: Mapped[int] = mapped_column(Integer)
    image_height: Mapped[int] = mapped_column(Integer)
    file_format: Mapped[str] = mapped_column(String(10))
    storage_path: Mapped[str] = mapped_column(String(512))

    gaze_sequence: Mapped[dict | None] = mapped_column(JSON, nullable=True)
    hotspots: Mapped[dict | None] = mapped_column(JSON, nullable=True)
    overall_score: Mapped[float | None] = mapped_column(Float, nullable=True)
    entropy_score: Mapped[float | None] = mapped_column(Float, nullable=True)
    ai_insights: Mapped[dict | None] = mapped_column(JSON, nullable=True)
    ai_score: Mapped[int | None] = mapped_column(Integer, nullable=True)
    ai_score_reason: Mapped[str | None] = mapped_column(String(500), nullable=True)
    ai_cost_usd: Mapped[float | None] = mapped_column(Float, nullable=True)

    created_at: Mapped[datetime] = mapped_column(server_default=func.now())

    project: Mapped["Project"] = relationship(back_populates="analyses")  # noqa: F821
    aois: Mapped[list["AOI"]] = relationship(  # noqa: F821
        back_populates="analysis", cascade="all, delete-orphan"
    )