fixed filtering bugs for data exp

tabs/data_exploration.py

@@ -576,8 +576,25 @@ def create_exploration_tab(df):
                 )
 
             with gr.Column(scale=1, min_width=100):
+                # Get initial count from default data
+                df_initial = get_chat_and_score_df(MODELS[0], DATASETS[0])
+                initial_count = len(df_initial)
+
                 index_display = gr.HTML(
-                    value="<div style='text-align: center; color: var(--text-muted);'>0/0</div>",
+                    value=f"""<div style="
+                        display: flex;
+                        align-items: center;
+                        justify-content: center;
+                        font-weight: 500;
+                        color: var(--primary-text);
+                        background-color: var(--surface-color-alt);
+                        padding: 0.5rem 1rem;
+                        border-radius: 20px;
+                        font-size: 0.9rem;
+                        width: fit-content;
+                        margin: 0 auto;">
+                        <span style="margin-right: 0.5rem;">📄</span>1/{initial_count}
+                    </div>""",
                     elem_id="index-display",
                 )
 
@@ -602,6 +619,53 @@ def create_exploration_tab(df):
         # State for tracking current index (simple integer state)
         current_index = gr.State(value=0)
 
+        def reset_index():
+            """Reset the current index to 0"""
+            return 0
+
+        # Add these explicit event handlers for model and dataset changes
+        explore_model.change(
+            reset_index,
+            inputs=[],
+            outputs=[current_index],
+        )
+
+        explore_dataset.change(
+            reset_index,
+            inputs=[],
+            outputs=[current_index],
+        )
+
+        min_score.change(
+            reset_index,
+            inputs=[],
+            outputs=[current_index],
+        )
+
+        max_score.change(
+            reset_index,
+            inputs=[],
+            outputs=[current_index],
+        )
+
+        n_turns_filter.change(
+            reset_index,
+            inputs=[],
+            outputs=[current_index],
+        )
+
+        len_query_filter.change(
+            reset_index,
+            inputs=[],
+            outputs=[current_index],
+        )
+
+        n_tools_filter.change(
+            reset_index,
+            inputs=[],
+            outputs=[current_index],
+        )
+
         # Reset filters
         def reset_filters():
             return (
@@ -744,79 +808,3 @@ def create_exploration_tab(df):
             tool_info_display,
             index_display,
         ]
-
-
-def filter_and_update_display(model, dataset, min_score, max_score, current_index):
-    """Filter the dataset and update the display with comprehensive error handling."""
-    try:
-        df_chat = get_chat_and_score_df(model, dataset)
-        df_chat = df_chat[
-            (df_chat["score"] >= min_score) & (df_chat["score"] <= max_score)
-        ]
-
-        if df_chat.empty:
-            return (
-                '<div style="padding: 1.5rem; color: var(--text-muted); text-align: center; font-style: italic; background-color: var(--surface-color-alt); border-radius: 8px; border: 1px dashed var(--border-color);">No data available for selected filters</div>',
-                '<div style="padding: 1.5rem; color: var(--text-muted); text-align: center; font-style: italic;">No metrics available</div>',
-                '<div style="padding: 1.5rem; color: var(--text-muted); text-align: center; font-style: italic;">No tool information available</div>',
-                '<div style="font-weight: 500; color: var(--text-muted);">0/0</div>',
-            )
-
-        max_index = len(df_chat) - 1
-        current_index = min(current_index, max_index)
-        chat_html, metrics_html, tool_html = update_chat_display(df_chat, current_index)
-
-        index_display = f"""
-        <div style="
-            display: flex;
-            align-items: center;
-            justify-content: center;
-            font-weight: 500;
-            color: var(--primary-text);
-            background-color: var(--surface-color-alt);
-            padding: 0.5rem 1rem;
-            border-radius: 20px;
-            font-size: 0.9rem;
-            width: fit-content;
-            margin: 0 auto;">
-            <span style="margin-right: 0.25rem;">📄</span>{current_index + 1}/{len(df_chat)}
-        </div>
-        """
-        return chat_html, metrics_html, tool_html, index_display
-
-    except Exception as e:
-        error_html = f"""
-        <div style="
-            padding: 1.5rem;
-            color: var(--score-low);
-            background-color: var(--surface-color);
-            border: 1px solid var(--score-low);
-            border-radius: 8px;
-            display: flex;
-            align-items: flex-start;">
-            <div style="
-                flex-shrink: 0;
-                margin-right: 1rem;
-                font-size: 1.5rem;">⚠️</div>
-            <div>
-                <div style="
-                    font-weight: 600;
-                    margin-bottom: 0.5rem;">Error Occurred</div>
-                <div style="
-                    font-family: monospace;
-                    background-color: var(--surface-color-alt);
-                    padding: 1rem;
-                    border-radius: 4px;
-                    white-space: pre-wrap;
-                    font-size: 0.9rem;">
-                    {str(e)}
-                </div>
-            </div>
-        </div>
-        """
-        return (
-            error_html,
-            '<div style="padding: 1.5rem; color: var(--text-muted); text-align: center; font-style: italic;">No metrics available</div>',
-            '<div style="padding: 1.5rem; color: var(--text-muted); text-align: center; font-style: italic;">No tool information available</div>',
-            '<div style="font-weight: 500; color: var(--text-muted);">0/0</div>',
-        )