diff --git a/app/service/generate_image/utils/upload_sd_image.py b/app/service/generate_image/utils/upload_sd_image.py index c6a0632..0c3baec 100644 --- a/app/service/generate_image/utils/upload_sd_image.py +++ b/app/service/generate_image/utils/upload_sd_image.py @@ -10,6 +10,7 @@ import io import logging +import cv2 from PIL import Image from minio import Minio @@ -20,15 +21,12 @@ minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET def upload_png_sd(image, user_id, category, object_name): try: - image_file = io.BytesIO() - image = Image.fromarray(image) - image.save(image_file, format='JPEG') - image_file.seek(0) + _, img_byte_array = cv2.imencode('.jpg', image) minio_req = minio_client.put_object( GI_MINIO_BUCKET, f'{user_id}/{category}/{object_name}', - image_file, - len(image_file.getvalue()), + io.BytesIO(img_byte_array), + len(img_byte_array), content_type='image/jpeg' ) image_url = f"aida-users/{minio_req.object_name}"