models.py

#models.py

from sqlalchemy import Column, Integer, String, Float, DateTime, Text, JSON
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.sql import func

Base = declarative_base()


class ScamAnalysis(Base):
    __tablename__ = "scam_analyses"

    id = Column(Integer, primary_key=True, index=True)
    content = Column(Text, nullable=False)
    content_type = Column(String(20), nullable=False)  # "text" or "url"
    risk_level = Column(String(20), nullable=False)  # "Safe", "Suspicious", "Scam"
    confidence = Column(Float, nullable=False)
    reasoning = Column(Text, nullable=True)
    detected_language = Column(String(10), nullable=True)
    processing_time = Column(Float, nullable=True)
    additional_data = Column(JSON, nullable=True)
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())


class WhatsAppAnalysis(Base):
    __tablename__ = "whatsapp_analyses"

    id = Column(Integer, primary_key=True, index=True)
    message_id = Column(String(100), unique=True, nullable=False)
    sender = Column(String(100), nullable=False)
    content = Column(Text, nullable=False)
    message_type = Column(String(20), default="text")  # "text" or "voice"
    risk_level = Column(String(20), nullable=False)
    confidence = Column(Float, nullable=False)
    reasoning = Column(Text, nullable=True)
    processing_time = Column(Float, nullable=True)
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())