#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project ：trinity_client
@File    ：upload_image.py
@Author  ：周成融
@Date    ：2023/8/28 13:49:20
@detail  ：
"""
import io
import logging

from PIL import Image
from minio import Minio

from app.core.config import *

minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)


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)
        minio_req = minio_client.put_object(
            GI_MINIO_BUCKET,
            f'{user_id}/{category}/{object_name}',
            image_file,
            len(image_file.getvalue()),
            content_type='image/jpeg'
        )
        image_url = f"aida-users/{minio_req.object_name}"
        return image_url
    except Exception as e:
        logging.warning(f"upload_png_mask runtime exception : {e}")