Update app.py

app.py

@@ -12,6 +12,7 @@ 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)
 
+
 # Dictionary to store user IDs and their corresponding unique strings
 user_tokens = {}
 
@@ -41,17 +42,23 @@ def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
     query_params = parse_qs(urlparse(url_str).query)
     user_id = query_params.get('user_id', [None])[0]
     token = query_params.get('token', [None])[0]
+    user = await bot.fetch_user(user_id)
 
-    if user_id is None or token is None:
-        return "# ❌ Invalid link. Generate a new one [here](https://discord.com/channels/879548962464493619/900125909984624713) !."
+    print(f"token:{token}\n user_id:{user_id}\n profile:{profile}\n user_tokens:{user_tokens}")
 
-    if int(user_id) not in user_tokens or user_tokens[int(user_id)] != token:
-        return "# ❌ Invalid or expired link. Generate a new one [here](https://discord.com/channels/879548962464493619/900125909984624713) ! "
+    if token is None:
+        return "# ❌ Invalid link, unique code is missing. Generate a new link [here](https://discord.com/channels/879548962464493619/900125909984624713) !."
+
+    if user_id is None:
+        return "# ❌ Invalid link, discord user_id is missing. Generate a new link [here](https://discord.com/channels/879548962464493619/900125909984624713) !."
 
     if profile is None:
-        return f"# ❌ Not logged in with Hugging Face yet."
+        return f"# ❌ Log in with Hugging Face to verify!"
+    
+    if int(user_id) not in user_tokens or user_tokens[int(user_id)] != token:
+        return "# ❌ Invalid or expired link. Generate a new link [here](https://discord.com/channels/879548962464493619/900125909984624713) ! "
 
-    return f"# ✅ Successfully logged in as {profile.username}. User ID: {user_id}"
+    return f"# ✅ Successfully logged in as {profile.username}. Discord username: {user.name}"
 
 with gr.Blocks() as demo:
     with gr.Row():