diff --git a/cogs/prompt_optimizer_cog.py b/cogs/prompt_optimizer_cog.py index 28e1613e..df479177 100644 --- a/cogs/prompt_optimizer_cog.py +++ b/cogs/prompt_optimizer_cog.py @@ -182,7 +182,11 @@ class DrawButton(discord.ui.Button["OptimizeView"]): def __init__( self, converser_cog, image_service_cog, deletion_queue, custom_api_key ): - super().__init__(style=discord.ButtonStyle.green, label="Draw", custom_id="draw_button_optimizer") + super().__init__( + style=discord.ButtonStyle.green, + label="Draw", + custom_id="draw_button_optimizer", + ) self.converser_cog = converser_cog self.image_service_cog = image_service_cog self.deletion_queue = deletion_queue @@ -237,7 +241,11 @@ class RedoButton(discord.ui.Button["OptimizeView"]): def __init__( self, converser_cog, image_service_cog, deletion_queue, custom_api_key=None ): - super().__init__(style=discord.ButtonStyle.danger, label="Retry", custom_id="redo_button_optimizer") + super().__init__( + style=discord.ButtonStyle.danger, + label="Retry", + custom_id="redo_button_optimizer", + ) self.converser_cog = converser_cog self.image_service_cog = image_service_cog self.deletion_queue = deletion_queue diff --git a/services/image_service.py b/services/image_service.py index 1c517a6e..0089e7f8 100644 --- a/services/image_service.py +++ b/services/image_service.py @@ -350,7 +350,11 @@ async def callback(self, interaction: discord.Interaction): class RedoButton(discord.ui.Button["SaveView"]): def __init__(self, cog, converser_cog, custom_api_key): - super().__init__(style=discord.ButtonStyle.danger, label="Retry", custom_id="redo_button_draw_main") + super().__init__( + style=discord.ButtonStyle.danger, + label="Retry", + custom_id="redo_button_draw_main", + ) self.cog = cog self.converser_cog = converser_cog self.custom_api_key = custom_api_key diff --git a/services/moderations_service.py b/services/moderations_service.py index c07dbbff..173c35e5 100644 --- a/services/moderations_service.py +++ b/services/moderations_service.py @@ -323,7 +323,9 @@ def __init__( class ApproveMessageButton(discord.ui.Button["ModerationAdminView"]): def __init__(self, message, moderation_message, current_num): - super().__init__(style=discord.ButtonStyle.green, label="Approve", custom_id="approve_button") + super().__init__( + style=discord.ButtonStyle.green, label="Approve", custom_id="approve_button" + ) self.message = message self.moderation_message = moderation_message self.current_num = current_num @@ -336,7 +338,11 @@ async def callback(self, interaction: discord.Interaction): class DeleteMessageButton(discord.ui.Button["ModerationAdminView"]): def __init__(self, message, moderation_message, current_num): - super().__init__(style=discord.ButtonStyle.danger, label="Delete Message", custom_id="delete_button") + super().__init__( + style=discord.ButtonStyle.danger, + label="Delete Message", + custom_id="delete_button", + ) self.message = message self.moderation_message = moderation_message self.current_num = current_num @@ -361,7 +367,9 @@ async def callback(self, interaction: discord.Interaction): class KickUserButton(discord.ui.Button["ModerationAdminView"]): def __init__(self, message, moderation_message, current_num): - super().__init__(style=discord.ButtonStyle.danger, label="Kick User", custom_id="kick_button") + super().__init__( + style=discord.ButtonStyle.danger, label="Kick User", custom_id="kick_button" + ) self.message = message self.moderation_message = moderation_message self.current_num = current_num @@ -400,7 +408,11 @@ class TimeoutUserButton(discord.ui.Button["ModerationAdminView"]): def __init__( self, message, moderation_message, current_num, hours, nodelete, source_deleted ): - super().__init__(style=discord.ButtonStyle.danger, label=f"Timeout {hours}h", custom_id="timeout_button") + super().__init__( + style=discord.ButtonStyle.danger, + label=f"Timeout {hours}h", + custom_id="timeout_button", + ) self.message = message self.moderation_message = moderation_message self.hours = hours