models.py

from typing import List, Optional
from pydantic import BaseModel, Field
from datetime import datetime


class Persona(BaseModel):
    """Represents a persona that will interact on Reddit"""
    username: str
    info: str  # Single text field with all persona information


class CompanyInfo(BaseModel):
    """Company information for content generation"""
    name: str
    website: str
    description: str


class ContentInput(BaseModel):
    """Input for content calendar generation"""
    company_info: CompanyInfo
    personas: List[Persona] = Field(min_length=2)
    subreddits: List[str]
    chatgpt_queries: List[str]
    posts_per_week: int = Field(ge=1, le=50)
    start_week: Optional[int] = Field(default=0, description="Week number, 0 is current week")


class Comment(BaseModel):
    """A comment on a post"""
    persona_name: str
    content: str
    delay_hours: float = Field(description="Hours after post creation to comment")


class Post(BaseModel):
    """A Reddit post with associated comments"""
    subreddit: str
    title: str
    content: str
    persona_name: str
    target_query: str
    scheduled_time: str
    comments: List[Comment]
    reasoning: str = Field(description="Why this post/comment combination is valuable")
    keywords: List[str] = Field(default_factory=list, description="Keywords/tags for this post")


class ContentCalendar(BaseModel):
    """A week's worth of content"""
    week_number: int
    start_date: str
    end_date: str
    posts: List[Post]
    metadata: dict = Field(default_factory=dict)