models.py

from sqlalchemy import Column, Integer, String, Text, DateTime, Float
from datetime import datetime
from database import Base


class RawData(Base):
    __tablename__ = "raw_data"

    id = Column(Integer, primary_key=True, index=True)
    source = Column(String)
    title = Column(String)
    content = Column(Text)
    tags = Column(String)
    created_at = Column(DateTime, default=datetime.utcnow)


class Trend(Base):
    __tablename__ = "trends"

    id = Column(Integer, primary_key=True, index=True)
    topic = Column(String, index=True)
    mentions = Column(Integer)
    growth_rate = Column(Float)
    source_count = Column(Integer)
    stage = Column(String)
    last_updated = Column(DateTime, default=datetime.utcnow)