from app.database import SessionLocal
from app.models.chat_message import ChatMessage
from app.models.stream import Stream

db = SessionLocal()
streams = db.query(Stream).all()
print(f'Total streams: {len(streams)}')
for s in streams[:5]:
    chat_count = db.query(ChatMessage).filter(ChatMessage.stream_id == s.id).count()
    print(f'Stream {s.id}: {chat_count} chat messages')
db.close()