dalle.py

import base64
from PIL import Image
import asyncio
import aiohttp
import json
import io

url = "http://127.0.0.1:8080/dalle"

def get_concat_h():
    images = [Image.open(x) for x in ['0.png', '1.png', '2.png', '3.png']]
    widths, heights = zip(*(i.size for i in images))

    total_width = sum(widths)
    max_height = max(heights)

    new_im = Image.new('RGB', (total_width, max_height))

    x_offset = 0
    for im in images:
        new_im.paste(im, (x_offset,0))
        x_offset += im.size[0]

    new_im.save('result.jpg')
    
async def asyncGetImgFromPrompt(num, prmpt):
    reqJson = {"text" : prmpt, "num_images" : num }
    req = json.dumps(reqJson)
    
    async with aiohttp.ClientSession() as session:
        async with session.post(url, data=req) as res:
            images = await res.json()
            for x in range(len(images)):
                imgData = base64.b64decode(images[x])
                buf = io.BytesIO(imgData)
                img = Image.open(buf)
                img.save("{}.png".format(x), format="png")
            get_concat_h()
    
async def asyncDiscordEntry(num, prompt):
    await asyncGetImgFromPrompt(num, prompt)