# models.py (new file)
from pydantic import BaseModel

class Trade(BaseModel):
    symbol: str
    quantity: int
    price: float
    transaction_type: str  # "buy" or "sell"

class PortfolioData(BaseModel):
    cash_balance: float
    holdings: dict
    total_investment: list
    money_end: float