Update app.py

app.py

@@ -36,25 +36,17 @@ def run_bot():
 threading.Thread(target=run_bot).start()
 
 # Gradio ------------------------------------------------------------------------------------------------------------
-def validate_link(request: gr.Request):
+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]
 
     if user_id is None or token is None:
-        return False, None, None
+        return "❌ Invalid link. Missing user_id or token."
 
     if int(user_id) not in user_tokens or user_tokens[int(user_id)] != token:
-        return False, None, None
-
-    return True, user_id, token
-
-def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
-    is_valid, user_id, token = validate_link(request)
-
-    if not is_valid:
-        return "❌ Invalid link or expired token."
+        return "❌ Invalid or expired token."
 
     if profile is None:
         return f"❌ Not logged in. User ID: {user_id}"
@@ -63,16 +55,8 @@ def hello(profile: gr.OAuthProfile | None, request: gr.Request) -> str:
 
 with gr.Blocks() as demo:
     gr.Markdown("# Gradio OAuth Space")
+    login_button = gr.LoginButton()
     m1 = gr.Markdown()
-
-    def conditional_login_button(request: gr.Request):
-        is_valid, _, _ = validate_link(request)
-        if is_valid:
-            return gr.LoginButton()
-        else:
-            return gr.Markdown("❌ Invalid link or expired token.")
-
-    login_button = conditional_login_button(gr.Request())
     demo.load(hello, inputs=None, outputs=m1)
 
     def check_login_status():
@@ -88,7 +72,6 @@ with gr.Blocks() as demo:
         else:
             return f"Logged in as {session.get('username', 'Unknown')}"
 
-    if isinstance(login_button, gr.LoginButton):
-        login_button.click(check_login_wrapper, inputs=None, outputs=m1)
+    login_button.click(check_login_wrapper, inputs=None, outputs=m1)
 
-demo.launch()
+demo.launch()