brian/visual_critical_discourse_analysis
1
0
Fork 0
Code Issues 11 Pull Requests Actions Projects 1 Releases Activity

implemented new interface

Browse Source
This commit is contained in:
brian
2025-01-06 11:29:31 +00:00
parent 9ee27b65a8
commit f28bff7e47
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
shared/mongodb/src/classes/visual_communication.py
+9 -9
View File
@@ -12,7 +12,7 @@ from PIL import Image
from pydantic import BaseModel, ConfigDict from pydantic import BaseModel, ConfigDict
from pymongo.collection import Collection from pymongo.collection import Collection
from shared.datastore import get_image, put_image from shared.datastore import Datastore
from shared.mongodb.src.classes import ModelData from shared.mongodb.src.classes import ModelData
@@ -39,10 +39,10 @@ class VisualCommunication(BaseModel):
"""Upload image to MinIO and return MD5 checksum of hashed image.""" """Upload image to MinIO and return MD5 checksum of hashed image."""
assert isinstance(image, Image.Image) assert isinstance(image, Image.Image)
assert isinstance(minio_client, Minio) assert isinstance(minio_client, Minio)
object_name = put_image( with Datastore() as ds:
client=minio_client, object_name = ds.put_image(
image=image, image=image,
) )
return object_name return object_name
@classmethod @classmethod
@@ -91,10 +91,10 @@ class VisualCommunication(BaseModel):
"""Load image data from minio.""" """Load image data from minio."""
assert isinstance(minio_client, Minio) assert isinstance(minio_client, Minio)
# get image from minio # get image from minio
image = get_image( with Datastore() as ds:
client=minio_client, image = ds.get_image(
object_name=self.object_name, object_name=self.object_name,
) )
return image return image
def save_to_mongo(self, collection: Collection) -> None: def save_to_mongo(self, collection: Collection) -> None: