from pydantic import BaseModel
from typing import Optional

class LaptopCreate(BaseModel):
    processor: str
    ram: int
    os: str
    ssd: int
    display_size: float
    resolution: str
    matrix_type: str
    gpu: str

class TVCreate(BaseModel):
    display: str
    tuners: str
    features: str
    os: str
    power_of_volume: str
    screen_size: int
    color: str

class LaptopResponse(BaseModel):
    id: int
    processor: str
    ram: int
    os: str
    ssd: int
    display_size: float
    resolution: str
    matrix_type: str
    gpu: str

    class Config:
        orm_mode = True

class TVResponse(BaseModel):
    id: int
    display: str
    tuners: str
    features: str
    os: str
    power_of_volume: str
    screen_size: int
    color: str

class PredictPriceResponse(BaseModel):
    predicted_price: float