"""
Pydantic models and state definitions for the Polymer Datasheet Agent.
"""

from __future__ import annotations

import uuid
from datetime import datetime
from typing import Any, Literal, Optional

from pydantic import BaseModel, Field


# ── LangGraph State ──────────────────────────────────────────────────────────

class AgentState(BaseModel):
    """State that flows through the LangGraph workflow."""

    # Input
    input_mode: Literal["search", "upload"] = "search"
    manufacturer: str = ""
    polymer_family: str = ""
    grade: str = ""
    uploaded_text: str = ""               # Raw text from uploaded PDF/file

    # Search phase
    search_queries: list[str] = Field(default_factory=list)
    search_results: list[dict[str, Any]] = Field(default_factory=list)
    raw_content: str = ""                  # Aggregated raw text from web/upload

    # Parsing phase
    parsed_datasheet: Optional[DatasheetRecord] = None
    parsing_errors: list[str] = Field(default_factory=list)

    # Output
    status: str = "pending"
    message: str = ""


# ── Datasheet Record ─────────────────────────────────────────────────────────

class DatasheetRecord(BaseModel):
    """A single parsed polymer datasheet stored in the database."""

    id: str = Field(default_factory=lambda: str(uuid.uuid4()))
    created_at: str = Field(default_factory=lambda: datetime.now().isoformat())

    # General
    material_name: str = ""
    trade_name: str = ""
    manufacturer: str = ""
    polymer_family: str = ""
    grade: str = ""
    description: str = ""
    processing_method: str = ""
    features: str = ""
    applications: str = ""
    source_url: str = ""

    # Mechanical
    tensile_strength_mpa: str = ""
    tensile_modulus_mpa: str = ""
    elongation_at_break_pct: str = ""
    flexural_strength_mpa: str = ""
    flexural_modulus_mpa: str = ""
    impact_strength_charpy_kj_m2: str = ""
    impact_strength_izod_j_m: str = ""
    hardness_shore_d: str = ""
    hardness_rockwell: str = ""
    compressive_strength_mpa: str = ""

    # Thermal
    melting_temperature_c: str = ""
    glass_transition_temperature_c: str = ""
    heat_deflection_temperature_c: str = ""
    vicat_softening_temperature_c: str = ""
    continuous_service_temperature_c: str = ""
    thermal_conductivity_w_mk: str = ""
    coefficient_of_thermal_expansion_um_mk: str = ""
    flammability_rating: str = ""

    # Physical
    density_g_cm3: str = ""
    melt_flow_index_g_10min: str = ""
    water_absorption_pct: str = ""
    moisture_absorption_pct: str = ""
    specific_gravity: str = ""
    transparency: str = ""
    color: str = ""

    # Electrical
    dielectric_strength_kv_mm: str = ""
    dielectric_constant: str = ""
    volume_resistivity_ohm_cm: str = ""
    surface_resistivity_ohm: str = ""
    dissipation_factor: str = ""

    # Chemical Resistance
    acid_resistance: str = ""
    alkali_resistance: str = ""
    solvent_resistance: str = ""
    uv_resistance: str = ""
    weatherability: str = ""

    # Regulatory
    fda_approved: str = ""
    rohs_compliant: str = ""
    reach_compliant: str = ""
    ul94_rating: str = ""

    def to_flat_dict(self) -> dict[str, str]:
        """Return all fields as a flat dict for database storage."""
        return self.model_dump()


# ── Search Result ────────────────────────────────────────────────────────────

class SearchResult(BaseModel):
    """One web search result from Tavily."""
    title: str = ""
    url: str = ""
    content: str = ""
    raw_content: str = ""
    score: float = 0.0