Update app.py

app.py

@@ -5,6 +5,7 @@ from urllib.parse import urlparse, parse_qs
 import discord
 from discord.ext import commands
 import secrets
+import asyncio
 
 # Discord bot -----------------------------------------------------------------------------------------------
 intents = discord.Intents.all()
@@ -30,11 +31,6 @@ 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()
-
 # Gradio ------------------------------------------------------------------------------------------------------------
 def validate_link(request: gr.Request) -> bool:
     referer = request.headers.get('Referer', '')
@@ -93,4 +89,15 @@ with gr.Blocks() as demo:
 
     login_button.click(check_login_wrapper, inputs=None, outputs=m1)
 
-demo.launch()
+def run_gradio():
+    demo.launch()
+
+# Run the Gradio app in a separate thread
+gradio_thread = threading.Thread(target=run_gradio)
+gradio_thread.start()
+
+# Run the Discord bot on the main thread
+if DISCORD_TOKEN:
+    bot.run(DISCORD_TOKEN)
+else:
+    print("Discord token not found. Please set the DISCORD_TOKEN environment variable.")