Update app.py

app.py

@@ -160,7 +160,7 @@ async def on_message(message:discord.Message):
       os.mkdir(guild_name)
       with open(os.path.join(guild_name, f"{msgchannel_name}.txt"), "a") as f:
         f.write(
-            """GPT4 Correct system: You are lr, a discord bot. You were made by Araeyn.
+            """GPT4 Correct system: You are Lyre, 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.
@@ -205,39 +205,40 @@ async def on_message(message:discord.Message):
           y = query({"inputs": y[0]["generated_text"]})
           if y[0]["generated_text"] == py:
             break
-      r = r.split("<|end_of_turn|>")[0]
-      r = r.split("GPT4 Correct")[0]
-      embed.description = r
-      await e.edit(embed=embed)
-      with open(os.path.join(guild_name, f"{msgchannel_name}.txt"), "a") as f:
-        f.write("GPT4 Correct Assistant: ")
-        f.write(r)
-        f.write("<|end_of_turn|>")
-      if "<image>" in r and "</image>" in r:
-        i = extract_content(r)[0]
-        r = r.replace("<image>" + i + "</image>", "")
-        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"])
-        if r.replace("\n", "") != "":  
+          r = r.split("<|end_of_turn|>")[0]
+          r = r.split("GPT4 Correct")[0]
           embed.description = r
           await e.edit(embed=embed)
-          embed.set_image(url=load)
-          await e.edit(embed=embed)
-        else:
-          embed.clear_fields()
-          embed.set_image(url=load)
-          await e.edit(embed=embed)
-        image_bytes = generate(i)
-        try: image = Image.open(io.BytesIO(image_bytes))
-        except: print(image_bytes)
-        image.save(f"latest.png")
-        prompt = "Make it look better."
-        url = "latest.png"
-        init_image = load_image(url).convert("RGB")
-        image = pipe(prompt, image=init_image).images[0]
-        image.save("latest.png")
-        embed.set_image(url="attachment://latest.png")
-        await e.edit(embed=embed, attachments=[discord.File(fp=f"latest.png")])
-        os.system("rm " + f"latest.png")
+          with open(os.path.join(guild_name, f"{msgchannel_name}.txt"), "a") as f:
+            f.write("GPT4 Correct Assistant: ")
+            f.write(r)
+            f.write("<|end_of_turn|>")
+          if "<image>" in r and "</image>" in r:
+            i = extract_content(r)[0]
+            r = r.replace("<image>" + i + "</image>", "")
+            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"])
+            if r.replace("\n", "") != "":  
+              embed.description = r
+              await e.edit(embed=embed)
+              embed.set_image(url=load)
+              await e.edit(embed=embed)
+            else:
+              embed.clear_fields()
+              embed.set_image(url=load)
+              await e.edit(embed=embed)
+            image_bytes = generate(i)
+            try: image = Image.open(io.BytesIO(image_bytes))
+            except: print(image_bytes)
+            image.save(f"latest.png")
+            # prompt = "Make it look better."
+            url = "latest.png"
+            init_image = load_image(url).convert("RGB")
+            image = pipe("", image=init_image).images[0]
+            print(image)
+            image.save("latest.png")
+            embed.set_image(url="attachment://latest.png")
+            await e.edit(embed=embed, attachments=[discord.File(fp=f"latest.png")])
+            os.system("rm " + f"latest.png")
       else:
         embed.description = r
         await e.edit(embed=embed)