Update app.py

app.py

@@ -90,6 +90,22 @@ button, input, textarea, select, option,
     border-top: 1px solid var(--border-color-primary, #eee);
 }
 
+/* Login section styling */
+.login-section {
+    padding: 10px;
+    margin-bottom: 15px;
+    border-radius: 8px;
+    background-color: rgba(250, 250, 250, 0.1);
+    text-align: center;
+}
+
+/* Login button styling */
+.login-button {
+    background-color: #4CAF50 !important;
+    color: white !important;
+    font-weight: bold;
+}
+
 /* Force high contrast on specific input areas */
 input[type="text"], input[type="password"], textarea {
     background-color: var(--background-fill-primary) !important;
@@ -118,6 +134,15 @@ input[type="text"], input[type="password"], textarea {
 def is_running_in_hf_space():
     return 'SPACE_ID' in os.environ
 
+# Function to trigger HuggingFace OAuth login
+def trigger_hf_oauth():
+    return """
+    <script>
+    // Redirect to HuggingFace Space login
+    window.location.href = window.location.origin + "?__auth_callback=true";
+    </script>
+    """
+
 # Start evaluation queue worker
 def start_queue_worker():
     # Wait a moment to ensure app is initialized
@@ -132,28 +157,11 @@ def start_queue_worker():
 with gr.Blocks(css=css, title="Dynamic Highscores", theme=gr.themes.Soft()) as app:
     auth_status = gr.State(None)  # Store user auth state
     
-    # Auth status check on page load
-    def check_auth_on_load(request: gr.Request):
-        # Check if running in HF Space with OAuth
-        if is_running_in_hf_space():
-            username = request.headers.get("HF-User")
-            if username:
-                print(f"Detected logged-in user via Space OAuth: {username}")
-                # Get or create user
-                user = db.get_user_by_username(username)
-                if not user:
-                    print(f"Creating new user: {username}")
-                    is_admin = (username == "Quazim0t0")  # Replace with your admin username
-                    db.add_user(username, username, is_admin)
-                    user = db.get_user_by_username(username)
-                return user, f"Logged in as {username}"
-        else:
-            # Fallback to cookie-based auth for local development
-            user = auth_manager.check_login(request)
-            if user:
-                return user, f"Logged in as {user['username']}"
-        
-        return None, "Not logged in"
+    # Add a prominent login section at the top
+    with gr.Row(visible=True, elem_classes=["login-section"]) as login_section:
+        with gr.Column():
+            login_status = gr.Markdown("### 🔒 Not logged in", elem_id="login-status")
+            login_button = gr.Button("Login with HuggingFace", size="lg", variant="primary", elem_classes=["login-button"])
     
     gr.Markdown("# 🏆 Dynamic Highscores", elem_classes=["header"])
     gr.Markdown("""
@@ -165,10 +173,6 @@ with gr.Blocks(css=css, title="Dynamic Highscores", theme=gr.themes.Soft()) as a
     - **Filter results** by model type (Merge, Agent, Reasoning, Coding, etc.)
     """, elem_classes=["info-text"])
     
-    # Authentication UI
-    login_button, logout_button, user_info = create_login_ui()
-    setup_auth_handlers(login_button, logout_button, user_info, auth_manager)
-    
     # Main tabs
     with gr.Tabs() as tabs:
         with gr.TabItem("📊 Leaderboard", id=0):
@@ -190,11 +194,39 @@ with gr.Blocks(css=css, title="Dynamic Highscores", theme=gr.themes.Soft()) as a
     Created by Quazim0t0
     """, elem_classes=["footer"])
     
+    # Auth status check on page load
+    def check_auth_on_load(request: gr.Request):
+        # Check if running in HF Space with OAuth
+        if is_running_in_hf_space():
+            username = request.headers.get("HF-User")
+            if username:
+                print(f"Detected logged-in user via Space OAuth: {username}")
+                # Get or create user
+                user = db.get_user_by_username(username)
+                if not user:
+                    print(f"Creating new user: {username}")
+                    is_admin = (username == "Quazim0t0")  # Replace with your admin username
+                    db.add_user(username, username, is_admin)
+                    user = db.get_user_by_username(username)
+                
+                # Update UI for logged in state
+                return user, f"### ✅ Logged in as {username}", gr.update(visible=False)
+        
+        # Not logged in - make sure login section is visible
+        return None, "### 🔒 Not logged in", gr.update(visible=True)
+    
+    # Connect event handlers
+    login_button.click(
+        fn=trigger_hf_oauth,
+        inputs=[],
+        outputs=[gr.HTML()]
+    )
+    
     # Check auth on load
     app.load(
         fn=check_auth_on_load, 
         inputs=[],
-        outputs=[auth_status, user_info]
+        outputs=[auth_status, login_status, login_section]
     )
 
 # Launch the app