From b5ed1be93adaed0fcd26d236da9100ad3967ff79 Mon Sep 17 00:00:00 2001 From: Rene Teigen Date: Sat, 14 Jan 2023 09:28:07 +0000 Subject: [PATCH] More redouser --- cogs/draw_image_generation.py | 4 ++-- cogs/image_prompt_optimizer.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/cogs/draw_image_generation.py b/cogs/draw_image_generation.py index aa2c629..9cbb590 100644 --- a/cogs/draw_image_generation.py +++ b/cogs/draw_image_generation.py @@ -123,7 +123,7 @@ class DrawDallEService(discord.Cog, name="DrawDallEService"): if from_context: result_message = await ctx.fetch_message(result_message.id) - redo_users[user_id] = RedoUser(prompt, ctx, ctx, result_message) + redo_users[user_id] = RedoUser(prompt=prompt, message=ctx, ctx=ctx, response=response_message, instruction=None,codex=False,paginator=None) else: if not vary: # Editing case @@ -177,7 +177,7 @@ class DrawDallEService(discord.Cog, name="DrawDallEService"): ) ) - redo_users[user_id] = RedoUser(prompt, ctx, ctx, result_message) + redo_users[user_id] = RedoUser(prompt=prompt, message=ctx, ctx=ctx, response=response_message, instruction=None,codex=False,paginator=None) self.converser_cog.users_to_interactions[user_id].append( response_message.id diff --git a/cogs/image_prompt_optimizer.py b/cogs/image_prompt_optimizer.py index a7891bf..1cdb5a2 100644 --- a/cogs/image_prompt_optimizer.py +++ b/cogs/image_prompt_optimizer.py @@ -114,7 +114,7 @@ class ImgPromptOptimizer(discord.Cog, name="ImgPromptOptimizer"): ) self.converser_cog.redo_users[user.id] = RedoUser( - prompt=final_prompt, message=ctx, ctx=ctx, response=response_message, instruction=None,codex=None,paginator=None + prompt=final_prompt, message=ctx, ctx=ctx, response=response_message, instruction=None,codex=False,paginator=None ) self.converser_cog.redo_users[user.id].add_interaction(response_message.id) await response_message.edit(