format

main.py

@@ -9,7 +9,7 @@ from pyrogram import Client, filters
 from pyrogram.types import *
 from dotenv import load_dotenv
 
-# Done! Congratulations on your new bot. You will find it at 
+# Done! Congratulations on your new bot. You will find it at
 # t.me/gootmornbot
 # You can now add a description, about section and profile picture for your bot, see /help for a list of commands. By the way, when you've finished creating your cool bot, ping our Bot Support if you want a better username for it. Just make sure the bot is fully operational before you do this.
 
@@ -19,53 +19,45 @@ from dotenv import load_dotenv
 # For a description of the Bot API, see this page: https://core.telegram.org/bots/api
 
 load_dotenv()
-API_ID = os.environ.get("API_ID", None) 
+API_ID = os.environ.get("API_ID", None)
-API_HASH = os.environ.get("API_HASH", None) 
+API_HASH = os.environ.get("API_HASH", None)
-TOKEN = os.environ.get("TOKEN", None) 
+TOKEN = os.environ.get("TOKEN", None)
-SD_URL = os.environ.get("SD_URL", None) 
+SD_URL = os.environ.get("SD_URL", None)
 print(SD_URL)
-app = Client(
+app = Client("stable", api_id=API_ID, api_hash=API_HASH, bot_token=TOKEN)
-    "stable",
+
-    api_id=API_ID,
+
-    api_hash=API_HASH,
+def slice_positive_negative(string):
-    bot_token=TOKEN
+    delimiter = "ng"
-)
+    if delimiter in string:
+        index = string.index(delimiter)
+        positive = string[:index].rstrip()
+        negative = string[index + len(delimiter) :].lstrip()
+        return positive, negative
+    else:
+        return string, "null"  # if we are missing the ng bad thing happen....
 
 
 @app.on_message(filters.command(["draw"]))
 def draw(client, message):
-    msgs = message.text.split(' ', 1)
+    msgs = message.text.split(" ", 1)
     if len(msgs) == 1:
-        message.reply_text("Format : /draw < text to image >")
+        message.reply_text(
+            "Format :\n/draw < text to image >\nng < negative (optional) >"
+        )
         return
-    msg = msgs[1]
 
-    K = message.reply_text("Please Wait 10-15 Second")
+    msg = slice_positive_negative(msgs[1])
-
+    print(msg)
-    prompt = msg
+    payload = {
-    negative_prompt = "lowres, low resolution, bad resolution, bad, poor quality, bad quality, blurry, bad art, incomplete, logo, signature, text, jpeg artifacts, compression artifacts,child, childish"
+        "prompt": msg[0],
-
+        "negative_prompt": msg[1],
-    # Look for negative prompt in brackets after "NG" in the message
+    }
-    ng_index = msg.find("ng" or "NG")
-    if ng_index != -1:
-        bracket_start_index = msg.find("[", ng_index)
-        bracket_end_index = msg.find("]", ng_index)
-        if bracket_start_index != -1 and bracket_end_index != -1:
-            new_negative_prompt = msg[bracket_start_index+1:bracket_end_index]
-            if negative_prompt == "":
-                negative_prompt = new_negative_prompt
-            else:
-                negative_prompt = f"{negative_prompt}, {new_negative_prompt}"
-            prompt = msg[:ng_index] + msg[bracket_end_index+1:]
-        else:
-            prompt = msg[:ng_index]
-
-    payload = {"prompt": prompt}
-    if negative_prompt != "":
-        payload["negative_prompt"] = negative_prompt
 
     print(payload)
-    r = requests.post(url=f'{SD_URL}/sdapi/v1/txt2img', json=payload).json()
+
+    K = message.reply_text("Please Wait 10-15 Second")
+    r = requests.post(url=f"{SD_URL}/sdapi/v1/txt2img", json=payload).json()
 
     def genr():
         chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
@@ -81,44 +73,48 @@ def draw(client, message):
         gen9 = random.choice(chars)
         gen10 = random.choice(chars1)
         return f"{message.from_user.id}-MOE{gen1}{gen2}{gen3}{gen4}{gen5}{gen6}{gen7}{gen8}{gen9}{gen10}"
-    
+
     word = genr()
-    for i in r['images']:
+    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'{SD_URL}/sdapi/v1/png-info',
+        response2 = requests.post(url=f"{SD_URL}/sdapi/v1/png-info", json=png_payload)
-                                  json=png_payload)
 
         pnginfo = PngImagePlugin.PngInfo()
         pnginfo.add_text("parameters", response2.json().get("info"))
-        image.save(f'{word}.png', pnginfo=pnginfo)
+        image.save(f"{word}.png", pnginfo=pnginfo)
 
         message.reply_photo(
             photo=f"{word}.png",
-            caption=
+            caption=f"Prompt - **{msg[0]}**\nNegative Prompt - **{msg[1]}**\n**[{message.from_user.first_name}-Kun](tg://user?id={message.from_user.id})**",
-            f"Prompt - **{msg}**\n **[{message.from_user.first_name}-Kun](tg://user?id={message.from_user.id})**"
         )
         # os.remove(f"{word}.png")
         K.delete()
 
 
-
 @app.on_message(filters.command(["getmodels"]))
 async def get_models(client, message):
-    response = requests.get(url=f'{SD_URL}/sdapi/v1/sd-models')
+    response = requests.get(url=f"{SD_URL}/sdapi/v1/sd-models")
     if response.status_code == 200:
         models_json = response.json()
         # create buttons for each model name
         buttons = []
         for model in models_json:
-            buttons.append([InlineKeyboardButton(model['title'], callback_data=model['model_name'])])
+            buttons.append(
+                [
+                    InlineKeyboardButton(
+                        model["title"], callback_data=model["model_name"]
+                    )
+                ]
+            )
         # send the message
         await message.reply_text(
             text="Select a model [checkpoint] to use",
-            reply_markup=InlineKeyboardMarkup(buttons)
+            reply_markup=InlineKeyboardMarkup(buttons),
         )
 
+
 @app.on_callback_query()
 async def process_callback(client, callback_query):
     # if a model button is clicked, set sd_model_checkpoint to the selected model's title
@@ -127,36 +123,36 @@ async def process_callback(client, callback_query):
     # The sd_model_checkpoint needs to be set to the title from /sdapi/v1/sd-models
     # post using /sdapi/v1/options
 
-    options = {
+    options = {"sd_model_checkpoint": sd_model_checkpoint}
-        "sd_model_checkpoint": sd_model_checkpoint
-    }
 
     # post the options
-    response = requests.post(url=f'{SD_URL}/sdapi/v1/options', json=options)
+    response = requests.post(url=f"{SD_URL}/sdapi/v1/options", json=options)
     if response.status_code == 200:
         # if the post was successful, send a message
-        await callback_query.message.reply_text("checpoint set to " + sd_model_checkpoint)
+        await callback_query.message.reply_text(
+            "checpoint set to " + sd_model_checkpoint
+        )
     else:
         # if the post was unsuccessful, send an error message
         await callback_query.message.reply_text("Error setting options")
 
 
-
 @app.on_message(filters.command(["start"], prefixes=["/", "!"]))
 async def start(client, message):
     # Photo = "https://i.imgur.com/79hHVX6.png"
 
-    buttons = [[
+    buttons = [
-        InlineKeyboardButton("Add to your group",
+        [
-                             url="https://t.me/gootmornbot?startgroup=true")
+            InlineKeyboardButton(
-    ]]
+                "Add to your group", url="https://t.me/gootmornbot?startgroup=true"
+            )
+        ]
+    ]
     await message.reply_text(
-
         # photo=Photo,
-        text=
+        text=f"Hello!\nask me to imagine anything\n\n/draw text to image",
-        f"Hello!\nask me to imagine anything\n\n/draw text to image",
+        reply_markup=InlineKeyboardMarkup(buttons),
-        reply_markup=InlineKeyboardMarkup(buttons)
+    )
-        )
 
 
 app.run()