research-agent/models.py

"""
Data models for Disaster Risk Construction Planner
Pydantic models for FastAPI compatibility and Blaxel deployment
"""

from typing import Optional, List, Literal, Dict, Any
from pydantic import BaseModel, Field
from datetime import datetime


# Input Types
BuildingType = Literal[
    "residential_single_family",
    "residential_multi_family",
    "residential_high_rise",
    "commercial_office",
    "commercial_retail",
    "industrial_warehouse",
    "institutional_school",
    "institutional_hospital",
    "infrastructure_bridge",
    "mixed_use"
]

RiskLevel = Literal["CRITICAL", "HIGH", "MODERATE", "LOW"]


# Base Models
class Coordinates(BaseModel):
    """Geographic coordinates"""
    latitude: float
    longitude: float


class LocationInfo(BaseModel):
    """Location information"""
    name: str
    coordinates: Coordinates
    administrative_area: str


# Risk Assessment Models
class HazardDetail(BaseModel):
    """Detailed information about a specific hazard"""
    status: str
    description: str
    distance: Optional[str] = None
    direction: Optional[str] = None
    severity: Optional[str] = None


class SeismicHazards(BaseModel):
    """Seismic hazard information"""
    active_fault: HazardDetail
    ground_shaking: HazardDetail
    liquefaction: HazardDetail
    tsunami: HazardDetail
    earthquake_induced_landslide: HazardDetail
    fissure: HazardDetail
    ground_rupture: HazardDetail


class VolcanicHazards(BaseModel):
    """Volcanic hazard information"""
    active_volcano: HazardDetail
    potentially_active_volcano: HazardDetail
    inactive_volcano: HazardDetail
    ashfall: HazardDetail
    pyroclastic_flow: HazardDetail
    lahar: HazardDetail
    lava: HazardDetail
    ballistic_projectile: HazardDetail
    base_surge: HazardDetail
    volcanic_tsunami: HazardDetail


class HydroHazards(BaseModel):
    """Hydrometeorological hazard information"""
    flood: HazardDetail
    rain_induced_landslide: HazardDetail
    storm_surge: HazardDetail
    severe_winds: HazardDetail


class HazardData(BaseModel):
    """Complete hazard data from risk assessment"""
    seismic: SeismicHazards
    volcanic: VolcanicHazards
    hydrometeorological: HydroHazards


class RiskSummary(BaseModel):
    """Summary of overall risk assessment"""
    overall_risk_level: RiskLevel
    total_hazards_assessed: int
    high_risk_count: int
    moderate_risk_count: int
    critical_hazards: List[str] = Field(default_factory=list)


class FacilityInfo(BaseModel):
    """Critical facilities information from risk assessment"""
    schools: Dict[str, Any] | List[Dict[str, Any]] = Field(default_factory=dict)
    hospitals: Dict[str, Any] | List[Dict[str, Any]] = Field(default_factory=dict)
    road_networks: Dict[str, Any] | List[Dict[str, Any]] = Field(default_factory=list)


class Metadata(BaseModel):
    """Metadata for data sources"""
    timestamp: str
    source: str
    cache_status: str
    ttl: int


class RiskData(BaseModel):
    """Complete risk assessment data"""
    success: bool
    summary: RiskSummary
    location: LocationInfo
    hazards: HazardData
    facilities: FacilityInfo
    metadata: Metadata


# Construction Recommendations Models
class RecommendationDetail(BaseModel):
    """Detailed construction recommendation"""
    hazard_type: str
    recommendation: str
    rationale: str
    source_url: Optional[str] = None


class BuildingCodeReference(BaseModel):
    """Building code reference"""
    code_name: str
    section: str
    requirement: str


class Recommendations(BaseModel):
    """Construction recommendations"""
    general_guidelines: List[str] = Field(default_factory=list)
    seismic_recommendations: List[RecommendationDetail] = Field(default_factory=list)
    volcanic_recommendations: List[RecommendationDetail] = Field(default_factory=list)
    hydrometeorological_recommendations: List[RecommendationDetail] = Field(default_factory=list)
    priority_actions: List[str] = Field(default_factory=list)
    building_codes: List[BuildingCodeReference] = Field(default_factory=list)


# Material Cost Models
class MaterialCost(BaseModel):
    """Material cost information"""
    material_name: str
    category: str
    unit: str
    price_per_unit: float
    currency: str
    quantity_needed: Optional[float] = None
    total_cost: Optional[float] = None
    source: Optional[str] = None


class CostEstimate(BaseModel):
    """Cost estimate range"""
    low: float
    mid: float
    high: float
    currency: str


class CostData(BaseModel):
    """Complete cost analysis data"""
    materials: List[MaterialCost] = Field(default_factory=list)
    total_estimate: Optional[CostEstimate] = None
    market_conditions: str = ""
    last_updated: str = ""


# Critical Facilities Models
class FacilityDetail(BaseModel):
    """Detailed facility information"""
    name: str
    type: str
    distance_meters: float
    travel_time_minutes: float
    directions: str
    coordinates: Coordinates


class RoadDetail(BaseModel):
    """Road network information"""
    name: str
    type: Literal["primary", "secondary"]
    distance_meters: float


class FacilityData(BaseModel):
    """Complete facility location data"""
    schools: List[FacilityDetail] = Field(default_factory=list)
    hospitals: List[FacilityDetail] = Field(default_factory=list)
    emergency_services: List[FacilityDetail] = Field(default_factory=list)
    utilities: List[FacilityDetail] = Field(default_factory=list)
    road_networks: List[RoadDetail] = Field(default_factory=list)


# Final Output Models
class PlanMetadata(BaseModel):
    """Construction plan metadata"""
    generated_at: str
    building_type: BuildingType
    building_area: Optional[float]
    location: LocationInfo
    coordinates: Coordinates


class ExecutiveSummary(BaseModel):
    """Executive summary of construction plan"""
    overall_risk: str
    critical_concerns: List[str] = Field(default_factory=list)
    key_recommendations: List[str] = Field(default_factory=list)
    building_specific_notes: List[str] = Field(default_factory=list)


class ExportFormats(BaseModel):
    """Export format URLs"""
    pdf_url: Optional[str] = None
    json_url: Optional[str] = None


class ConstructionPlan(BaseModel):
    """Complete construction plan output"""
    metadata: PlanMetadata
    executive_summary: ExecutiveSummary
    risk_assessment: RiskData
    construction_recommendations: Recommendations
    material_costs: CostData
    critical_facilities: FacilityData
    export_formats: ExportFormats


# Error Handling Models
class ErrorDetail(BaseModel):
    """Error detail information"""
    code: str
    message: str
    details: Optional[Dict[str, Any]] = None
    retry_possible: bool = False


class ErrorResponse(BaseModel):
    """Error response structure"""
    success: bool = False
    error: Optional[ErrorDetail] = None
    partial_results: Optional[Dict[str, Any]] = None