from pydantic import BaseModel, Field
from typing import List

# Input 정의
class MarketingPromptInput(BaseModel):
    customer_name : str = Field(..., description = "마케팅 대상 사업체 이름")
    region : str = Field(..., description = "마케팅 대상 지역")
    detail_region_info : str = Field(..., description = "마케팅 대상 지역 상세")


# Output 정의
class BrandIdentity(BaseModel):
    location_feature_analysis: str = Field(..., description="입지 특성 분석 (80자 이상 150자 이하)", min_length = 80, max_length = 150) # min/max constraint는 현재 openai json schema 등에서 작동하지 않는다는 보고가 있음. 
    concept_scalability: str = Field(..., description="컨셉 확장성 (80자 이상 150자 이하)", min_length = 80, max_length = 150)


class MarketPositioning(BaseModel):
    category_definition: str = Field(..., description="마케팅 카테고리")
    core_value: str = Field(..., description="마케팅 포지션 핵심 가치")

class AgeRange(BaseModel):
    min_age : int = Field(..., ge=0, le=100)
    max_age : int = Field(..., ge=0, le=100)


class TargetPersona(BaseModel):
    persona: str = Field(..., description="타겟 페르소나 이름/설명")
    age: AgeRange = Field(..., description="타겟 페르소나 나이대")
    favor_target: List[str] = Field(..., description="페르소나의 선호 요소")
    decision_trigger: str = Field(..., description="구매 결정 트리거")


class SellingPoint(BaseModel):
    english_category: str = Field(..., description="셀링포인트 카테고리(영문)")
    korean_category: str = Field(..., description="셀링포인트 카테고리(한글)")
    description: str = Field(..., description="상세 설명")
    score: int = Field(..., ge=0, le=100, description="점수 (100점 만점)")
    
class MarketingPromptOutput(BaseModel):
    brand_identity: BrandIdentity = Field(..., description="브랜드 아이덴티티")
    market_positioning: MarketPositioning = Field(..., description="시장 포지셔닝")
    target_persona: List[TargetPersona] = Field(..., description="타겟 페르소나")
    selling_points: List[SellingPoint] = Field(..., description="셀링 포인트")
    target_keywords: List[str] = Field(..., description="타겟 키워드 리스트")