from pathlib import Path
from dotenv import load_dotenv
import os

from src.database import (
    connect,
    total_annotated
)

if __name__ == "__main__":
    # prepare env vars
    env_path = Path(__file__).parent.parent / "local.env"
    assert env_path.exists()
    load_dotenv(env_path)
    os.environ["MONGO_HOST"] = "localhost"
    # connect to database
    collection, db, client = connect()
    # get visual communication
    num_docs = total_annotated(collection)
    print(f"number of annotated  documents in database: {num_docs}")