forust
7fb0a0e179
- S113: Add timeout=10 to all requests calls (74 fixes) - E722: Replace bare except: with except Exception: - B904: Replace redundant re-raise with bare raise - E402: Add noqa for intentional late imports after import_library() - S102/S307/S310/S311/S603/S605/S606/S607/S108: Add noqa for intentional usage - F601: Fix duplicate dict key in unsplash.py - N802: Rename ReplyCheck -> reply_check with backward compat alias - N813: Rename bs -> BS in icons.py - B007/B020: Rename loop var _j in animations.py - SIM102: Collapse nested if in autofwd.py - SIM113: Use enumerate() in calculator.py - A002: Add noqa for builtin shadowing in admlist.py - F811: Add noqa for cohere redefinition - edu_master: Fix ARG001, S108, S110, SIM117, apply --unsafe-fixes - Add modules_list.txt with full module inventory
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import os
|
|
|
|
from pyrogram import Client, filters
|
|
from pyrogram.types import Message
|
|
from utils.misc import modules_help, prefix
|
|
from utils.scripts import format_exc, import_library
|
|
|
|
clarifai = import_library('clarifai')
|
|
|
|
from clarifai.client.model import Model # noqa: E402
|
|
|
|
|
|
@Client.on_message(filters.command('cdxl', prefix) & filters.me)
|
|
async def cdxl(c: Client, message: Message):
|
|
try:
|
|
chat_id = message.chat.id
|
|
await message.edit_text('<code>Please Wait...</code>')
|
|
|
|
if len(message.command) > 1:
|
|
prompt = message.text.split(maxsplit=1)[1]
|
|
elif message.reply_to_message:
|
|
prompt = message.reply_to_message.text
|
|
else:
|
|
await message.edit_text(f'<b>Usage: </b><code>{prefix}vdxl [prompt/reply to prompt]</code>')
|
|
return
|
|
|
|
inference_params = {'width': 1024, 'height': 1024, 'steps': 50, 'cfg_scale': 9.0}
|
|
|
|
model_prediction = Model(
|
|
'https://clarifai.com/stability-ai/stable-diffusion-2/models/stable-diffusion-xl'
|
|
).predict_by_bytes(prompt.encode(), input_type='text', inference_params=inference_params)
|
|
|
|
output_base64 = model_prediction.outputs[0].data.image.base64
|
|
|
|
with open('sdxl_out.png', 'wb') as f:
|
|
f.write(output_base64)
|
|
|
|
await message.delete()
|
|
await c.send_photo(
|
|
chat_id,
|
|
photo='sdxl_out.png',
|
|
caption=f'<b>Prompt:</b><code>{prompt}</code>',
|
|
)
|
|
os.remove('sdxl_out.png')
|
|
|
|
except Exception as e:
|
|
await message.edit_text(f'An error occurred: {format_exc(e)}')
|
|
|
|
|
|
modules_help['cdxl'] = {
|
|
'cdxl [prompt/reply to prompt]*': 'Text to Image with SDXL model',
|
|
}
|