button test

app.py

@@ -1,12 +1,15 @@
-import gradio as gr
 import os
-import threading
-from urllib.parse import urlparse, parse_qs
 import discord
-from discord.ext import commands
 import secrets
+import threading
+import gradio as gr
 
-# Discord bot -----------------------------------------------------------------------------------------------
+from discord.ext import commands
+from discord.ui import Button, View
+from urllib.parse import urlparse, parse_qs
+
+
+# Discord bot ------------------------------------------------------------------------------------------------------
 intents = discord.Intents.all()
 bot = commands.Bot(command_prefix="!", intents=intents)
 GRADIO_APP_URL = "https://huggingface.co/spaces/lunarflu/gradio-oauth2"
@@ -22,6 +25,8 @@ async def on_ready():
 def generate_unique_string(length=6):
     return secrets.token_hex(length // 2)
 
+
+# commands ---------------------------------------------------------------------------------------------------------
 @bot.command()
 async def sendlink(ctx, user: discord.User):
     if ctx.author.id == 811235357663297546:
@@ -30,10 +35,27 @@ async def sendlink(ctx, user: discord.User):
         unique_link = f"{GRADIO_APP_URL}?user_id={user.id}&token={unique_string}"
         await user.send(f"Click the link to sign in with Hugging Face: {unique_link}")
 
-def run_bot():
-    bot.run(DISCORD_TOKEN)
 
-threading.Thread(target=run_bot).start()
+class DMButton(Button):
+    def __init__(self, label, style, message):
+        super().__init__(label=label, style=style)
+        self.message = message
+
+    async def callback(self, interaction: discord.Interaction):
+        await interaction.user.send(self.message)
+        await interaction.response.send_message(f"DM sent to {interaction.user.mention}", ephemeral=True)
+
+@bot.command(name='sendbutton')
+async def send_button(ctx):
+    if ctx.author.id == 811235357663297546:
+        unique_string = generate_unique_string()
+        user_tokens[user.id] = unique_string
+        unique_link = f"{GRADIO_APP_URL}?user_id={user.id}&token={unique_string}"
+        button = DMButton(label="Verify Discord Account", style=discord.ButtonStyle.primary, message=f"To complete the verification process, visit this link and click the 'sign in with Hugging Face' button: {unique_link}")
+        view = View()
+        view.add_item(button)
+        await ctx.send("Click the button below:", view=view)
+
 
 # Gradio ------------------------------------------------------------------------------------------------------------
 def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
@@ -83,3 +105,8 @@ with gr.Blocks() as demo:
     login_button.click(check_login_wrapper, inputs=None, outputs=m1)
 
 demo.launch()
+
+def run_bot():
+    bot.run(DISCORD_TOKEN)
+
+threading.Thread(target=run_bot).start()