from sqlalchemy import Column, Integer, String, Text, DateTime, ForeignKey, JSON
from datetime import datetime, timezone
from app.database import Base

class Document(Base):
    __tablename__ = "documents"
    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
    title = Column(String(500))
    category = Column(String(50), index=True)
    original_image = Column(Text)
    extracted_text = Column(Text)
    summary = Column(Text)
    extracted_data = Column(JSON)
    risk_alerts = Column(JSON)
    tags = Column(JSON)
    file_size = Column(Integer)
    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc), index=True)