from app.db.base import Base
from app.db.session import engine

# Import models so Base.metadata knows about all tables
from app.models.document import Document  # noqa: F401
from app.models.document_review_state import DocumentReviewState  # noqa: F401
from app.models.document_version import DocumentVersion  # noqa: F401
from app.models.text_version import TextVersion  # noqa: F401
from app.models.extracted_field import ExtractedField  # noqa: F401
from app.models.layer1_candidate import Layer1Candidate  # noqa: F401


def init_db() -> None:
    Base.metadata.create_all(bind=engine)


if __name__ == "__main__":
    init_db()
    print("Database tables created.")