from pathlib import Path
from dotenv import load_dotenv
import os

from src.database import (
    connect,
    get_visual_communication
)

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()
    print(client.server_info())
    # get visual communication
    vis_com = get_visual_communication(collection)
    print(vis_com)