From c2ed2f4173b86a4613473c1f6c3437b5a9b5187e Mon Sep 17 00:00:00 2001
From: Pranav Ajay <86167689+MoeZilla@users.noreply.github.com>
Date: Thu, 12 Jan 2023 10:07:56 +0530
Subject: [PATCH] Create main.py

---
 main.py | 88 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..c6ac902
--- /dev/null
+++ b/main.py
@@ -0,0 +1,88 @@
+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()