Update app.py

app.py

@@ -580,6 +580,31 @@ political_columns = [
     }
 ]
 
+template_column = {
+    "field": "Prompt Template",
+    "headerName": "Template",
+    "width": 150,
+    "filter": "agTextColumnFilter",
+    "filterParams": {
+        "defaultOption": "contains",
+        "filterOptions": ['contains', 'notContains', 'startsWith', 'endsWith']
+    },
+    "headerClass": "ag-left-aligned-header wrap-text",
+    "cellClass": "ag-left-aligned-cell",
+    "wrapHeaderText": True,
+    "autoHeaderHeight": True,
+    "comparator": {
+        "function": """
+            function(valueA, valueB) {
+                if (!valueA && !valueB) return 0;
+                if (!valueA) return 1;
+                if (!valueB) return -1;
+                return valueA.toLowerCase().localeCompare(valueB.toLowerCase());
+            }
+        """
+    }
+}
+
 # Add axes columns with different widths
 for i, col in enumerate(AXES_COLS_1):
     col_def = {
@@ -815,20 +840,32 @@ app.layout = html.Div([
     
     # Additional Columns Filter
     html.Div([
-        html.Label("Show Additional Columns:", 
-            className="model-type-filter"),  # Use same class for consistent styling
-        dcc.Checklist(
-            id='additional-columns-filter',
-            options=[
-                {'label': 'UGI Categories', 'value': 'ugi_categories'},
-                {'label': 'W/10 Types', 'value': 'w10_types'},
-                {'label': 'Political Test Axes', 'value': 'political_axes'}
-            ],
-            value=[],
-            inline=True,
-            style={'display': 'inline-block'},
-            labelStyle={'fontWeight': 'normal', 'marginRight': '15px'}
-        )
+        html.Div([
+            html.Label("Show Additional Columns:", 
+                className="model-type-filter"),
+            dcc.Checklist(
+                id='additional-columns-filter',
+                options=[
+                    {'label': 'UGI Categories', 'value': 'ugi_categories'},
+                    {'label': 'W/10 Types', 'value': 'w10_types'},
+                    {'label': 'Political Test Axes', 'value': 'political_axes'}
+                ],
+                value=[],
+                inline=True,
+                style={'display': 'inline-block'},
+                labelStyle={'fontWeight': 'normal', 'marginRight': '15px'}
+            )
+        ], style={'float': 'left'}),
+        html.Div([
+            dcc.Checklist(
+                id='template-filter',
+                options=[{'label': 'Template', 'value': 'template'}],
+                value=[],
+                inline=True,
+                style={'display': 'inline-block'},
+                labelStyle={'fontWeight': 'normal'}
+            )
+        ], style={'float': 'right'})
     ], style={'marginBottom': '13px', 'padding': '0 20px', 'overflow': 'hidden'}),
     
     # Grid
@@ -1044,33 +1081,35 @@ def update_grid(selected_types, show_na, pinned_rows):
 
 @app.callback(
     Output('leaderboard-grid', 'columnDefs'),
-    [Input('additional-columns-filter', 'value')]
+    [Input('additional-columns-filter', 'value'),
+     Input('template-filter', 'value')]
 )
-def update_columns(additional_columns):
+def update_columns(additional_columns, template_filter):
     # Start with base columns up to UGI column
-    current_columns = columnDefs[:7]  # Include up to UGI column
+    current_columns = columnDefs[:6]  # Include up to Model column
+    
+    # Add Template column if selected
+    if template_filter and 'template' in template_filter:
+        current_columns.append(template_column)
+    
+    # Rest of the function remains the same...
+    current_columns.extend(columnDefs[6:7])
     
-    # Add UGI category columns if selected
     if 'ugi_categories' in additional_columns:
         current_columns.extend(ugi_category_columns)
     
-    # Add W/10 column
-    current_columns.extend(columnDefs[7:8])  # Add just the W/10 column
+    current_columns.extend(columnDefs[7:8])
     
-    # Add W/10 type columns if selected
     if 'w10_types' in additional_columns:
         current_columns.extend(w10_type_columns)
     
-    # Add remaining base columns (NatInt, Coding, Political Lean)
     current_columns.extend(columnDefs[8:11])
     
-    # Add political columns if selected
     if 'political_axes' in additional_columns:
         current_columns.extend(political_columns)
         current_columns.extend([col for col in columnDefs if col['field'] in AXES_COLS_1])
         current_columns.extend([col for col in columnDefs if col['field'] in AXES_COLS_2])
     
-    # Always add date columns at the end
     current_columns.extend([col for col in columnDefs if col['field'] in ['Release Date', 'Test Date']])
     
     return current_columns