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

# ====================== Analysis ======================
class AnalysisRequest(BaseModel):
    subreddits: List[str] = Field(..., description="List of subreddit names")
    posts_per_subreddit: int = Field(10, ge=1, le=50, description="Number of posts per subreddit")
    time_filter: str = Field("day", description="Time filter (hour, day, week, month, year, all)")

class AnalysisResponse(BaseModel):
    status: str
    message: str
    insights: List[Dict[str, Any]] = []

# ====================== Insights ======================
class InsightResponse(BaseModel):
    id: str
    subreddit: str
    post_title: str
    post_url: str
    timestamp: datetime
    model: str
    insight: Dict[str, Any]

# ====================== Query / RAG ======================
class QueryRequest(BaseModel):
    query: str = Field(..., description="The user's question")
    n_results: int = Field(5, ge=1, le=20, description="Number of sources to retrieve")

class QueryResponse(BaseModel):
    answer: str
    sources: List[Dict[str, Any]]  # each source: {title, subreddit, url}

# ====================== Market Intelligence ======================
class MarketIntelligenceRequest(BaseModel):
    idea: str = Field(..., description="Startup idea to analyze")

class MarketIntelligenceResponse(BaseModel):
    interest_score: int
    competitors: List[str]
    funding_news: List[Dict[str, str]]

# ====================== Startup Builder ======================
class GeneratePlanRequest(BaseModel):
    insight: Dict[str, Any] = Field(..., description="The insight object to generate a plan for")

class GeneratePlanResponse(BaseModel):
    plan: str

# ====================== Trends ======================
class TrendRequest(BaseModel):
    subreddit: Optional[str] = Field(None, description="Filter by subreddit (optional)")

class TrendResponse(BaseModel):
    dates: List[str]
    counts: List[int]

# ====================== Auth & Users ======================
class UserSignup(BaseModel):
    email: str
    password: str
    full_name: Optional[str] = None

class UserLogin(BaseModel):
    email: str
    password: str

class UserProfile(BaseModel):
    id: str
    email: str
    full_name: Optional[str]
    subscription_tier: str
    subscription_status: str
    created_at: datetime

# ====================== Subscription Plans ======================
class SubscriptionPlan(BaseModel):
    id: str
    name: str
    price: float
    features: Dict[str, Any]