Update app.py

app.py

@@ -128,8 +128,9 @@ async def on_disconnect():
 @client.event
 async def on_message(message):
     for user in message.mentions:
-        message.content = message.content.replace(f"<@{user.id}>",
-                                                  f"<@{str(user)}>")
+        if user.bot:
+            message.content = message.content.replace(f"<@{user.id}>",
+                                                  f"<@{str(user)}[bot]>")
     try:
         msgchannel = message.channel
         try:
@@ -138,7 +139,7 @@ async def on_message(message):
             guild_name = guild.name
         except:
             guild_name = "Direct"
-            msgchannel_name = message.author.display_name
+            msgchannel_name = str(message.author)
         s = f":green[{message.author}]: :violet[{message.content}] :blue[{msgchannel_name}] :orange[{guild_name}]"
         if message.author == client.user:
             return
@@ -151,7 +152,8 @@ async def on_message(message):
             If the user asks you to generate an image, use the start tag <|image|> and the end tag <|image|> around the prompt to generate it. Put this at the end of your response. Do not use a link for the image.
             For example, if the user asks you to generate an image of a cat, you could say <|image|>A cute cat with long fur that is looking out a window with curious eyes, hd, realistic<|image|>
             Always include a title.
-            Do not generate images unless the user speciifes that they want an image.""" 
+            Do not generate images unless the user speciifes that they want an image.
+            If a user has [bot] next to their username, they are a bot.""" 
         try:
             os.mkdir("data/" + guild_name)
         except:
@@ -159,14 +161,24 @@ async def on_message(message):
         if os.path.exists(f"data/{guild_name}/{msgchannel_name}"):
             with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
                 n = "\n"
-                f.write(
-                    f"""GPT4 Correct {message.author.name}: {message.content.strip(n)}<|end_of_turn|>"""
-                )
+                if message.author.bot:
+                    f.write(
+                        f"""GPT4 Correct {message.author}[bot]: {message.content.strip(n)}<|end_of_turn|>"""
+                    )
+                else:
+                    f.write(
+                        f"""GPT4 Correct {message.author}: {message.content.strip(n)}<|end_of_turn|>"""
+                    )
         else:
             with open(f"data/{guild_name}/{msgchannel_name}", "w") as f:
-                f.write(
-                    f"GPT4 Correct system: {sysp}<|end_of_turn|>GPT4 Correct {message.author}: {message.content}<|end_of_turn|>"
-                )
+                if message.author.bot:
+                        f.write(
+                            f"GPT4 Correct system: {sysp}<|end_of_turn|>GPT4 Correct {message.author}[bot]: {message.content}<|end_of_turn|>"
+                        )
+                    else:
+                        f.write(
+                            f"GPT4 Correct system: {sysp}<|end_of_turn|>GPT4 Correct {message.author}: {message.content}<|end_of_turn|>"
+                        )
         with open(f"data/{guild_name}/{msgchannel_name}", "r") as f:
             context = f.read()
         if "Feeway" in guild_name:
@@ -203,7 +215,7 @@ async def on_message(message):
                 e = await message.reply(embed=embed)
             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"""
-                image = await PT.text_to_image(imgp, negative_prompt=np)
+                image = await SDT.text_to_image(imgp, negative_prompt=np)
                 image.save("image.png")
                 embed.set_footer(
                     text=