import io
import pymongo as pymongo
from PIL import Image

myclient = pymongo.MongoClient(
    "mongo_uri")
# database creation
db = myclient["Endoscopic_Guidance"]


def convert_image_to_byte_array(file):
    im = Image.open(file)
    image_bytes = io.BytesIO()
    im.save(image_bytes, format='JPEG')
    image = {
        'data': image_bytes.getvalue()
    }
    return image


def upload_image_data_mongodb(file):
    images = db.images
    # mongodb server
    imgByteArr = convert_image_to_byte_array(file=file)
    # database creation
    image_id = images.insert_one(imgByteArr).inserted_id
    print(image_id, 'Successfully inserted')