Update app.py

app.py

@@ -25,7 +25,7 @@ def syncMessages():
         hours, remainder = divmod(int(delta_uptime.total_seconds()), 3600)
         minutes, seconds = divmod(remainder, 60)
         days, hours = divmod(hours, 24)
-        st.markdown(f":violet[Online Time]: :green[{days:02d}]d | :green[{hours:02d}]h | :green[{minutes:02d}]m | :green[{seconds:02d}]s")
+        st.markdown(f"`[Online Time]: [{days:02d}]d | [{hours:02d}]h | [{minutes:02d}]m | [{seconds:02d}]s`")
         dirs = st.tabs(os.listdir("data"))
         i = -1
         for dir in os.listdir("data"):
@@ -71,7 +71,7 @@ HF_TOKEN = os.environ["HF_TOKEN"]
 
 login(HF_TOKEN)
 
-SD = InferenceClient(model="stabilityai/stable-diffusion-2-1")
+SD = InferenceClient(model="dataautogpt3/ProteusV0.2")
 LLM = InferenceClient(model="openchat/openchat-3.5-0106")
 RF = InferenceClient(model="stabilityai/stable-diffusion-xl-refiner-1.0")
 
@@ -147,8 +147,10 @@ async def on_message(message):
             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|>
+            Use the keywords 'best quality, HD, ~*~aesthetic~*~' in your prompt to improve it.
             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.
+            Talk like the user does,""" 
         try:
             os.mkdir("data/" + guild_name)
         except:
@@ -201,26 +203,28 @@ async def on_message(message):
                     )
                     
                 e = await message.reply(embed=embed)
-            print(imgp)
             if imgp != "":
-                image = SD.text_to_image(imgp)
-                image.save("image.png")
-                embed.set_footer(
-                    text=
-                    """Refining..."""
-                )
-                file = discord.File("image.png", filename="image.png")
-                embed.set_image(url="attachment://image.png")
-                await e.edit(embed=embed, attachments=[file])
-                image = RF.image_to_image("image.png")
-                embed.set_footer(
-                    text=
-                    """Information or code generated by Lyre may not always be correct. Lyre was made by Araeyn."""
-                )
-                image.save("image.png")
-                file = discord.File("image.png", filename="image.png")
-                embed.set_image(url="attachment://image.png")
-                await e.edit(embed=embed, attachments=[file])
+                try:
+                    image = SD.text_to_image(imgp)
+                    image.save("image.png")
+                    embed.set_footer(
+                        text=
+                        """Refining..."""
+                    )
+                    file = discord.File("image.png", filename="image.png")
+                    embed.set_image(url="attachment://image.png")
+                    await e.edit(embed=embed, attachments=[file])
+                    image = RF.image_to_image("image.png")
+                    embed.set_footer(
+                        text=
+                        """Information or code generated by Lyre may not always be correct. Lyre was made by Araeyn."""
+                    )
+                    image.save("image.png")
+                    file = discord.File("image.png", filename="image.png")
+                    embed.set_image(url="attachment://image.png")
+                    await e.edit(embed=embed, attachments=[file])
+                except:
+                    print(image)
             
     except Exception as e:
         print(e)