from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Float
from datetime import datetime, UTC
from app.db.database import Base

class File(Base):
    __tablename__ = "file"
    
    id = Column(Integer, primary_key=True, index=True)
    filename = Column(String, nullable=False)
    filepath = Column(String, nullable=False)
    filetype = Column(String, nullable=False)
    filesize = Column(Integer, nullable=False)
    heatmap_path = Column(String, nullable=True)
    geometry_path = Column(String, nullable=True)
    geometry_score = Column(Float, nullable=True)
    timeline_data = Column(String, nullable=True)
    ip_address = Column(String, nullable=True)
    
    owner_id = Column(Integer, ForeignKey("users.id"))
    created_at = Column(DateTime, default=lambda: datetime.now(UTC))
    
    status = Column(String, default="uploaded")
    result = Column(String, nullable=True)
    confidence = Column(Float, nullable=True)
    ai_explanation = Column(String, nullable=True)

    model_version_used = Column(String, nullable=True)