Update app.py

app.py

@@ -135,7 +135,16 @@ async def on_message(message):
         s = f":green[{message.author}]: :violet[{message.content}] :blue[{msgchannel_name}] :orange[{guild_name}]"
         if message.author == client.user:
             return
-        sysp = """You are Lyre, a helpful discord chatbot. Respond in markdown. Your username is lyre#9828."""
+        sysp = """GPT4 Correct system: You are lr, a discord bot. You were made by Araeyn.
+            You must use the markdown format for your responses.
+            Do not excessively use bullet points.
+            Use emojis at the start of your responses.
+            Use <title> at the start of your title for the response and </title> at the end of the title.
+            Your title must summarize the user's 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 would say <image>A cute cat with long fur that is looking out a window with curious eyes.</image>
+            Always include a title.
+            Do not generate images unless the user speciifes that they want an image."""
         try:
             os.mkdir("data/" + guild_name)
         except:
@@ -156,22 +165,39 @@ async def on_message(message):
                 context = f.read()
         await syncMessages()
         if "Feeway" in guild_name:
-            context += f"GPT4 Correct Assistant:"
-            title = ec(context, "<title>", "</title>")[0]
-            output = LLM.text_generation(context,
-                                         stop_sequences=["<|end_of_turn|>"],
-                                         max_new_tokens=4096)
-            with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
-                f.write(f"GPT4 Correct Assistant: {output}<|end_of_turn|>")
-            embed = discord.Embed(title=title,
-                                  description=output.replace(
-                                      f"<title>{title}</title>", ""),
-                                  color=0x1E81B0)
-            embed.set_footer(
-                text=
-                """Information or code generated by Lyre may not always be correct. Lyre was made by Araeyn."""
-            )
+            async with msgchannel.typing():
+                context += f"GPT4 Correct Assistant:"
+                title = ec(context, "<title>", "</title>")[0]
+                imgp = ec(context)[0]
+                load = random.choice(
+                    [
+                        "https://cdn.dribbble.com/users/744913/screenshots/4094897/media/771a495231b798c0ccf7a59a19f31946.gif",
+                        "https://cdn.dribbble.com/users/563824/screenshots/3633228/media/b620ccb3ae8c14ea5447d159ebb1da58.gif",
+                        "https://cdn.dribbble.com/users/563824/screenshots/4155980/media/d3828cd14ed415eb6f90310991e06f27.gif",
+                        "https://cdn.dribbble.com/users/107759/screenshots/3498589/media/5bc45101de34a80ea71238a02f3a75b5.gif",
+                    ]
+                )
+                output = LLM.text_generation(context,
+                                             stop_sequences=["<|end_of_turn|>"],
+                                             max_new_tokens=4096)
+                with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
+                    f.write(f"GPT4 Correct Assistant: {output}<|end_of_turn|>")
+                embed = discord.Embed(title=title,
+                                      description=output.replace(
+                                          f"<title>{title}</title>", ""),
+                                      color=0x1E81B0)
+                embed.set_footer(
+                    text=
+                    """Information or code generated by Lyre may not always be correct. Lyre was made by Araeyn."""
+                )
+                embed.set_image(load)
             e = await message.reply(embed=embed)
+            image = SD.text_to_image(imgp)
+            image.save("image.png")
+            file = discord.File("image.png")
+            embed.set_image("attachment://image.png")
+            e.edit(embed=embed, file=file)
+            
     except Exception as e:
         print(Fore.RED)
         print(e)