Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 import pandas as pd
+import os
+import zipfile
 
 CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
 CITATION_BUTTON_TEXT = r"""@misc{aienergyscore-leaderboard,
@@ -37,20 +39,38 @@ def make_link(mname):
     display_name = parts[1] if len(parts) > 1 else mname
     return f'<a href="https://huggingface.co/{mname}" target="_blank">{display_name}</a>'
 
+def extract_link_text(html_link):
+    """Extracts the inner text from an HTML link."""
+    start = html_link.find('>') + 1
+    end = html_link.rfind('</a>')
+    if start > 0 and end > start:
+        return html_link[start:end]
+    else:
+        return html_link
+
 def generate_html_table_from_df(df):
     """
     Given a dataframe with a numeric energy column (gpu_energy_numeric),
     generate an HTML table with three columns:
-      - Model (the link)
+      - Model (the link, with a fixed width based on the longest model name)
       - GPU Energy (Wh) plus a horizontal bar whose width is proportional
         to the energy value relative to the maximum in the table.
       - Score (displayed as stars)
     """
+    # Compute a static width (in pixels) for the Model column based on the longest model name.
+    if not df.empty:
+        max_length = max(len(extract_link_text(link)) for link in df['Model'])
+    else:
+        max_length = 10
+    # Multiply by an estimated average character width (10 pixels) and add some extra padding.
+    static_width = max_length * 10 + 16
+
     max_energy = df['gpu_energy_numeric'].max() if not df.empty else 1
     color_map = {"1": "red", "2": "orange", "3": "yellow", "4": "lightgreen", "5": "green"}
     html = '<table style="width:100%; border-collapse: collapse; font-family: Arial, sans-serif;">'
-    html += '<thead><tr style="background-color: #f2f2f2;">'
-    html += '<th style="text-align: left; padding: 8px;">Model</th>'
+    # Apply the static width to the Model column header.
+    html += f'<thead><tr style="background-color: #f2f2f2;">'
+    html += f'<th style="text-align: left; padding: 8px; width: {static_width}px;">Model</th>'
     html += '<th style="text-align: left; padding: 8px;">GPU Energy (Wh)</th>'
     html += '<th style="text-align: left; padding: 8px;">Score</th>'
     html += '</tr></thead>'
@@ -63,7 +83,8 @@ def generate_html_table_from_df(df):
         score_val = row['energy_score']
         bar_color = color_map.get(str(score_val), "gray")
         html += '<tr>'
-        html += f'<td style="padding: 8px;">{row["Model"]}</td>'
+        # Apply the static width to the Model column cell.
+        html += f'<td style="padding: 8px; width: {static_width}px;">{row["Model"]}</td>'
         html += (
             f'<td style="padding: 8px;">{energy_str}<br>'
             f'<div style="background-color: {bar_color}; width: {bar_width:.1f}%; height: 10px;"></div></td>'
@@ -73,6 +94,17 @@ def generate_html_table_from_df(df):
     html += '</tbody></table>'
     return html
 
+# --- Function to zip all CSV files ---
+def zip_csv_files():
+    data_dir = "data/energy"
+    zip_filename = "data.zip"
+    with zipfile.ZipFile(zip_filename, "w", zipfile.ZIP_DEFLATED) as zipf:
+        for filename in os.listdir(data_dir):
+            if filename.endswith(".csv"):
+                filepath = os.path.join(data_dir, filename)
+                zipf.write(filepath, arcname=filename)
+    return zip_filename
+
 # --- Modified functions to include a sort_order parameter ---
 def get_model_names_html(task, sort_order="High to Low"):
     df = pd.read_csv('data/energy/' + task)
@@ -181,17 +213,14 @@ with demo:
 ### Welcome to the leaderboard for the [AI Energy Score Project!](https://huggingface.co/AIEnergyScore) — Select different tasks to see scored models."""
     )
     
-    # Header links:
-    gr.HTML('''
-    <div style="text-align: center; margin-bottom: 20px;">
-        <a href="https://huggingface.co/spaces/AIEnergyScore/submission_portal" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">Submission Portal</a>
-        <a href="https://huggingface.co/spaces/AIEnergyScore/Label" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">Label Generator</a>
-        <a href="https://huggingface.github.io/AIEnergyScore/#faq" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">FAQ</a>
-        <a href="https://huggingface.github.io/AIEnergyScore/#documentation" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">Documentation</a>
-        <a href="https://huggingface.co/spaces/AIEnergyScore/README/discussions" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">Community</a>
-
-    </div>
-    ''')
+    # Header links (now using a row of components, including a Download Data button)
+    with gr.Row():
+        submission_link = gr.HTML('<a href="https://huggingface.co/spaces/AIEnergyScore/submission_portal" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">Submission Portal</a>')
+        label_link = gr.HTML('<a href="https://huggingface.co/spaces/AIEnergyScore/Label" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">Label Generator</a>')
+        download_button = gr.DownloadButton("Download Data", fn=zip_csv_files, file_name="data.zip", variant="link", style={"margin": "0 15px", "font-weight": "bold", "font-size": "1.1em"})
+        faq_link = gr.HTML('<a href="https://huggingface.github.io/AIEnergyScore/#faq" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">FAQ</a>')
+        documentation_link = gr.HTML('<a href="https://huggingface.github.io/AIEnergyScore/#documentation" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">Documentation</a>')
+        community_link = gr.HTML('<a href="https://huggingface.co/spaces/AIEnergyScore/README/discussions" style="margin: 0 15px; text-decoration: none; font-weight: bold; font-size: 1.1em;">Community</a>')
     
     with gr.Tabs():
         # --- Text Generation Tab ---
@@ -327,4 +356,4 @@ with demo:
         )
     gr.Markdown("""Last updated: February 2025""")
 
-demo.launch()
+demo.launch()