Code Changes - Reranking Implementation

app.py

@@ -278,12 +278,16 @@ def plot_chunking_strategies_old1(subset_name):
         print(f"Error plotting chunking strategies: {str(e)}")
 
 def plot_chunking_strategies(subset_name):
-    """Visualize chunking strategy distribution with consistent formatting."""
+    """Visualize chunking strategy distribution with consistent formatting and border."""
     try:
         chunking_data = ragbench_details[subset_name]["chunking"]
         
-        # Create figure with constrained layout
-        fig, ax = plt.subplots(figsize=(10, 5), constrained_layout=True)
+        # Create figure with constrained layout and white background
+        fig, ax = plt.subplots(figsize=(10, 5), constrained_layout=True, facecolor='white')
+        
+        # Add a subtle border around the entire figure
+        fig.patch.set_edgecolor('lightgray')
+        fig.patch.set_linewidth(2)
         
         # Prepare data
         strategies = list(chunking_data.keys())
@@ -292,10 +296,10 @@ def plot_chunking_strategies(subset_name):
         # Plot bars with consistent styling
         bars = ax.bar(strategies, counts, color='skyblue', edgecolor='white', linewidth=0.7)
         
-        # Add value labels (consistent with plot_subset_metrics)
+        # Add value labels
         ax.bar_label(bars, fmt='%d', padding=3, fontsize=9)
         
-        # Customize plot (aligned with plot_subset_metrics style)
+        # Customize plot
         ax.set_title(
             f"Chunking Strategy Distribution - {subset_name}",
             fontsize=12,
@@ -304,11 +308,11 @@ def plot_chunking_strategies(subset_name):
         )
         ax.set_ylabel("Number of Chunks", fontsize=10)
         
-        # Rotate x-labels (consistent angle)
+        # Rotate x-labels
         ax.set_xticks(range(len(strategies)))
         ax.set_xticklabels(
             strategies,
-            rotation=25,  # Matches 25° from plot_subset_metrics
+            rotation=25,
             ha='right',
             fontsize=9,
             rotation_mode='anchor'
@@ -317,17 +321,19 @@ def plot_chunking_strategies(subset_name):
         # Consistent grid and spines
         ax.grid(axis='y', linestyle=':', alpha=0.6)
         ax.spines[['top', 'right']].set_visible(False)
+        # Add border around plot area
+        ax.spines[['left', 'bottom']].set_color('lightgray')
+        ax.spines[['left', 'bottom']].set_linewidth(1.5)
         
         # Auto-scale with same headroom
         ax.set_ylim(0, max(counts) * 1.1)
         
         return fig
-        
     except Exception as e:
         print(f"Error plotting chunking strategies: {str(e)}")
 
 def plot_subset_metrics(subset_name):
-    """Generate a bar plot of key metrics with consistent formatting."""
+    """Generate a bar plot of key metrics with consistent formatting and border."""
     try:
         summary = ragbench_details[subset_name]["summary"]
         
@@ -340,13 +346,15 @@ def plot_subset_metrics(subset_name):
             'UniqueDocsPercent': '% Unique Docs'
         }
         
+        # Create figure with white background and border
+        fig, ax = plt.subplots(figsize=(10, 5), constrained_layout=True, facecolor='white')
+        fig.patch.set_edgecolor('lightgray')
+        fig.patch.set_linewidth(2)
+        
         # Prepare data
         display_names = list(metrics.values())
         values = [summary.get(metric, 0) for metric in metrics.keys()]
         
-        # Create figure with same layout
-        fig, ax = plt.subplots(figsize=(10, 5), constrained_layout=True)
-        
         # Plot bars with same style
         bars = ax.bar(display_names, values, color='skyblue', edgecolor='white', linewidth=0.7)
         
@@ -360,16 +368,19 @@ def plot_subset_metrics(subset_name):
             
         ax.set_title(title, fontsize=12, pad=20, loc='left')
         
-        # Consistent axis styling
+        # Consistent axis styling with border
         ax.set_ylabel("Count", fontsize=10)
         ax.grid(axis='y', linestyle=':', alpha=0.6)
         ax.spines[['top', 'right']].set_visible(False)
+        # Add border around plot area
+        ax.spines[['left', 'bottom']].set_color('lightgray')
+        ax.spines[['left', 'bottom']].set_linewidth(1.5)
         
         # Same label rotation
         ax.set_xticks(range(len(display_names)))
         ax.set_xticklabels(display_names, rotation=25, ha='right', fontsize=9)
         
-        # Special percentage handling (now matches chunking plot's y-limit logic)
+        # Special percentage handling
         if 'UniqueDocsPercent' in summary:
             current_ylim = ax.get_ylim()
             ax.set_ylim(current_ylim[0], max(current_ylim[1], summary['UniqueDocsPercent'] * 1.2))
@@ -384,7 +395,6 @@ def plot_subset_metrics(subset_name):
             )
         
         return fig
-        
     except Exception as e:
         print(f"Error plotting metrics: {str(e)}")