"""
Pydantic schemas for business prediction endpoints
"""

from pydantic import BaseModel, Field
from typing import Dict


class BusinessPredictionRequest(BaseModel):
    """Request model for business requirements prediction"""

    model_config = {
        "json_schema_extra": {
            "example": {
                "business_description": "We are a sustainable fashion brand selling eco-friendly clothing for young professionals. Looking for creators who align with our values of sustainability and ethical fashion.",
                "budget_min": 500.0,
                "budget_max": 2000.0
            }
        }
    }

    business_description: str = Field(
        ...,
        description="Free-form text describing the business and requirements"
    )
    budget_min: float = Field(
        ...,
        ge=0,
        description="Minimum budget per post/campaign"
    )
    budget_max: float = Field(
        ...,
        ge=0,
        description="Maximum budget per post/campaign"
    )


# Response model moved to common.py as PredictionResponse (unified for both influencer and business)

