| |
| import logging |
|
|
| logging.basicConfig(level=logging.WARNING) |
|
|
| import gradio as gr |
| import urllib.parse |
|
|
| from apscheduler.schedulers.background import BackgroundScheduler |
| from huggingface_hub import HfApi |
|
|
| from config import LEADERBOARD_PATH, LOCAL_DEBUG |
| from content import css |
| from main_page import build_page as build_main_page |
| from literature_understanding import build_page as build_lit_page |
| from c_and_e import build_page as build_c_and_e_page |
| from data_analysis import build_page as build_data_analysis_page |
| from e2e import build_page as build_e2e_page |
| from submission import build_page as build_submission_page |
| from about import build_page as build_about_page |
|
|
| api = HfApi() |
| LOGO_PATH = "assets/logo.svg" |
| |
| scroll_script = """ |
| <script> |
| function scroll_to_element(id) { |
| console.log("Global scroll_to_element called for ID:", id); |
| const element = document.querySelector('#' + id); |
| if (element) { |
| console.log("Element found:", element); |
| element.scrollIntoView({ behavior: 'smooth', block: 'start' }); |
| } else { |
| console.error("Error: Element with ID '" + id + "' not found in the document."); |
| } |
| } |
| </script> |
| """ |
| redirect_script = """ |
| <script> |
| if (window.location.pathname === '/') { window.location.replace('/home'); } |
| </script> |
| """ |
| tooltip_script = """ |
| <script> |
| function initializeSmartTooltips() { |
| // Find all tooltip trigger icons |
| const tooltipIcons = document.querySelectorAll('.tooltip-icon-legend'); |
| |
| tooltipIcons.forEach(icon => { |
| // Find the tooltip card associated with this icon |
| const tooltipCard = icon.querySelector('.tooltip-card'); |
| if (!tooltipCard) return; |
| |
| // Move the card to the end of the <body>. This is the KEY to escaping |
| // any parent containers that might clip it. |
| document.body.appendChild(tooltipCard); |
| |
| // --- MOUSE HOVER EVENT --- |
| icon.addEventListener('mouseenter', () => { |
| // Get the exact position of the icon on the screen |
| const iconRect = icon.getBoundingClientRect(); |
| // Get the dimensions of the tooltip card |
| const cardRect = tooltipCard.getBoundingClientRect(); |
| |
| // Calculate the ideal top position (above the icon with a 10px gap) |
| const top = iconRect.top - cardRect.height - 10; |
| |
| // --- Smart Centering Logic --- |
| // Start by calculating the perfect center |
| let left = iconRect.left + (iconRect.width / 2) - (cardRect.width / 2); |
| |
| // Check if it's going off the left edge of the screen |
| if (left < 10) { |
| left = 10; // Pin it to the left with a 10px margin |
| } |
| // Check if it's going off the right edge of the screen |
| if (left + cardRect.width > window.innerWidth) { |
| left = window.innerWidth - cardRect.width - 10; // Pin it to the right |
| } |
| |
| // Apply the calculated position and show the card |
| tooltipCard.style.top = `${top}px`; |
| tooltipCard.style.left = `${left}px`; |
| tooltipCard.classList.add('visible'); |
| }); |
| |
| // --- MOUSE LEAVE EVENT --- |
| icon.addEventListener('mouseleave', () => { |
| // Hide the card |
| tooltipCard.classList.remove('visible'); |
| }); |
| }); |
| } |
| |
| // Poll the page until the tooltips exist, then run the initialization. |
| const tooltipInterval = setInterval(() => { |
| if (document.querySelector('.tooltip-icon-legend')) { |
| clearInterval(tooltipInterval); |
| initializeSmartTooltips(); |
| } |
| }, 200); |
| </script> |
| """ |
| redirect_submission_on_close_script = """ |
| <script> |
| function initializeRedirectObserver() { |
| const successModal = document.querySelector('#success-modal'); |
| |
| if (successModal) { |
| const observer = new MutationObserver((mutationsList) => { |
| for (const mutation of mutationsList) { |
| // We only care about changes to the 'class' attribute. |
| if (mutation.type === 'attributes' && mutation.attributeName === 'class') { |
| |
| // Check if the 'hide' class has been ADDED to the class list. |
| // This is how Gradio hides the modal. |
| if (successModal.classList.contains('hide')) { |
| console.log("Success modal was closed. Redirecting to homepage..."); |
| // This is the command to redirect the browser. |
| window.location.href = '/home'; |
| } |
| } |
| } |
| }); |
| |
| // Tell the observer to watch the modal for attribute changes. |
| observer.observe(successModal, { attributes: true }); |
| } |
| } |
| |
| // Polling mechanism to wait for Gradio to build the UI. |
| const redirectInterval = setInterval(() => { |
| if (document.querySelector('#success-modal')) { |
| clearInterval(redirectInterval); |
| initializeRedirectObserver(); |
| } |
| }, 200); |
| </script> |
| """ |
| |
| theme = gr.themes.Base( |
| primary_hue=gr.themes.Color(c100="#CFF5E8", c200="#B7EFDD", c300="#9FEAD1", c400="#87E5C5", c50="#E7FAF3", c500="#6FE0BA", c600="#57DBAF", c700="#3FD5A3", c800="#27D09C", c900="#0FCB8C", c950="#0fcb8c"), |
| secondary_hue=gr.themes.Color(c100="#FCDCEB", c200="#FBCBE1", c300="#F9BAD7", c400="#F7A8CD", c50="#FDEEF5", c500="#F697C4", c600="#F586BA", c700="#F375B0", c800="#F263A6", c900="#F0529C", c950="#F0529C"), |
| neutral_hue=gr.themes.Color(c100="#FDF9F4", c200="#C9C9C3", c300="#B0B5AF", c400="#97A09C", c50="#FAF2E9", c500="#7F8C89", c600="#667876", c700="#344F4F", c800="#1C3A3C", c900="#032629", c950="032629"), |
| font=[gr.themes.GoogleFont('Manrope'), 'ui-sans-serif', 'sans-serif', 'sans-serif'], |
| font_mono=[gr.themes.GoogleFont('Roboto Mono'), 'ui-monospace', 'monospace', 'monospace'], |
| ).set( |
| body_text_color='*neutral_950', |
| body_text_color_subdued='*neutral_950', |
| body_text_color_subdued_dark='*neutral_50', |
| body_text_color_dark='*neutral_50', |
| background_fill_primary='*neutral_50', |
| background_fill_primary_dark='*neutral_900', |
| background_fill_secondary='*neutral_100', |
| background_fill_secondary_dark='*neutral_800', |
| border_color_accent='*secondary_900', |
| border_color_accent_subdued='*neutral_400', |
| border_color_accent_subdued_dark='*neutral_400', |
| color_accent='*primary_900', |
| color_accent_soft='*neutral_200', |
| color_accent_soft_dark='*neutral_800', |
| link_text_color='*secondary_900', |
| link_text_color_dark='*primary_900', |
| link_text_color_active_dark='*primary_600', |
| link_text_color_hover_dark='*primary_700', |
| link_text_color_visited_dark='*primary_600', |
| table_even_background_fill='*neutral_100', |
| table_even_background_fill_dark='*neutral_800', |
| button_primary_background_fill='*secondary_900', |
| button_primary_background_fill_dark='*primary_900', |
| button_primary_background_fill_hover='*secondary_600', |
| button_primary_background_fill_hover_dark='*primary_600', |
| button_secondary_background_fill="#9FEAD1", |
| button_secondary_background_fill_dark="#9FEAD1", |
| button_secondary_text_color="*neutral_900", |
| button_secondary_text_color_dark="*neutral_900", |
| block_title_text_color="*neutral_900", |
| button_primary_text_color='*neutral_900', |
| block_title_text_color_dark="#ffffff", |
| button_primary_text_color_dark='*neutral_900', |
| block_border_color="#032629", |
| block_border_color_dark="#9fead1", |
| block_background_fill_dark="#032629", |
| block_background_fill="#FAF2E9", |
| checkbox_label_text_color="#032629", |
| checkbox_label_background_fill="#D8D6CF", |
| checkbox_label_background_fill_dark="#254243", |
| checkbox_background_color_selected="#F0529C", |
| checkbox_background_color_selected_dark="#0FCB8C", |
| ) |
| try: |
| with open(LOGO_PATH, "r") as f: |
| svg_content = f.read() |
| encoded_svg = urllib.parse.quote(svg_content) |
| home_icon_data_uri = f"data:image/svg+xml,{encoded_svg}" |
| except FileNotFoundError: |
| print(f"Warning: Home icon file not found at {LOGO_PATH}.") |
| home_icon_data_uri = "none" |
|
|
| |
| final_css = css + f""" |
| /* --- Find the "Home" button and replace its text with an icon --- */ |
| .nav-holder nav a[href$="/"] {{ |
| display: none !important; |
| }} |
| .nav-holder nav a[href*="/home"] {{ |
| grid-row: 1 !important; |
| grid-column: 1 !important; |
| justify-self: start !important; |
| display: flex !important; |
| align-items: center !important; |
| justify-content: center !important; |
| |
| /* 2. Hide the original "Home" text */ |
| font-size: 0 !important; |
| text-indent: -9999px; |
| |
| /* 3. Apply the icon as the background */ |
| background-image: url("{home_icon_data_uri}") !important; |
| background-size: contain !important; |
| background-repeat: no-repeat !important; |
| background-position: center !important; |
| |
| width: 240px !important; |
| height: 50px !important; |
| padding: 0 !important; |
| border: none !important; |
| outline: none !important; |
| }} |
| """ |
| |
| demo = gr.Blocks( |
| theme=theme, |
| css=final_css, |
| head=scroll_script + redirect_script + tooltip_script + redirect_submission_on_close_script, |
| title="AstaBench Leaderboards", |
| ) |
| with demo.route("Home", "/home"): |
| build_main_page() |
|
|
| with demo.route("Literature Understanding", "/literature-understanding"): |
| build_lit_page() |
|
|
| with demo.route("Code & Execution", "/code-execution"): |
| build_c_and_e_page() |
|
|
| with demo.route("Data Analysis", "/data-analysis"): |
| build_data_analysis_page() |
|
|
| with demo.route("End-to-End Discovery", "/discovery"): |
| build_e2e_page() |
|
|
| with demo.route("About", "/about"): |
| build_about_page() |
|
|
| with demo.route("🚀 Submit an Agent", "/submit"): |
| build_submission_page() |
| |
| def restart_space_job(): |
| print("Scheduler: Attempting to restart space.") |
| try: |
| api.restart_space(repo_id=LEADERBOARD_PATH) |
| print("Scheduler: Space restart request sent.") |
| except Exception as e: |
| print(f"Scheduler: Error restarting space: {e}") |
| scheduler = BackgroundScheduler(timezone="UTC") |
| scheduler.add_job(restart_space_job, "interval", hours=1) |
| scheduler.start() |
|
|
|
|
| |
| if __name__ == "__main__": |
| if LOCAL_DEBUG: |
| print("Launching in LOCAL_DEBUG mode.") |
| demo.launch(debug=True, allowed_paths=["assets"], favicon_path="assets/favicon/favicon.ico") |
| else: |
| print("Launching in Space mode.") |
| |
| |
| demo.launch(server_name="0.0.0.0", server_port=7860, debug=True, share=False, allowed_paths=["assets"], favicon_path="assets/favicon/favicon.ico") |
|
|
|
|
