diff --git a/cogs/imagine_cog.py b/cogs/imagine_cog.py index 5acb384..4b0763f 100644 --- a/cogs/imagine_cog.py +++ b/cogs/imagine_cog.py @@ -224,7 +224,7 @@ async def model_autocomplete(self, @app_commands.guild_only() @app_commands.checks.cooldown(1, 15) @app_commands.describe(prompt="Imagine a prompt", height="Height of the image", width="Width of the image", negative="The things not to include in the image", cached="Removes the image seed", nologo="Remove the logo", enhance="Disables Prompt enhancing if set to False", private="Only you can see the generated Image if set to True") - async def imagine_command(self, interaction, prompt:str, model: str = "Dreamshaper", width:int = 1000, height:int = 1000, negative:str|None = None, cached:bool = False, nologo:bool = False, enhance:bool = True, private:bool = False): + async def imagine_command(self, interaction, prompt:str, model: str = "Swizz8", width:int = 1000, height:int = 1000, negative:str|None = None, cached:bool = False, nologo:bool = False, enhance:bool = True, private:bool = False): await interaction.response.send_message(embed=discord.Embed(title="Generating Image", description="Please wait while we generate your image", color=discord.Color.blurple()), ephemeral=True) if len(prompt) > 1500: diff --git a/cogs/multi_imagine_cog.py b/cogs/multi_imagine_cog.py index 1371bb6..e339c37 100644 --- a/cogs/multi_imagine_cog.py +++ b/cogs/multi_imagine_cog.py @@ -89,33 +89,33 @@ async def regerate_4(self, interaction: discord.Interaction, button: discord.ui. await Multi_imagine.regenerate(interaction, button, data, 3) - @discord.ui.button(label="V5", style=discord.ButtonStyle.secondary, custom_id="v5") - async def regerate_5(self, interaction: discord.Interaction, button: discord.ui.Button): - await interaction.response.send_message(embed=discord.Embed(title="Regenerating Image with model 5", description="Please wait while we regenerate your image", color=discord.Color.blurple()), ephemeral=True) - data = get_multi_imagined_prompt_data(interaction.message.id) + # @discord.ui.button(label="V5", style=discord.ButtonStyle.secondary, custom_id="v5") + # async def regerate_5(self, interaction: discord.Interaction, button: discord.ui.Button): + # await interaction.response.send_message(embed=discord.Embed(title="Regenerating Image with model 5", description="Please wait while we regenerate your image", color=discord.Color.blurple()), ephemeral=True) + # data = get_multi_imagined_prompt_data(interaction.message.id) - await Multi_imagine.regenerate(interaction, button, data, 4) + # await Multi_imagine.regenerate(interaction, button, data, 4) - @discord.ui.button(label="V6", style=discord.ButtonStyle.secondary, custom_id="v6") - async def regerate_6(self, interaction: discord.Interaction, button: discord.ui.Button): - await interaction.response.send_message(embed=discord.Embed(title="Regenerating Image with model 6", description="Please wait while we regenerate your image", color=discord.Color.blurple()), ephemeral=True) - data = get_multi_imagined_prompt_data(interaction.message.id) + # @discord.ui.button(label="V6", style=discord.ButtonStyle.secondary, custom_id="v6") + # async def regerate_6(self, interaction: discord.Interaction, button: discord.ui.Button): + # await interaction.response.send_message(embed=discord.Embed(title="Regenerating Image with model 6", description="Please wait while we regenerate your image", color=discord.Color.blurple()), ephemeral=True) + # data = get_multi_imagined_prompt_data(interaction.message.id) - await Multi_imagine.regenerate(interaction, button, data, 5) + # await Multi_imagine.regenerate(interaction, button, data, 5) - @discord.ui.button(label="V7", style=discord.ButtonStyle.secondary, custom_id="v7") - async def regerate_7(self, interaction: discord.Interaction, button: discord.ui.Button): - await interaction.response.send_message(embed=discord.Embed(title="Regenerating Image with model 7", description="Please wait while we regenerate your image", color=discord.Color.blurple()), ephemeral=True) - data = get_multi_imagined_prompt_data(interaction.message.id) + # @discord.ui.button(label="V7", style=discord.ButtonStyle.secondary, custom_id="v7") + # async def regerate_7(self, interaction: discord.Interaction, button: discord.ui.Button): + # await interaction.response.send_message(embed=discord.Embed(title="Regenerating Image with model 7", description="Please wait while we regenerate your image", color=discord.Color.blurple()), ephemeral=True) + # data = get_multi_imagined_prompt_data(interaction.message.id) - await Multi_imagine.regenerate(interaction, button, data, 6) + # await Multi_imagine.regenerate(interaction, button, data, 6) - @discord.ui.button(label="V8", style=discord.ButtonStyle.secondary, custom_id="v8") - async def regerate_8(self, interaction: discord.Interaction, button: discord.ui.Button): - await interaction.response.send_message(embed=discord.Embed(title="Regenerating Image with model 8", description="Please wait while we regenerate your image", color=discord.Color.blurple()), ephemeral=True) - data = get_multi_imagined_prompt_data(interaction.message.id) + # @discord.ui.button(label="V8", style=discord.ButtonStyle.secondary, custom_id="v8") + # async def regerate_8(self, interaction: discord.Interaction, button: discord.ui.Button): + # await interaction.response.send_message(embed=discord.Embed(title="Regenerating Image with model 8", description="Please wait while we regenerate your image", color=discord.Color.blurple()), ephemeral=True) + # data = get_multi_imagined_prompt_data(interaction.message.id) - await Multi_imagine.regenerate(interaction, button, data, 7) + # await Multi_imagine.regenerate(interaction, button, data, 7) @discord.ui.button(style=discord.ButtonStyle.red, custom_id="multiimagine_delete", emoji="<:delete:1187102382312652800>") diff --git a/constants.py b/constants.py index 320444d..6b55d9b 100644 --- a/constants.py +++ b/constants.py @@ -7,14 +7,10 @@ MONGODB_URI = os.environ["MONGODB_URI"] APP_URI = os.environ["APP_URI"] MODELS = [ - "Deliberate", - "Playground", - "Pixart", - "Dreamshaper", - "Turbo", - "Formulaxl", - "Dpo", - "Realvis", + "swizz8", + "dreamshaper", + "deliberate", + "juggernaut", ] with open("nsfw.txt", "r") as r: