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 Lyre, a discord bot. You were made by Araeyn.
+            """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.
@@ -205,43 +205,43 @@ 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]
+      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", "") != "":  
           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", "") != "":  
-              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)
+          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]
+        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)
   except Exception as e:
     embed = discord.Embed(title="Error", description=f"```{traceback.format_exc()}```")
     await message.channel.send(embed=embed)