mirror of
https://github.com/5shekel/stable-diffusion-telegram-bot.git
synced 2024-05-22 19:33:14 +03:00
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
from pyrogram import Client, filters
|
|
from pyrogram.types import *
|
|
import os
|
|
import json
|
|
import requests
|
|
import io
|
|
import random
|
|
from PIL import Image, PngImagePlugin
|
|
import base64
|
|
|
|
API_ID = os.environ.get("API_ID", None)
|
|
API_HASH = os.environ.get("API_HASH", None)
|
|
TOKEN = os.environ.get("TOKEN", None)
|
|
SD_URL = os.environ.get("TOKEN", None)
|
|
|
|
bot = Client(
|
|
"stable",
|
|
api_id=API_ID,
|
|
api_hash=API_HASH,
|
|
bot_token=TOKEN
|
|
)
|
|
|
|
@app.on_message(filters.command(["draw"]))
|
|
def draw(client, message):
|
|
msgs = message.text.split(' ', 1)
|
|
if len(msgs) == 1:
|
|
message.reply_text("Format : /draw < text to anime image >")
|
|
return
|
|
msg = msgs[1]
|
|
|
|
K = message.reply_text("Please Wait 10-15 Second")
|
|
|
|
payload = {"prompt": msg}
|
|
|
|
r = requests.post(url=f'{SD_URL}/sdapi/v1/txt2img', json=payload).json()
|
|
|
|
chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
chars1 = "1234564890"
|
|
gen1 = random.choice(chars)
|
|
gen2 = random.choice(chars)
|
|
gen3 = random.choice(chars1)
|
|
gen4 = random.choice(chars)
|
|
gen5 = random.choice(chars)
|
|
gen6 = random.choice(chars)
|
|
gen7 = random.choice(chars1)
|
|
gen8 = random.choice(chars)
|
|
gen9 = random.choice(chars)
|
|
gen10 = random.choice(chars1)
|
|
word = f"{message.from_user.id}-MOE{gen1}{gen2}{gen3}{gen4}{gen5}{gen6}{gen7}{gen8}{gen9}{gen10}"
|
|
|
|
for i in r['images']:
|
|
image = Image.open(io.BytesIO(base64.b64decode(i.split(",", 1)[0])))
|
|
|
|
png_payload = {"image": "data:image/png;base64," + i}
|
|
response2 = requests.post(url=f'{url}/sdapi/v1/png-info',
|
|
json=png_payload)
|
|
|
|
pnginfo = PngImagePlugin.PngInfo()
|
|
pnginfo.add_text("parameters", response2.json().get("info"))
|
|
image.save(f'{word}.png', pnginfo=pnginfo)
|
|
|
|
message.reply_photo(
|
|
photo=f"{word}.png",
|
|
caption=
|
|
f"Prompt - **{msg}**\n **[{message.from_user.first_name}-Kun](tg://user?id={message.from_user.id})**\n Join @WaifuAiSupport"
|
|
)
|
|
os.remove(f"{word}.png")
|
|
K.delete()
|
|
|
|
|
|
@app.on_message(filters.command(["start"], prefixes=["/", "!"]))
|
|
async def start(client, message):
|
|
Photo = "https://media.discordapp.net/attachments/1028156834944655380/1062018608022171788/3aac7aaf-0065-40aa-9e4d-430c717b3d87.jpg"
|
|
|
|
buttons = [[
|
|
InlineKeyboardButton("Add to your group",
|
|
url="http://t.me/botname?startgroup=true"),
|
|
InlineKeyboardButton("Channel", url="https://t.me/otakatsu"),
|
|
InlineKeyboardButton("Support", url="https://t.me/otakatsu_chat")
|
|
]]
|
|
await message.reply_photo(
|
|
photo=Photo,
|
|
caption=
|
|
f"Hello! I'm botname Ai and I can make an anime-styled picture!\n\n/generate - Reply to Image\n/draw text to anime image\n\nPowered by @Otakatsu",
|
|
reply_markup=InlineKeyboardMarkup(buttons))
|
|
|
|
|
|
app.run()
|