fixed types

2025-01-06 13:42:21 +00:00
parent 16ad2b80ee
commit aff0ae8fc6
16 changed files with 88 additions and 74 deletions
@@ -4,7 +4,7 @@ from pathlib import Path

 from dotenv import load_dotenv

-from shared.datastore import connect_minio
+from shared.datastore import Datastore
 from shared.mongodb import connect_mongodb, get_visual_communication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST
@@ -19,11 +19,13 @@ if __name__ == '__main__':
    # setup logging
    setup_logging()
    # connect to minIO
-    minio_client = connect_minio()
+    datastore = Datastore()
+    datastore.connect()
+    assert datastore._client is not None
    # connect to MongoDB
    collection, db, client = connect_mongodb()
    # get visual communication
    vis_com = get_visual_communication(collection)
    print(repr(vis_com))
-    image = vis_com.get_image(minio_client=minio_client)
+    image = vis_com.get_image(minio_client=datastore._client)
    image.show()