button test

app.py

@@ -1,5 +1,7 @@
 import discord
 from discord.ui import Button
+from discord.ext import commands #for buttons
+
 import gradio_client
 from gradio_client import Client
 import gradio as gr
@@ -49,26 +51,30 @@ class ButtonView(discord.ui.View):
         for child in self.children:
             child.disabled = True
         self.stop()
+    #-------------------------------------------
+    async def invoke_dfif2(self, image_path):
+        ctx = await self.ctx.bot.get_context(self.ctx)
+        await ctx.invoke(ctx.bot.get_command('dfif2'), image_path=image_path, stage_1_result_path=self.stage_1_result_path)        
 
     @discord.ui.button(label='Image 1', style=discord.ButtonStyle.blurple)
     async def image1_button(self, button: discord.ui.Button, interaction: discord.Interaction):
-        await self.ctx.invoke(self.ctx.bot.get_command('dfif2'), image_path=self.image_paths[0], stage_1_result_path=self.stage_1_result_path)
-        self.stop()
+        await self.invoke_dfif2(self.image_paths[0])
+        self.stop()         
 
     @discord.ui.button(label='Image 2', style=discord.ButtonStyle.blurple)
     async def image2_button(self, button: discord.ui.Button, interaction: discord.Interaction):
-        await self.ctx.invoke(self.ctx.bot.get_command('dfif2'), image_path=self.image_paths[1], stage_1_result_path=self.stage_1_result_path)
-        self.stop()
+        await self.invoke_dfif2(self.image_paths[1])
+        self.stop()  
 
     @discord.ui.button(label='Image 3', style=discord.ButtonStyle.blurple)
     async def image3_button(self, button: discord.ui.Button, interaction: discord.Interaction):
-        await self.ctx.invoke(self.ctx.bot.get_command('dfif2'), image_path=self.image_paths[2], stage_1_result_path=self.stage_1_result_path)
-        self.stop()
+        await self.invoke_dfif2(self.image_paths[2])
+        self.stop()  
 
     @discord.ui.button(label='Image 4', style=discord.ButtonStyle.blurple)
     async def image4_button(self, button: discord.ui.Button, interaction: discord.Interaction):
-        await self.ctx.invoke(self.ctx.bot.get_command('dfif2'), image_path=self.image_paths[3], stage_1_result_path=self.stage_1_result_path)
-        self.stop()
+        await self.invoke_dfif2(self.image_paths[3])
+        self.stop()  
 #---------------------------------------------------------------------------------------------------------------------------------------------------