import enum

from pydantic import BaseModel, ConfigDict, Field


class RiskCategory(enum.StrEnum):
    HIGH_RISK_FINANCE = "high-risk financial services"
    LEAD_GEN = "third-party lead generation services and marketing"
    DEBT_COLLECTION = "debt collection and forgiveness"
    GET_RICH_QUICK = "get rich quick schema"
    ILLEGAL_SUBSTANCES = "illegal substances/articles"
    PRESCRIPTION_DRUGS = "prescription drugs"
    GAMBLING = "gambling"
    SHAFT = "S.H.A.F.T."
    URL_SHORTENER = "URL shortener"


class PhraseAnalysis(BaseModel):
    phrase: str
    category: RiskCategory


class ContentScanResponse(BaseModel):
    phrases: list[PhraseAnalysis]


class ContentRecommendations(BaseModel):
    sensitive_words: list[str] | None = Field(None, alias="sensitiveWords")
    included_artist_name: bool | None = Field(None, alias="includedArtistName")

    model_config = ConfigDict(populate_by_name=True)