overspend1/Moon-UB-overver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update vdxl api & new models

Browse Source
This commit is contained in:
Abhi
2024-08-17 08:03:12 +00:00
parent 7f6c1e3b32
commit 06af0febb3
1 changed files with 580 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

585
modules/aiutils.py
View File

@@ -12,10 +12,10 @@ from utils.scripts import format_exc
api_url = "https://visioncraft.top"
async def fetch_models():
async def fetch_models(category: str):
"""Get all available SDXL models"""
async with aiohttp.ClientSession() as session, session.get(
"https://visioncraft.top/sd/models-sdxl"
f"https://visioncraft.top/image/models/{category}"
) as response:
return await response.json()
@@ -31,7 +31,7 @@ async def generate_gifs(data):
async def generate_images(data):
"""Helper Function to generate image using SDXL"""
async with aiohttp.ClientSession() as session, session.post(
f"{api_url}/sd", json=data
f"{api_url}/image/generate", json=data
) as response:
return await response.read()
@@ -52,7 +52,8 @@ async def vdxl(c: Client, message: Message):
try:
chat_id = message.chat.id
models = await fetch_models()
model_category = "SDXL-1.0"
models = await fetch_models(category=model_category)
if not message.reply_to_message and len(message.command) > 2:
model_found = False
@@ -116,7 +117,574 @@ async def vdxl(c: Client, message: Message):
except MessageTooLong:
await message.edit_text(
"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/sd/models-sdxl'> Here </a>"
f"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/image/models/{model_category}'> Here </a>"
)
return
except Exception as e:
await message.edit_text(f"An error occurred: {format_exc(e)}")
@Client.on_message(filters.command("vdxl2", prefix) & filters.me)
async def vdxl2(c: Client, message: Message):
"""Text to Image Generation Using SDXL"""
await message.edit_text("<code>Please Wait...</code>")
try:
chat_id = message.chat.id
model_category = "SD-2.0"
models = await fetch_models(category=model_category)
if not message.reply_to_message and len(message.command) > 2:
model_found = False
for m in models:
if message.text.startswith(f"{prefix}vdxl {m}"):
model = m
prompt = message.text[len(f"{prefix}vdxl {m}") :].strip()
model_found = True
break
if not model_found:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
elif message.reply_to_message and len(message.command) > 1:
model = message.text.split(maxsplit=1)[1]
print(model)
if model in models:
prompt = message.reply_to_message.text
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
data = {
"prompt": prompt,
"model": model,
"negative_prompt": "canvas frame, cartoon, ((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), weird colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy",
"token": vca_api_key,
"image_count": 1,
"width": 1024,
"height": 1024,
"steps": 30,
"cfg_scale": 8,
"sampler": "Euler",
"upscale": True,
}
response = await generate_images(data)
try:
with open("generated_image.png", "wb") as f:
f.write(response)
await message.delete()
await c.send_document(
chat_id,
document="generated_image.png",
caption=f"<b>Prompt: </b><code>{prompt}</code>\n<b>Model: </b><code>{model}</code>",
)
os.remove("generated_image.png")
except KeyError:
try:
error = response["error"]
await message.edit_text(f"<code>{error}</code>")
except KeyError:
detail = response["detail"]
await message.edit_text(f"<code>{detail}</code>")
except MessageTooLong:
await message.edit_text(
f"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/image/models/{model_category}'> Here </a>"
)
return
except Exception as e:
await message.edit_text(f"An error occurred: {format_exc(e)}")
@Client.on_message(filters.command("vdxl2", prefix) & filters.me)
async def vdxl3(c: Client, message: Message):
"""Text to Image Generation Using SDXL"""
await message.edit_text("<code>Please Wait...</code>")
try:
chat_id = message.chat.id
model_category = "SD-3.0"
models = await fetch_models(category=model_category)
if not message.reply_to_message and len(message.command) > 2:
model_found = False
for m in models:
if message.text.startswith(f"{prefix}vdxl {m}"):
model = m
prompt = message.text[len(f"{prefix}vdxl {m}") :].strip()
model_found = True
break
if not model_found:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
elif message.reply_to_message and len(message.command) > 1:
model = message.text.split(maxsplit=1)[1]
print(model)
if model in models:
prompt = message.reply_to_message.text
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
data = {
"prompt": prompt,
"model": model,
"negative_prompt": "canvas frame, cartoon, ((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), weird colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy",
"token": vca_api_key,
"image_count": 1,
"width": 1024,
"height": 1024,
"steps": 30,
"cfg_scale": 8,
"sampler": "Euler",
"upscale": True,
}
response = await generate_images(data)
try:
with open("generated_image.png", "wb") as f:
f.write(response)
await message.delete()
await c.send_document(
chat_id,
document="generated_image.png",
caption=f"<b>Prompt: </b><code>{prompt}</code>\n<b>Model: </b><code>{model}</code>",
)
os.remove("generated_image.png")
except KeyError:
try:
error = response["error"]
await message.edit_text(f"<code>{error}</code>")
except KeyError:
detail = response["detail"]
await message.edit_text(f"<code>{detail}</code>")
except MessageTooLong:
await message.edit_text(
f"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/image/models/{model_category}'> Here </a>"
)
return
except Exception as e:
await message.edit_text(f"An error occurred: {format_exc(e)}")
@Client.on_message(filters.command("vfxl", prefix) & filters.me)
async def vfxl(c: Client, message: Message):
"""Text to Image Generation Using SDXL"""
await message.edit_text("<code>Please Wait...</code>")
try:
chat_id = message.chat.id
model_category = "FLUX.1"
models = await fetch_models(category=model_category)
if not message.reply_to_message and len(message.command) > 2:
model_found = False
for m in models:
if message.text.startswith(f"{prefix}vdxl {m}"):
model = m
prompt = message.text[len(f"{prefix}vdxl {m}") :].strip()
model_found = True
break
if not model_found:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
elif message.reply_to_message and len(message.command) > 1:
model = message.text.split(maxsplit=1)[1]
print(model)
if model in models:
prompt = message.reply_to_message.text
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
data = {
"prompt": prompt,
"model": model,
"negative_prompt": "canvas frame, cartoon, ((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), weird colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy",
"token": vca_api_key,
"image_count": 1,
"width": 1024,
"height": 1024,
"steps": 30,
"cfg_scale": 8,
"sampler": "Euler",
"upscale": True,
}
response = await generate_images(data)
try:
with open("generated_image.png", "wb") as f:
f.write(response)
await message.delete()
await c.send_document(
chat_id,
document="generated_image.png",
caption=f"<b>Prompt: </b><code>{prompt}</code>\n<b>Model: </b><code>{model}</code>",
)
os.remove("generated_image.png")
except KeyError:
try:
error = response["error"]
await message.edit_text(f"<code>{error}</code>")
except KeyError:
detail = response["detail"]
await message.edit_text(f"<code>{detail}</code>")
except MessageTooLong:
await message.edit_text(
f"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/image/models/{model_category}'> Here </a>"
)
return
except Exception as e:
await message.edit_text(f"An error occurred: {format_exc(e)}")
@Client.on_message(filters.command("vpxl", prefix) & filters.me)
async def vpxl(c: Client, message: Message):
"""Text to Image Generation Using SDXL"""
await message.edit_text("<code>Please Wait...</code>")
try:
chat_id = message.chat.id
model_category = "Playground-v2"
models = await fetch_models(category=model_category)
if not message.reply_to_message and len(message.command) > 2:
model_found = False
for m in models:
if message.text.startswith(f"{prefix}vdxl {m}"):
model = m
prompt = message.text[len(f"{prefix}vdxl {m}") :].strip()
model_found = True
break
if not model_found:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
elif message.reply_to_message and len(message.command) > 1:
model = message.text.split(maxsplit=1)[1]
print(model)
if model in models:
prompt = message.reply_to_message.text
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
data = {
"prompt": prompt,
"model": model,
"negative_prompt": "canvas frame, cartoon, ((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), weird colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy",
"token": vca_api_key,
"image_count": 1,
"width": 1024,
"height": 1024,
"steps": 30,
"cfg_scale": 8,
"sampler": "Euler",
"upscale": True,
}
response = await generate_images(data)
try:
with open("generated_image.png", "wb") as f:
f.write(response)
await message.delete()
await c.send_document(
chat_id,
document="generated_image.png",
caption=f"<b>Prompt: </b><code>{prompt}</code>\n<b>Model: </b><code>{model}</code>",
)
os.remove("generated_image.png")
except KeyError:
try:
error = response["error"]
await message.edit_text(f"<code>{error}</code>")
except KeyError:
detail = response["detail"]
await message.edit_text(f"<code>{detail}</code>")
except MessageTooLong:
await message.edit_text(
f"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/image/models/{model_category}'> Here </a>"
)
return
except Exception as e:
await message.edit_text(f"An error occurred: {format_exc(e)}")
@Client.on_message(filters.command("vpixl", prefix) & filters.me)
async def vpixl(c: Client, message: Message):
"""Text to Image Generation Using SDXL"""
await message.edit_text("<code>Please Wait...</code>")
try:
chat_id = message.chat.id
model_category = "PixArt"
models = await fetch_models(category=model_category)
if not message.reply_to_message and len(message.command) > 2:
model_found = False
for m in models:
if message.text.startswith(f"{prefix}vdxl {m}"):
model = m
prompt = message.text[len(f"{prefix}vdxl {m}") :].strip()
model_found = True
break
if not model_found:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
elif message.reply_to_message and len(message.command) > 1:
model = message.text.split(maxsplit=1)[1]
print(model)
if model in models:
prompt = message.reply_to_message.text
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
data = {
"prompt": prompt,
"model": model,
"negative_prompt": "canvas frame, cartoon, ((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), weird colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy",
"token": vca_api_key,
"image_count": 1,
"width": 1024,
"height": 1024,
"steps": 30,
"cfg_scale": 8,
"sampler": "Euler",
"upscale": True,
}
response = await generate_images(data)
try:
with open("generated_image.png", "wb") as f:
f.write(response)
await message.delete()
await c.send_document(
chat_id,
document="generated_image.png",
caption=f"<b>Prompt: </b><code>{prompt}</code>\n<b>Model: </b><code>{model}</code>",
)
os.remove("generated_image.png")
except KeyError:
try:
error = response["error"]
await message.edit_text(f"<code>{error}</code>")
except KeyError:
detail = response["detail"]
await message.edit_text(f"<code>{detail}</code>")
except MessageTooLong:
await message.edit_text(
f"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/image/models/{model_category}'> Here </a>"
)
return
except Exception as e:
await message.edit_text(f"An error occurred: {format_exc(e)}")
@Client.on_message(filters.command("vkxl", prefix) & filters.me)
async def vkxl(c: Client, message: Message):
"""Text to Image Generation Using SDXL"""
await message.edit_text("<code>Please Wait...</code>")
try:
chat_id = message.chat.id
model_category = "Kolors"
models = await fetch_models(category=model_category)
if not message.reply_to_message and len(message.command) > 2:
model_found = False
for m in models:
if message.text.startswith(f"{prefix}vdxl {m}"):
model = m
prompt = message.text[len(f"{prefix}vdxl {m}") :].strip()
model_found = True
break
if not model_found:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
elif message.reply_to_message and len(message.command) > 1:
model = message.text.split(maxsplit=1)[1]
print(model)
if model in models:
prompt = message.reply_to_message.text
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
data = {
"prompt": prompt,
"model": model,
"negative_prompt": "canvas frame, cartoon, ((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), weird colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy",
"token": vca_api_key,
"image_count": 1,
"width": 1024,
"height": 1024,
"steps": 30,
"cfg_scale": 8,
"sampler": "Euler",
"upscale": True,
}
response = await generate_images(data)
try:
with open("generated_image.png", "wb") as f:
f.write(response)
await message.delete()
await c.send_document(
chat_id,
document="generated_image.png",
caption=f"<b>Prompt: </b><code>{prompt}</code>\n<b>Model: </b><code>{model}</code>",
)
os.remove("generated_image.png")
except KeyError:
try:
error = response["error"]
await message.edit_text(f"<code>{error}</code>")
except KeyError:
detail = response["detail"]
await message.edit_text(f"<code>{detail}</code>")
except MessageTooLong:
await message.edit_text(
f"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/image/models/{model_category}'> Here </a>"
)
return
except Exception as e:
await message.edit_text(f"An error occurred: {format_exc(e)}")
@Client.on_message(filters.command("vfxl", prefix) & filters.me)
async def vfxl(c: Client, message: Message):
"""Text to Image Generation Using SDXL"""
await message.edit_text("<code>Please Wait...</code>")
try:
chat_id = message.chat.id
model_category = "FLUX.1"
models = await fetch_models(category=model_category)
if not message.reply_to_message and len(message.command) > 2:
model_found = False
for m in models:
if message.text.startswith(f"{prefix}vdxl {m}"):
model = m
prompt = message.text[len(f"{prefix}vdxl {m}") :].strip()
model_found = True
break
if not model_found:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
elif message.reply_to_message and len(message.command) > 1:
model = message.text.split(maxsplit=1)[1]
print(model)
if model in models:
prompt = message.reply_to_message.text
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
else:
return await message.edit_text(
f"<b>Usage: </b><code>{prefix}vdxl [model]* [prompt/reply to prompt]*</code>\n <b>Available Models:</b> <blockquote>{models}</blockquote>"
)
data = {
"prompt": prompt,
"model": model,
"negative_prompt": "canvas frame, cartoon, ((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), weird colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy",
"token": vca_api_key,
"image_count": 1,
"width": 1024,
"height": 1024,
"steps": 30,
"cfg_scale": 8,
"sampler": "Euler",
"upscale": True,
}
response = await generate_images(data)
try:
with open("generated_image.png", "wb") as f:
f.write(response)
await message.delete()
await c.send_document(
chat_id,
document="generated_image.png",
caption=f"<b>Prompt: </b><code>{prompt}</code>\n<b>Model: </b><code>{model}</code>",
)
os.remove("generated_image.png")
except KeyError:
try:
error = response["error"]
await message.edit_text(f"<code>{error}</code>")
except KeyError:
detail = response["detail"]
await message.edit_text(f"<code>{detail}</code>")
except MessageTooLong:
await message.edit_text(
f"<b>Model List is too long</b> See the Full List <a href='https://visioncraft.top/image/models/{model_category}'> Here </a>"
)
return
@@ -176,6 +744,13 @@ async def vgif(c: Client, message: Message):
modules_help["aiutils"] = {
"vdxl [model]* [prompt/reply to prompt]*": "Text to Image with SDXL model",
"vdxl2 [model]* [prompt/reply to prompt]*": "Text to Image with SDXL-2 model",
"vdxl3 [model]* [prompt/reply to prompt]*": "Text to Image with SDXL-3 model",
"vpxl [model]* [prompt/reply to prompt]*": "Text to Image with Playground model",
"vfxl [model]* [prompt/reply to prompt]*": "Text to Image with FLUX model",
"vpixl [model]* [prompt/reply to prompt]*": "Text to Image with PixArt model",
"vkxl [model]* [prompt/reply to prompt]*": "Text to Image with Kolors model",
"vdxl [model]* [prompt/reply to prompt]*": "Text to Image with SDXL model",
"vgif [prompt/reply to prompt]*": "Text to GIF",
}