Code Changes - Reranking Implementation

app.py

@@ -114,6 +114,67 @@ def get_info_from_sample_questions(subset, question):
     return "No answer found.", "No metrics found."
 
 def plot_subset_metrics(subset_name):
+    """Generate a bar plot of key metrics for a given subset with proper title display."""
+    try:
+        summary = ragbench_details[subset_name]["summary"]
+        
+        # Metrics to plot
+        metrics = {
+            'Entries': 'Total Entries',
+            'TotalDocs': 'Total Documents',
+            'TotalUniqueIds': 'Unique IDs', 
+            'TotalUniqueDocs': 'Unique Documents',
+            'UniqueDocsPercent': '% Unique Docs'
+        }
+        
+        # Prepare data
+        display_names = list(metrics.values())
+        values = [summary.get(metric, 0) for metric in metrics.keys()]
+        
+        # Create figure with constrained layout
+        fig, ax = plt.subplots(figsize=(10, 6), constrained_layout=True)
+        
+        # Plot bars
+        bars = ax.bar(display_names, values, color='skyblue')
+        
+        # Add value labels
+        ax.bar_label(bars, fmt='%.0f', padding=3, fontsize=9)
+        
+        # Title adjustments
+        title = f"Dataset Metrics: {subset_name}"
+        if 'Domain' in summary:
+            title += f"\nDomain: {summary['Domain']}"
+            
+        ax.set_title(title, fontsize=12, pad=20, loc='left')  # Explicit title positioning
+        
+        # Axis formatting
+        ax.set_ylabel("Count", fontsize=10)
+        ax.grid(axis='y', linestyle=':', alpha=0.6)
+        
+        # Rotate x-labels
+        ax.set_xticks(range(len(display_names)))
+        ax.set_xticklabels(display_names, rotation=25, ha='right', fontsize=9)
+        
+        # Special handling for percentage value
+        if 'UniqueDocsPercent' in summary:
+            ax.text(
+                len(metrics)-1, 
+                summary['UniqueDocsPercent'], 
+                f"{summary['UniqueDocsPercent']}%",
+                ha='center', 
+                va='bottom', 
+                fontsize=10,
+                bbox=dict(facecolor='white', alpha=0.8, edgecolor='none')
+            )
+        
+        return fig
+        
+    except KeyError:
+        print(f"Error: Subset '{subset_name}' not found")
+    except Exception as e:
+        print(f"Plotting error: {str(e)}")
+
+def plot_subset_metrics_old(subset_name):
     summary = ragbench_details[subset_name]["summary"]
 
     # Create a DataFrame for plotting