big oauth test

app.py

@@ -1,31 +1,55 @@
 import gradio as gr
-from huggingface_hub import list_models
+import os
+from urllib.parse import urlparse, parse_qs
+import discord
+from discord.ext import commands
 
 
-def hello(profile: gr.OAuthProfile | None) -> str:
-    # ^ expect a gr.OAuthProfile object as input to get the user's profile
-    # if the user is not logged in, profile will be None
-    if profile is None:
-        return "❌ Not logged in."
-    return f"✅ Successfully logged in as `{profile.username}`."
+
+# discord bot -----------------------------------------------------------------------------------------------
+intents = discord.Intents.all()
+bot = commands.Bot(command_prefix="!", intents=intents)
+GRADIO_APP_URL = "https://lunarflu-gradio-oauth2"
+DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
+
+
+@bot.event
+async def on_ready():
+    print(f'Logged in as {bot.user}')
+
+
+@bot.command()
+async def sendlink(ctx, user: discord.User):
+    if ctx.author.id == 811235357663297546:
+        unique_link = f"{GRADIO_APP_URL}?user_id={user.id}"
+        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()
+
+
+#gradio------------------------------------------------------------------------------------------------------------
+def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
+    query_params = parse_qs(urlparse(request.url).query)
+    user_id = query_params.get('user_id', [None])[0]
+    if profile is None:
+        return f"❌ Not logged in. User ID: {user_id}"
+    return f"✅ Successfully logged in as {profile.username}. User ID: {user_id}"
+
 
 
 
 with gr.Blocks() as demo:
     gr.Markdown(
         "# Gradio OAuth Space"
-        "\n\nThis Space is a demo for the **Sign in with Hugging Face** feature. "
-        "Duplicate this Space to get started."
-        "\n\nFor more details, check out:"
-        "\n- https://www.gradio.app/guides/sharing-your-app#o-auth-login-via-hugging-face"
-        "\n- https://huggingface.co/docs/hub/spaces-oauth"
     )
     gr.LoginButton()
     # ^ add a login button to the Space
     m1 = gr.Markdown()
-    m2 = gr.Markdown()
     demo.load(hello, inputs=None, outputs=m1)
 
 demo.launch()
+
+