fix style

pages/benchmark_viewer.py

@@ -46,7 +46,7 @@ def show():
     with col2:
         # Single Model Multiselect (filtered by selected datasets)
         available_models = filtered_df['model'].unique()
-        selected_models = st.multiselect("Model(s)", available_models, default=['qwen-2.5-7b-instruct'] if 'qwen-2.5-7b-instruct' in available_models else available_models) # Default to qwen-2.5-7b-instruct if available, otherwise select all
+        selected_models = st.multiselect("Model(s)", available_models, default=['llama-3.1-8b-instruct','qwen-2.5-7b-instruct','llama-3.1-70b-instruct','qwen-2.5-72b-instruct','deepseek-v3'])
 
     with col3:
         min_op, max_op = st.slider("Op Range", int(filtered_df['op'].min()), int(filtered_df['op'].max()), (int(filtered_df['op'].min()), int(filtered_df['op'].max())))

utils/style.py

@@ -1,11 +1,18 @@
 import pandas as pd
-
+import streamlit as st
+# Define color maps for both light and dark modes
 COLOR_MAP = {
-    "yellow": "background-color: #FFFFCC",  # Reasoning models
-    "green": "background-color: #E3FBE9",   # Linear attention hybrid
-    "blue": "background-color: #E6F4FF"     # SSM hybrid models
+    "light": {
+        "yellow": "background-color: rgba(255, 255, 204, 0.5)",  # Reasoning models
+        "green": "background-color: rgba(227, 251, 233, 0.5)",   # Linear attention hybrid
+        "blue": "background-color: rgba(230, 244, 255, 0.5)"     # SSM hybrid models
+    },
 }
 
+def get_color_map():
+    """Returns the appropriate color map based on the current theme."""
+    return COLOR_MAP["light"]
+
 def style_zero_context(df):
     """
     Similar approach to style_long_context:
@@ -16,11 +23,11 @@ def style_zero_context(df):
 
     # Example color dict, tweak as needed:
     color_mapping = {
-        "minimax-text-01": COLOR_MAP["green"],
-        "jamba-1.5-large": COLOR_MAP["blue"],
-        "deepseek-r1": COLOR_MAP["yellow"],
-        "o1-mini": COLOR_MAP["yellow"],
-        "qwq-32b-preview": COLOR_MAP["yellow"],
+        "minimax-text-01": get_color_map()["green"],
+        "jamba-1.5-large": get_color_map()["blue"],
+        "deepseek-r1": get_color_map()["yellow"],
+        "o1-mini": get_color_map()["yellow"],
+        "qwq-32b-preview": get_color_map()["yellow"],
         # Add any other special-cased models here
         # "o1-mini": COLOR_MAP["yellow"], etc.
     }
@@ -55,8 +62,8 @@ def style_zero_context(df):
 # Add styling for model types
 def style_long_context(df):
     color_mapping = {
-        "minimax-text-01": COLOR_MAP["green"],
-        "jamba-1.5-large": COLOR_MAP["blue"]
+        "minimax-text-01": get_color_map()["green"],
+        "jamba-1.5-large": get_color_map()["blue"]
     }
     
     return df.style.apply(