Spaces:
Paused
Paused
File size: 2,791 Bytes
fefc4fb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import aiohttp
import json
import os
from os import system, remove
from . import ultroid_cmd, check_filename, fast_download
async def paal_image(prompt):
url = "https://bot-management-4tozrh7z2a-ue.a.run.app/chat/image"
headers = {
"content-type": "application/json",
"sec-ch-ua": '"Not-A.Brand";v="99", "Chromium";v="124"',
"accept": "application/json, text/plain, */*",
"sec-ch-ua-mobile": "?1",
"user-agent": "Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Mobile Safari/537.36",
"sec-ch-ua-platform": '"Android"',
"origin": "https://app.paal.ai",
"sec-fetch-site": "cross-site",
"sec-fetch-mode": "cors",
"sec-fetch-dest": "empty",
"referer": "https://app.paal.ai/",
"accept-encoding": "gzip, deflate, br",
"accept-language": "en-GB,en-US;q=0.9,en;q=0.8",
}
payload = {"prompt": prompt, "bid": "edwo6pg1", "history": []}
async with aiohttp.ClientSession() as session:
async with session.post(url, headers=headers, data=json.dumps(payload)) as response:
data = await response.json()
if "url" not in data:
raise Exception("Image URL not found in the response.")
image_url = data["url"]
async with session.get(image_url) as image_response:
if image_response.status != 200:
raise Exception(f"Failed to retrieve image from URL: {image_response.status}")
image_content = await image_response.read()
image_path = check_filename("generated_image.png")
with open(image_path, "wb") as image_file:
image_file.write(image_content)
return image_path
@ultroid_cmd(pattern="dall( (.*)|$)")
async def generate_image(event):
args = event.pattern_match.group(2)
if not args and event.is_reply:
replied_message = await event.get_reply_message()
args = replied_message.text.strip()
if not args:
return await event.eor("Please provide a prompt for generating the image.")
moi = await event.eor("Generating image...")
try:
# Await the image creation
image_path = await paal_image(args.strip())
if len(f"Prompt: {args}") <= 1024:
await event.client.send_file(
event.chat_id,
image_path,
caption=f"Prompt: {args}",
reply_to=event.reply_to_msg_id,
)
else:
await event.client.send_file(
event.chat_id,
image_path,
reply_to=event.reply_to_msg_id,
)
remove(image_path)
await moi.delete()
except Exception as exc:
await moi.edit(f"Error: {exc}")
|