fix auto-expire retry and vary buttons

cogs/draw_image_generation.py

@@ -80,8 +80,8 @@ class DrawDallEService(commands.Cog, name="DrawDallEService"):
             result_message = await message.channel.send(
                 embed=embed,
                 file=file,
-                view=SaveView(image_urls, self, self.converser_cog),
             )
+            await result_message.edit(view=SaveView(image_urls, self, self.converser_cog, result_message))
 
             self.converser_cog.users_to_interactions[message.author.id] = []
             self.converser_cog.users_to_interactions[message.author.id].append(
@@ -112,8 +112,8 @@ class DrawDallEService(commands.Cog, name="DrawDallEService"):
                         content="I've drawn the optimized prompt!",
                         embed=embed,
                         file=file,
-                        view=SaveView(image_urls, self, self.converser_cog),
                     )
+                    await result_message.edit(view=SaveView(image_urls, self, self.converser_cog, result_message))
 
                     redo_users[user_id] = RedoUser(prompt, message, result_message)
 
@@ -204,14 +204,15 @@ class DrawDallEService(commands.Cog, name="DrawDallEService"):
 
 
 class SaveView(discord.ui.View):
-    def __init__(self, image_urls, cog, converser_cog, no_retry=False, only_save=None):
+    def __init__(self, image_urls, cog, converser_cog, message, no_retry=False, only_save=None):
         super().__init__(
-            timeout=600 if not only_save else None
+            timeout=10 if not only_save else None
         )  # 10 minute timeout for Retry, Save
         self.image_urls = image_urls
         self.cog = cog
         self.no_retry = no_retry
         self.converser_cog = converser_cog
+        self.message = message
         for x in range(1, len(image_urls) + 1):
             self.add_item(SaveButton(x, image_urls[x - 1]))
         if not only_save:
@@ -232,7 +233,7 @@ class SaveView(discord.ui.View):
 
         # Create a new view with the same params as this one, but pass only_save=True
         new_view = SaveView(
-            self.image_urls, self.cog, self.converser_cog, self.no_retry, only_save=True
+            self.image_urls, self.cog, self.converser_cog, self.message, self.no_retry, only_save=True
         )
 
         # Set the view of the message to the new view