from app.database import get_sync_db
from app.models import BillDocument, BillBrief, Bill

db = get_sync_db()
total_bills = db.query(Bill).count()
docs = db.query(BillDocument).filter(BillDocument.raw_text.isnot(None)).count()
briefs = db.query(BillBrief).count()
full_briefs = db.query(BillBrief).filter_by(brief_type="full").count()
amendment_briefs = db.query(BillBrief).filter_by(brief_type="amendment").count()
print(f"Total bills:       {total_bills}")
print(f"Documents fetched: {docs}")
print(f"Briefs generated:  {briefs} ({full_briefs} full, {amendment_briefs} amendments)")
print(f"Remaining:         {total_bills - briefs}")
db.close()