some styling

summary_page.py

@@ -1,5 +1,9 @@
 import matplotlib.pyplot as plt
 import pandas as pd
+from matplotlib.offsetbox import OffsetImage, AnnotationBbox
+from matplotlib.patches import FancyBboxPatch
+import matplotlib.image as mpimg
+import os
 from data import extract_model_data
 
 # Layout parameters
@@ -37,6 +41,22 @@ LABEL_FONT_SIZE = 14
 LABEL_OFFSET = 1  # Distance of label from bar
 FAILURE_RATE_FONT_SIZE = 28
 
+# Logo settings
+LOGO_BOX_WIDTH = 4.5  # Width of the logo box
+LOGO_BOX_HEIGHT = 0.8  # Height of the logo box
+LOGO_ZOOM = 0.08  # Size of the logo image
+
+# Load logos once at module level
+SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+try:
+    AMD_LOGO = mpimg.imread(os.path.join(SCRIPT_DIR, 'logos/amd_logo.png'))
+except:
+    AMD_LOGO = None
+try:
+    NVIDIA_LOGO = mpimg.imread(os.path.join(SCRIPT_DIR, 'logos/nvidia_logo.png'))
+except:
+    NVIDIA_LOGO = None
+
 
 def calculate_overall_failure_rates(df: pd.DataFrame, available_models: list[str]) -> tuple[float, float]:
     """Calculate overall failure rates for AMD and NVIDIA across all models."""
@@ -73,7 +93,7 @@ def calculate_overall_failure_rates(df: pd.DataFrame, available_models: list[str
     return amd_failure_rate, nvidia_failure_rate
 
 
-def draw_text_and_bar(
+def draw_logo_and_bar(
     label: str,
     stats: dict[str, int],
     y_bar: float,
@@ -81,19 +101,72 @@ def draw_text_and_bar(
     bar_height: float,
     ax: plt.Axes,
 ) -> None:
-    """Draw a horizontal bar chart for given stats and its label on the left."""
-    # Text
-    label_x = column_left_position - LABEL_OFFSET
+    """Draw a horizontal bar chart for given stats with a logo box on the left."""
+    # Determine if there are failures
     failures_present = any(stats[category] > 0 for category in ['failed', 'error'])
+    
+    # Select the appropriate logo
+    logo = AMD_LOGO if label.lower() == "amd" else NVIDIA_LOGO
+    
+    # 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:
-        props = dict(boxstyle='round', facecolor=COLORS['failed'], alpha=0.35)
+        box_color = COLORS['failed']  # Red for failures
+        box_alpha = 0.6
     else:
-        props = dict(alpha=0)
-    ax.text(
-        label_x, y_bar, label, ha='right', va='center', color='#CCCCCC', fontsize=LABEL_FONT_SIZE,
-        fontfamily='monospace', fontweight='normal', bbox=props
+        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
     )
-    # Bar
+    ax.add_patch(box)
+    
+    # Add logo image inside the box if available
+    if logo is not None:
+        try:
+            imagebox = OffsetImage(logo, zoom=LOGO_ZOOM)
+            ab = AnnotationBbox(
+                imagebox, 
+                (box_x + LOGO_BOX_WIDTH / 2, y_bar),
+                frameon=False,
+                box_alignment=(0.5, 0.5)
+            )
+            ax.add_artist(ab)
+        except:
+            # Fallback to text if logo doesn't work
+            ax.text(
+                box_x + LOGO_BOX_WIDTH / 2, y_bar, 
+                label.upper(), 
+                ha='center', va='center', 
+                color='#FFFFFF', 
+                fontsize=10,
+                fontfamily='monospace', 
+                fontweight='bold'
+            )
+    else:
+        # Fallback to text if logo not loaded
+        ax.text(
+            box_x + LOGO_BOX_WIDTH / 2, y_bar, 
+            label.upper(), 
+            ha='center', va='center', 
+            color='#FFFFFF', 
+            fontsize=10,
+            fontfamily='monospace', 
+            fontweight='bold'
+        )
+    
+    # Draw the bar
     total = sum(stats.values())
     if total > 0:
         left = column_left_position
@@ -172,10 +245,10 @@ 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
-        draw_text_and_bar("amd", amd_stats, y_amd_bar, col_left, bar_height, ax)
-        # Draw NVIDIA bar
-        draw_text_and_bar("nvidia", nvidia_stats, y_nvidia_bar, col_left, bar_height, ax)
+        # 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)
 
         # Increment counter for next visible model
         visible_model_count += 1