from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from models import ChatMessage, ChatSession, Base

SQLALCHEMY_DATABASE_URL = "sqlite:///./sql_app.db"
engine = create_engine(SQLALCHEMY_DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def check_messages():
    db = SessionLocal()
    sessions = db.query(ChatSession).all()
    print(f"Total Sessions: {len(sessions)}")
    
    for s in sessions:
        print(f"Session {s.id}: {s.title}")
        msgs = db.query(ChatMessage).filter(ChatMessage.session_id == s.id).all()
        for m in msgs:
            content_preview = m.content[:50] if m.content else "[NULL]"
            print(f"  - [{m.sender}] {content_preview}")
    
    db.close()

if __name__ == "__main__":
    check_messages()