test

app.py

@@ -76,7 +76,7 @@ async def on_interaction(interaction: discord.Interaction):
 
 
 class DMButton(Button):
-    def __init__(self, label, style):
+    def __init__(self):
         super().__init__(
             label="Verify Discord Account",
             style=discord.ButtonStyle.primary,
@@ -88,18 +88,35 @@ class DMButton(Button):
         await interaction.response.defer(ephemeral=True) # to fix interaction issue
 
         user_id = interaction.user.id
-    
         guild = interaction.guild
-        member = guild.get_member(user_id)
-        pending_role = guild.get_role(1380908157475360899) # pending verification role, this is to verify that the discord account actually clicked the button at least once
 
-        if member and pending_role:
-            await member.add_roles(pending_role)
-            print(f"Assigned 'Pending Verification' role to {member.name}") # should log this in #admin-logs properly
 
-        unique_link = f"<{GRADIO_APP_URL}?user_id={user_id}>" # don't need token
-        message = f"Login link generated! To complete the verification process:\n- 1 Visit this link,\n- 2 click the '🤗Sign in with Hugging Face' button\n\n{unique_link}"
-        await interaction.response.send_message(message, ephemeral=True)
+        # safer than cache:
+        try:
+            member = await guild.fetch_member(user_id)
+        except discord.NotFound:
+            await interaction.followup.send("Could not find your member record.", ephemeral=True)
+            return
+
+        pending_role = guild.get_role(1380908157475360899)
+
+        if pending_role:
+            try:
+                await member.add_roles(pending_role, reason="Clicked verification button")
+                logging.info("Assigned Pending to %s (%s)", member, member.id)
+            except discord.Forbidden:
+                logging.error("Missing permissions or role hierarchy below Pending.")
+            except Exception:
+                logging.exception("Failed to add Pending role")
+
+        unique_link = f"<{GRADIO_APP_URL}?user_id={user_id}>"
+        msg = (
+            "Login link generated! To complete verification:\n"
+            "- 1 Visit this link,\n- 2 click '🤗 Sign in with Hugging Face'\n\n"
+            f"{unique_link}"
+        )
+        # since we deferred:
+        await interaction.followup.send(msg, ephemeral=True)
 
 
 # new
@@ -109,54 +126,44 @@ class PersistentVerifyView(discord.ui.View):
         self.add_item(DMButton())
 
     async def on_error(self, error: Exception, item, interaction: discord.Interaction):
-        import logging
         logging.exception("Component error")
-        if interaction.response.is_done():
-            await interaction.followup.send("Something broke while handling that click.", ephemeral=True)
-        else:
-            await interaction.response.send_message("Something broke while handling that click.", ephemeral=True)
+        try:
+            if interaction.response.is_done():
+                await interaction.followup.send("Something broke while handling that click.", ephemeral=True)
+            else:
+                await interaction.response.send_message("Something broke while handling that click.", ephemeral=True)
+        except Exception:
+            logging.exception("Failed to notify user after component error.")
 
 
 
 @bot.event
 async def on_ready():
-    try:
-        print(f'Logged in as {bot.user.name}')
+    logging.info("Logged in as %s (%s)", bot.user, bot.user.id)
 
-        #new
-        bot.add_view(PersistentVerifyView())  # <- binds handler to custom_id
-        channel = bot.get_channel(900125909984624713)
-        msg = await channel.fetch_message(1271145797433557023)
-        await msg.edit(view=PersistentVerifyView())  # <- updates the message’s components
+    view = PersistentVerifyView()
+    bot.add_view(view)  # binds handler for the custom_id
 
-        
-    
-        guild = bot.get_guild(879548962464493619)
-        await guild.chunk() # get all users into bot cache
+    channel = bot.get_channel(900125909984624713)
+    msg = await channel.fetch_message(1271145797433557023)
 
+    # Step A: clear existing components
+    await msg.edit(view=None)
+    # Step B: attach the new persistent view (with stable custom_id)
+    await msg.edit(view=view)
 
-        """
-        channel = bot.get_channel(900125909984624713)
-        if channel:
-            try:
-                message = await channel.fetch_message(1271145797433557023)
-                if message:
-                    button = DMButton(label="Verify Discord Account", style=discord.ButtonStyle.primary)
-                    view = View(timeout=None)
-                    view.add_item(button)
-                    await message.edit(view=view)
-                    print("message edited")
-            except discord.NotFound:
-                print(f"Message with ID 1271145797433557023 not found.")
-            except discord.HTTPException as e:
-                print(f"Failed to fetch message with ID 1271145797433557023: {e}")              
-        
-        
-        """
-  
-        print("------------------------------------------------------------------------")
-    except Exception as e:
-        print(f"on_ready Error: {e}")
+    # fetch and print the custom_ids currently on the message
+    msg2 = await channel.fetch_message(msg.id)
+    try:
+        current_ids = []
+        for action_row in msg2.components:
+            for comp in action_row.children:
+                current_ids.append(getattr(comp, "custom_id", None))
+        logging.info("Message components now: %s", current_ids)
+    except Exception:
+        logging.exception("Could not introspect message components")
+
+    logging.info("------------------------------------------------------------------------")
 
 
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
@@ -172,31 +179,47 @@ threading.Thread(target=run_bot).start()
 def verify_button(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
     query_params = parse_qs(urlparse(str(request.url)).query)
     user_id = query_params.get('user_id', [None])[0]
-
     if not user_id:
         return "# ❌ Missing Discord user ID."
-
     if profile is None:
         return "# ❌ You're not logged in with Hugging Face."
 
     async def upgrade_role():
+        await bot.wait_until_ready()
         guild = bot.get_guild(879548962464493619)
-        member = guild.get_member(int(user_id))
-        pending_role = guild.get_role(1380908157475360899) # pending
-        verified_role = guild.get_role(900063512829755413) # verified
+        if not guild:
+            logging.error("Guild not ready")
+            return
 
-        if not member:
-            print(f"❌ Could not find Discord user {user_id}")
+        try:
+            member = await guild.fetch_member(int(user_id))
+        except discord.NotFound:
+            logging.error("User %s not found in guild", user_id)
             return
 
-        if pending_role in member.roles:
-            await member.remove_roles(pending_role)
-            await member.add_roles(verified_role)
-            print(f"✅ {member.name} verified and upgraded to verified role.")
-        else:
-            print(f"⚠️ {member.name} did not have the pending role. Ignoring.")
+        pending_role = guild.get_role(1380908157475360899)
+        verified_role = guild.get_role(900063512829755413)
+        if not verified_role:
+            logging.error("Verified role missing")
+            return
+
+        # remove pending if present
+        if pending_role and pending_role in member.roles:
+            try:
+                await member.remove_roles(pending_role, reason="Verification complete")
+            except Exception:
+                logging.exception("Remove pending failed")
+
+        # always try to add verified
+        if verified_role not in member.roles:
+            try:
+                await member.add_roles(verified_role, reason=f"HF verified {profile.username}")
+                logging.info("Added Verified to %s", member)
+            except discord.Forbidden:
+                logging.error("Forbidden adding Verified (perm/hierarchy)")
+            except Exception:
+                logging.exception("Add Verified failed")
 
-    global discord_loop
     if discord_loop:
         asyncio.run_coroutine_threadsafe(upgrade_role(), discord_loop)
     return f"# ✅ Verification successful! Welcome, {profile.username} 🎉"