Spaces:
Running
Running
| import gradio as gr | |
| import os | |
| import threading | |
| from urllib.parse import urlparse, parse_qs | |
| import discord | |
| from discord.ext import commands | |
| import secrets | |
| # Discord bot ----------------------------------------------------------------------------------------------- | |
| intents = discord.Intents.all() | |
| 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 = {} | |
| async def on_ready(): | |
| print(f'Logged in as {bot.user}') | |
| def generate_unique_string(length=6): | |
| return secrets.token_hex(length // 2) | |
| async def sendlink(ctx, user: discord.User): | |
| if ctx.author.id == 811235357663297546: | |
| unique_string = generate_unique_string() | |
| user_tokens[user.id] = unique_string | |
| 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 hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str: | |
| 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] | |
| user = await bot.fetch_user(user_id) | |
| print(f"token:{token}\n user_id:{user_id}\n profile:{profile}\n user_tokens:{user_tokens}") | |
| if token is None: | |
| return "# ❌ Invalid link, unique code is missing. Generate a new link [here](https://discord.com/channels/879548962464493619/900125909984624713) !." | |
| if user_id is None: | |
| return "# ❌ Invalid link, discord user_id is missing. Generate a new link [here](https://discord.com/channels/879548962464493619/900125909984624713) !." | |
| if profile is None: | |
| return f"# ❌ Log in with Hugging Face to verify!" | |
| if int(user_id) not in user_tokens or user_tokens[int(user_id)] != token: | |
| return "# ❌ Invalid or expired link. Generate a new link [here](https://discord.com/channels/879548962464493619/900125909984624713) ! " | |
| 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() |