backend/database/models.py

from dataclasses import dataclass
from typing import List, Optional
from datetime import datetime

@dataclass
class TaughtObject:
    id: str
    name: str
    category: str
    description: str
    status: str = "active"
    created_at: str = ""
    image_count: int = 0
    embedding_count: int = 0

@dataclass
class ObjectImage:
    id: str
    object_id: str
    file_path: str
    image_hash: str
    created_at: str = ""

@dataclass
class ObjectEmbedding:
    id: str
    object_id: str
    model_name: str
    model_version: str
    embedding: List[float]
    created_at: str = ""

@dataclass
class DetectionLog:
    id: str
    object_name: str
    detection_type: str
    confidence: float
    timestamp: str = ""