Update app.py

app.py

@@ -5,33 +5,27 @@ from urllib.parse import urlparse, parse_qs
 import discord
 from discord.ext import commands
 
-
-
 # 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------------------------------------------------------------------------------------------------------------
+# gradio------------------------------------------------------------------------------------------------------------
 def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
     url_str = str(request.url)
     query_params = parse_qs(urlparse(url_str).query)
@@ -40,18 +34,25 @@ def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
         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"
-    )
-    gr.LoginButton()
-    # ^ add a login button to the Space
+    gr.Markdown("# Gradio OAuth Space")
+    login_button = gr.LoginButton()
     m1 = gr.Markdown()
     demo.load(hello, inputs=None, outputs=m1)
 
-demo.launch()
+    def check_login_status():
+        try:
+            return login_button.get_session().get("oauth_info", None)
+        except AttributeError:
+            return None
+
+    def check_login_wrapper():
+        session = check_login_status()
+        if session is None:
+            return "Not logged in."
+        else:
+            return f"Logged in as {session.get('username', 'Unknown')}"
 
+    login_button.click(check_login_wrapper, inputs=None, outputs=m1)
 
+demo.launch()