Update app.py

app.py

@@ -50,7 +50,7 @@ if not benchmarks or len(benchmarks) == 0:
 else:
     print(f"Found {len(benchmarks)} existing benchmarks.")
 
-# Custom CSS with better theme compatibility
+# Custom CSS with theme awareness
 css = """
 /* Theme-adaptive colored info box */
 .info-text {
@@ -102,40 +102,22 @@ input[type="text"], input[type="password"], textarea {
     color: var(--body-text-color) !important;
 }
 
-/* Make Textbox placeholders more visible */
-::placeholder {
-    color: var(--body-text-color-subdued) !important;
-    opacity: 0.7 !important;
-}
-
-/* Make tab labels more visible */
-.tab-nav button {
-    color: var(--body-text-color) !important;
-}
-
-/* Improve contrast in dataframes */
-table.dataframe th, table.dataframe td {
-    color: var(--body-text-color) !important;
-}
-
-/* Ensure buttons have visible text */
-button.primary {
-    color: white !important;
-}
-
-/* Force high contrast for dropdown/select options */
-.select-option {
-    color: var(--body-text-color) !important;
-}
-
-/* Improve visibility in dark mode specifically */
+/* Fix dark mode text visibility */
 @media (prefers-color-scheme: dark) {
-    .info-text {
-        background-color: rgba(53, 130, 220, 0.2);
+    input, textarea, select {
+        color: #ffffff !important;
+    }
+    
+    ::placeholder {
+        color: rgba(255, 255, 255, 0.5) !important;
     }
 }
 """
 
+# Check if the server is running in a HuggingFace Space
+def is_running_in_hf_space():
+    return 'SPACE_ID' in os.environ
+
 # Start evaluation queue worker
 def start_queue_worker():
     # Wait a moment to ensure app is initialized
@@ -147,7 +129,32 @@ def start_queue_worker():
         print(f"Error starting queue worker: {e}")
 
 # Create Gradio app
-with gr.Blocks(css=css, title="Dynamic Highscores", theme=gr.themes.Default()) as app:
+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"
+    
     gr.Markdown("# 🏆 Dynamic Highscores", elem_classes=["header"])
     gr.Markdown("""
     Welcome to Dynamic Highscores - a community benchmark platform for evaluating and comparing language models.
@@ -159,8 +166,8 @@ with gr.Blocks(css=css, title="Dynamic Highscores", theme=gr.themes.Default()) a
     """, elem_classes=["info-text"])
     
     # Authentication UI
-    login_button, logout_button, token_input, user_info = create_login_ui()
-    setup_auth_handlers(login_button, logout_button, token_input, user_info, auth_manager)
+    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:
@@ -182,6 +189,13 @@ with gr.Blocks(css=css, title="Dynamic Highscores", theme=gr.themes.Default()) a
     
     Created by Quazim0t0
     """, elem_classes=["footer"])
+    
+    # Check auth on load
+    app.load(
+        fn=check_auth_on_load, 
+        inputs=[],
+        outputs=[auth_status, user_info]
+    )
 
 # Launch the app
 if __name__ == "__main__":