from __future__ import annotations

import os
from pathlib import Path

from database import connect
from database import total_annotated
from dotenv import load_dotenv

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}")