updated to use new class

web_ui/src/app/init_app.py

@@ -16,11 +16,11 @@ from pydantic import ValidationError
 from pymongo.collection import Collection
 
 from .layout import app_layout
+from shared.data_store import delete as delete_from_minio
 from shared.database import count_documents
 from shared.database import get_visual_communication
 from shared.database import NoDocumentFoundException
 from shared.database import upsert_annotation
-from shared.database import upsert_visual_communication
 from shared.database import VisualCommunication
 from shared.dto import ModelData
 
@@ -120,34 +120,52 @@ def init_app(
         filename_list: list[str] | None,
     ) -> tuple[str | None, str | None]:
         """Upload image to database through web ui."""
-        # stop early if possible
+        try:
-        if content_list is None or filename_list is None:
+            # stop if no input
-            logging.info('nothing to upload.')
+            assert (
-            return None, 'nothing to upload'.title()
+                content_list is not None
-        # build list of visual communication
+            ) and (
-        vis_com_list = []
+                filename_list is not None
+            ), 'nothing to upload'
+            # handle input
+            failed_filename_list = []
             for content, filename in zip(content_list, filename_list):
                 try:
+                    # decode image content
                     image = VisualCommunication.decode_image(content=content)
-                vis_com = VisualCommunication(
+                except Exception:
+                    logging.debug('failed decoding %s', filename)
+                    failed_filename_list.append(filename)
+                    continue
+                try:
+                    # instantiate to upload image to minio
+                    vis_com = VisualCommunication.from_name_and_image(
                         name=filename,
                         image=image,
+                        minio_client=minio_client,
                     )
                 except Exception as exc:
-                logging.error('failed handling data from file: %s', filename)
                     logging.debug(exc)
+                    failed_filename_list.append(filename)
                     continue
-            vis_com_list.append(vis_com)
+                try:
-        # upsert documents
+                    # save to mongodb
-        success = upsert_visual_communication(
+                    vis_com.save_to_mongo(collection=mongo_collection)
-            collection=mongo_collection,
+                except Exception as exc:
-            visual_communication_list=vis_com_list,
+                    logging.debug(exc)
+                    failed_filename_list.append(filename)
+                    # remove document from minio
+                    delete_from_minio(
+                        client=minio_client,
+                        object_name=vis_com.object_name,
                     )
-        if success:
+            assert len(failed_filename_list) == 0, f"failed uploading:{
-            logging.info('uploaded %s files to database', len(vis_com_list))
+                '\n'.join(failed_filename_list)
+            }"
+        except Exception as exc:
+            logging.debug(exc)
+            return None, str(exc).title()
         return 'successfully uploaded images'.title(), None
-        logging.error('failed uploading images to database')
-        return None, 'failed uploading images'.title()
 
     # define callback: cycle visual communication data
     @app.callback(