upscale.py

"""from os import remove
from pyrogram import filters
from lexica import Client as LexicaClient
from pyrogram.errors.exceptions.bad_request_400 import PhotoInvalidDimensions
from YukkiMusic import app

lexica_client = LexicaClient()

def upscale_image(image: bytes) -> bytes:
    return lexica_client.upscale(image)

@app.on_message(filters.command("upscale"))
@utils.capture_err
async def upscale_reply_image(client, message):
    if not message.reply_to_message or not message.reply_to_message.photo:
        return await message.reply_text("ʀᴇᴘʟʏ ᴛᴏ ᴀ ᴘʜᴏᴛᴏ ᴛᴏ ᴜᴘsᴄᴀʟᴇ ɪᴛ....😑")
    if message.reply_to_message.photo:
        a = await message.reply_text("ᴘʟᴇᴀsᴇ ᴡᴀɪᴛ ᴀ ᴍᴏᴍᴇɴᴛ......")
        photo = await client.download_media(message.reply_to_message.photo.file_id)

        with open(photo, 'rb') as f:
            image_bytes = f.read()
        try:
            upscaled_image_bytes = upscale_image(image_bytes)
            await a.edit("ᴀʟᴍᴏsᴛ ᴅᴏɴᴇ......❣️")
            with open('upscaled.png', 'wb') as f:
                f.write(upscaled_image_bytes)
                try:
                    await message.reply_photo(photo='upscaled.png')
                    remove('upscaled.png')
                    await a.delete()
                except PhotoInvalidDimensions:
                    await message.reply_document('upscaled.png')
                    remove('upscaled.png')
                    await a.delete()
        except Exception as e:
            remove('upscaled.png')
            await a.edit(e)"""