from __future__ import annotations

from pathlib import Path

from database import VisualCommunication


if __name__ == '__main__':
    # get list of image paths
    test_dir = Path(__file__).parent
    img_dir = test_dir / 'imgs'
    img_path_list = [path for path in img_dir.glob('*.jpeg') if path.is_file()]
    print(img_path_list)
    # instantiate data object
    vis_com_list = [
        VisualCommunication.from_file(path)
        for path
        in img_path_list
    ]
    # generate random predictions
    for vis_com in vis_com_list:
        vis_com.generate_random_prediction()
    for vis_com in vis_com_list:
        print(vis_com)