revert

app.py

@@ -5,7 +5,6 @@ from urllib.parse import urlparse, parse_qs
 import discord
 from discord.ext import commands
 import secrets
-import asyncio
 
 # Discord bot -----------------------------------------------------------------------------------------------
 intents = discord.Intents.all()
@@ -31,10 +30,14 @@ 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', '')
-    url_str = str(referer)
+    url_str = str(request.url)
     query_params = parse_qs(urlparse(url_str).query)
     user_id = query_params.get('user_id', [None])[0]
     token = query_params.get('token', [None])[0]
@@ -51,8 +54,7 @@ def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
     if not validate_link(request):
         return "❌ Invalid link. Please use the link provided by the bot."
 
-    referer = request.headers.get('Referer', '')
-    url_str = str(referer)
+    url_str = str(request.url)
     query_params = parse_qs(urlparse(url_str).query)
     user_id = query_params.get('user_id', [None])[0]
 
@@ -89,15 +91,4 @@ with gr.Blocks() as demo:
 
     login_button.click(check_login_wrapper, inputs=None, outputs=m1)
 
-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.")
+demo.launch()