from pydantic import BaseModel, Field
from typing import List

class ProjectPlanRequest(BaseModel):
    project_title: str
    requirements: str
    features: List[str]
    platform: str = "AWS"  # Default to AWS
    additional_requirements: str = ""

class ProjectSection(BaseModel):
    title: str
    content: str

class ProjectPlan(BaseModel):
    executive_summary: str
    scope_objectives: dict
    architecture_overview: str
    component_design: List[dict]
    security_compliance: List[str]
    deployment_testing: List[str]
    team_roles: List[dict]
    cost_estimates: dict
    project_phases: List[dict]