from pydantic import BaseModel
from typing import List, Optional


class RiskFinding(BaseModel):
    risk_id: str
    description: str
    score: int


class AnalyzedClause(BaseModel):
    clause_number: int
    text: str
    risks: List[RiskFinding]
    suggestion: Optional[str] = None


class AnalysisReport(BaseModel):
    file_name: str
    language: str
    contract_type: str
    final_risk_score: int
    clauses: List[AnalyzedClause]