updated filenames to avoid accidental running during test

misc/generate_random_prediction.py

@@ -0,0 +1,40 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from dotenv import load_dotenv
+
+from shared.data_store import connect_minio
+from shared.database import connect_mongodb
+from shared.database.classes import VisualCommunication
+from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
+
+if __name__ == '__main__':
+    # load in env file
+    env_path = Path(__file__).parent.parent / 'server.env'
+    assert env_path.exists()
+    load_dotenv(env_path)
+    # ensure env vars set
+    check_env(NECESSARY_ENV_VAR_LIST)
+    # setup logging
+    setup_logging()
+    # connect to minIO
+    minio_client = connect_minio()
+    # connect to MongoDB
+    collection, db, client = connect_mongodb()
+    # 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, minio_client=minio_client)
+        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)