test: keep button functional across bot restarts

app.py

@@ -14,6 +14,8 @@ intents = discord.Intents.all()
 bot = commands.Bot(command_prefix="!", intents=intents)
 GRADIO_APP_URL = "https://huggingface.co/spaces/lunarflu/gradio-oauth2"
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
+BUTTON_CHANNEL_ID = "1106995001155649680"
+BUTTON_MESSAGE_ID = "1266438263531634708"
 
 # Dictionary to store user IDs and their corresponding unique strings
 user_tokens = {}
@@ -22,6 +24,20 @@ user_tokens = {}
 async def on_ready():
     print(f'Logged in as {bot.user}')
 
+    if BUTTON_MESSAGE_ID:
+        channel = bot.get_channel(BUTTON_CHANNEL_ID)
+        if channel:
+            try:
+                message = await channel.fetch_message(int(BUTTON_MESSAGE_ID))
+                if message:
+                    view = ButtonView()
+                    await message.edit(view=view)
+            except discord.NotFound:
+                print(f"Message with ID {KNOWN_MESSAGE_ID} not found.")
+            except discord.HTTPException as e:
+                print(f"Failed to fetch message with ID {KNOWN_MESSAGE_ID}: {e}")
+                
+
 def generate_unique_string(length=6):
     return secrets.token_hex(length // 2)
 
@@ -60,7 +76,7 @@ class DMButton(Button):
 async def send_button(ctx):
     if ctx.author.id == 811235357663297546:
         button = DMButton(label="Verify Discord Account", style=discord.ButtonStyle.primary, user_id=ctx.author.id)
-        view = View()
+        view = ButtonView()
         view.add_item(button)
         await ctx.send("Click the button below to generate your verification link:",view=view) #