Update app.py

app.py

@@ -10,6 +10,7 @@ import time
 import discord.utils
 from discord.ext import tasks
 from PIL import Image
+import requests
 import streamlit as st
 from huggingface_hub import AsyncInferenceClient, login
 import time
@@ -427,7 +428,10 @@ async def on_message(message):
             context = f.read()
         with open(f"{guild_name}.guild", "r") as f:
             o = f.read()
-        if str(message.channel.id) in o.split("\n") or guild_name == "Direct":
+        info = requests.get("https://raw.githubusercontent.com/aryananumula/lr/main/info.json")
+        bannedUsers = json.loads(info)["bannedUsers"]
+        imageModel = json.loads(info)["imageModel"]
+        if str(message.channel.id) in o.split("\n") or guild_name == "Direct" or message.author.id not in bannedUsers:
             async with msgchannel.typing():
                 context += f"GPT4 Correct Assistant:"
                 load = random.choice(
@@ -462,11 +466,18 @@ async def on_message(message):
             if imgp != "":
                 np = """lowres, text, error, cropped, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, out of frame, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck, username, watermark, signature"""
                 st = time.time()
-                # image, m = (await SD.text_to_image(imgp, negative_prompt=np), "Stable Diffusion 2-1")
-                # image, m = (await SDXL.text_to_image(imgp, negative_prompt=np, num_inference_steps=8), "stable-diffusion-xl-base-1.0")
-                image, m = (await SDXLT.text_to_image(imgp, negative_prompt=np, num_inference_steps=1, guidance_scale=0.0), "sdxl-turbo")
-                # image, m = (await SDT.text_to_image(imgp, negative_prompt=np, num_inference_steps=1, guidance_scale=0.0), "sd-turbo")
-                # image, m = (await PT.text_to_image(imgp, negative_prompt=np), "Proteus v0.2")
+                if imageModel == "Stable Diffusion 2-1":
+                    image, m = (await SD.text_to_image(imgp, negative_prompt=np), "Stable Diffusion 2-1")
+                elif imageModel == "stable-diffusion-xl-base-1.0":
+                    image, m = (await SDXL.text_to_image(imgp, negative_prompt=np, num_inference_steps=8), "stable-diffusion-xl-base-1.0")
+                elif imageModel == "sdxl-turbo":
+                    image, m = (await SDXLT.text_to_image(imgp, negative_prompt=np, num_inference_steps=1, guidance_scale=0.0), "sdxl-turbo")
+                elif imageModel == "sd-turbo":
+                    image, m = (await SDT.text_to_image(imgp, negative_prompt=np, num_inference_steps=1, guidance_scale=0.0), "sd-turbo")
+                elif imageModel == "Proteus v0.2":
+                    image, m = (await PT.text_to_image(imgp, negative_prompt=np), "Proteus v0.2")
+                else:
+                    raise NotImplementedError(f"Model {imageModel} not found. Report this to @araeyn if this keeps happening.")
                 image.save("image.png")
                 file = discord.File("image.png", filename="image.png", description=imgp)
                 embed.set_image(url="attachment://image.png")