# models/market_data.py
from pydantic import BaseModel
from typing import  Dict, Any, List
import pandas as pd

class StockDataResponse(BaseModel):
    history: Dict[str, Any] # Simplified, consider serializing DataFrame properly
    info: Dict[str, Any]
    financials: List[Dict[str, Any]] # List of records for financials
    balance_sheet: List[Dict[str, Any]] # List of records for balance sheet
    cash_flow: List[Dict[str, Any]] # List of records for cash flow

class IndexData(BaseModel):
    current_price: float
    change: float
    change_pct: float
    volume: float

class SectorPerformance(BaseModel):
    sector: str
    change_pct: float

class EconomicIndicators(BaseModel):
    repo_rate: float
    inflation_rate: float