asyncio

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-import threading
+import asyncio
 from urllib.parse import urlparse, parse_qs
 import discord
 from discord.ext import commands
@@ -12,7 +12,6 @@ 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 = {}
 
@@ -31,10 +30,8 @@ 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()
+async def run_bot():
+    await bot.start(DISCORD_TOKEN)
 
 # Gradio ------------------------------------------------------------------------------------------------------------
 async def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
@@ -60,30 +57,38 @@ async def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
 
     return f"# ✅ Successfully logged in as {profile.username}. Discord username: {user.name}"
 
-with gr.Blocks() as demo:
-    with gr.Row():
-        gr.Markdown("# Discord Verification Space")
-    with gr.Row():
-        login_button = gr.LoginButton()
-
-            
-    m1 = gr.Markdown()
-    demo.load(hello, inputs=None, outputs=m1)
-
-
-    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()
+def launch_gradio():
+    with gr.Blocks() as demo:
+        with gr.Row():
+            gr.Markdown("# Discord Verification Space")
+        with gr.Row():
+            login_button = gr.LoginButton()
+
+        m1 = gr.Markdown()
+        demo.load(hello, inputs=None, outputs=m1)
+
+        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()
+
+async def main():
+    await asyncio.gather(
+        run_bot(),
+        asyncio.to_thread(launch_gradio)
+    )
+
+if __name__ == "__main__":
+    asyncio.run(main())