summary_page.py

summary_page.py

@@ -1,6 +1,5 @@
 import matplotlib.pyplot as plt
 import pandas as pd
-from matplotlib.patches import FancyBboxPatch
 from data import extract_model_data
 
 # Layout parameters
@@ -38,9 +37,6 @@ LABEL_FONT_SIZE = 14
 LABEL_OFFSET = 1 
 FAILURE_RATE_FONT_SIZE = 28
 
-# Label box settings (for text labels)
-LOGO_BOX_WIDTH = 4.5
-LOGO_BOX_HEIGHT = 0.43
 
 
 def calculate_overall_failure_rates(df: pd.DataFrame, available_models: list[str]) -> tuple[float, float]:
@@ -78,7 +74,7 @@ def calculate_overall_failure_rates(df: pd.DataFrame, available_models: list[str
     return amd_failure_rate, nvidia_failure_rate
 
 
-def draw_logo_and_bar(
+def draw_text_and_bar(
     label: str,
     stats: dict[str, int],
     y_bar: float,
@@ -86,46 +82,19 @@ def draw_logo_and_bar(
     bar_height: float,
     ax: plt.Axes,
 ) -> None:
-    """Draw a horizontal bar chart for given stats with a logo box on the left."""
-    # Determine if there are failures
+    """Draw a horizontal bar chart for given stats and its label on the left."""
+    # Text
+    label_x = column_left_position - LABEL_OFFSET
     failures_present = any(stats[category] > 0 for category in ['failed', 'error'])
-    
-    # Calculate box position (centered on the bar vertically)
-    box_x = column_left_position - LABEL_OFFSET - LOGO_BOX_WIDTH
-    box_y = y_bar - LOGO_BOX_HEIGHT / 2
-    
-    # Draw the colored box
     if failures_present:
-        box_color = COLORS['failed']  # Red for failures
-        box_alpha = 0.6
+        props = dict(boxstyle='round', facecolor=COLORS['failed'], alpha=0.35)
     else:
-        box_color = '#2a2a2a'  # Dark gray for no failures
-        box_alpha = 0.5
-    
-    box = FancyBboxPatch(
-        (box_x, box_y), 
-        LOGO_BOX_WIDTH, 
-        LOGO_BOX_HEIGHT,
-        boxstyle="round,pad=0.05",
-        facecolor=box_color,
-        edgecolor='#444444',
-        linewidth=1,
-        alpha=box_alpha
-    )
-    ax.add_patch(box)
-    
-    # Add text label in the box
+        props = dict(alpha=0)
     ax.text(
-        box_x + LOGO_BOX_WIDTH / 2, y_bar, 
-        label.upper(), 
-        ha='center', va='center', 
-        color='#FFFFFF', 
-        fontsize=10,
-        fontfamily='monospace', 
-        fontweight='bold'
+        label_x, y_bar, label, ha='right', va='center', color='#CCCCCC', fontsize=LABEL_FONT_SIZE,
+        fontfamily='monospace', fontweight='normal', bbox=props
     )
-    
-    # Draw the bar
+    # Bar
     total = sum(stats.values())
     if total > 0:
         left = column_left_position
@@ -217,10 +186,9 @@ def create_summary_page(df: pd.DataFrame, available_models: list[str]) -> plt.Fi
 
         # AMD label and bar in this column
         bar_height = min(0.4, vertical_spacing * BAR_HEIGHT_RATIO)
-        # Draw AMD bar with logo
-        draw_logo_and_bar("amd", amd_stats, y_amd_bar, col_left, bar_height, ax)
-        # Draw NVIDIA bar with logo
-        draw_logo_and_bar("nvidia", nvidia_stats, y_nvidia_bar, col_left, bar_height, ax)
+        draw_text_and_bar("amd", amd_stats, y_amd_bar, col_left, bar_height, ax)
+        
+        draw_text_and_bar("nvidia", nvidia_stats, y_nvidia_bar, col_left, bar_height, ax)
 
         # Increment counter for next visible model
         visible_model_count += 1